package at.gv.egiz.eaaf.core.impl.idp.auth.builder;

import java.security.InvalidKeyException;
import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.security.NoSuchAlgorithmException;
import java.security.NoSuchProviderException;

import org.apache.commons.lang3.RandomStringUtils;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.BlockJUnit4ClassRunner;

import at.gv.egiz.eaaf.core.api.data.EaafConstants;
import at.gv.egiz.eaaf.core.exceptions.EaafBuilderException;
import at.gv.egiz.eaaf.core.impl.data.Pair;

@RunWith(BlockJUnit4ClassRunner.class)
public class BpkBuilderTest {

  private static final String BASEID = "RUxHQVRlc3RQQjBYWFjFkHpnw7xyX1hYWFTDvHpla8OnaQ==";
  
  private KeyPair keyPair;
  
  
  /**
   * jUnit test initializer.
   * @throws NoSuchProviderException In case of an error
   * @throws NoSuchAlgorithmException  In case of an error
   */
  @Before
  public void initialize() throws NoSuchAlgorithmException, NoSuchProviderException {
    KeyPairGenerator keyGen = KeyPairGenerator.getInstance("RSA");
    keyPair = keyGen.generateKeyPair();
    
  }
  
  @Test
  public void encBpkWrongTarget() throws InvalidKeyException {
    String bpk = RandomStringUtils.randomAlphanumeric(25);
    String target = RandomStringUtils.randomAlphanumeric(25);
    
    try {
      BpkBuilder.encryptBpk(bpk, target, keyPair.getPublic());
      Assert.fail("Wrong parameters not detected");
      
    } catch (EaafBuilderException e) {
      Assert.assertEquals("Wrong errorMsg", "builder.32", e.getErrorId());
      
    }
  }
  
  @Test
  public void decBpkWrongTarget() throws InvalidKeyException {
    String bpk = RandomStringUtils.randomAlphanumeric(25);
    String target = RandomStringUtils.randomAlphanumeric(25);
    
    try {
      BpkBuilder.decryptBpk(bpk, target, keyPair.getPrivate());
      Assert.fail("Wrong parameters not detected");
      
    } catch (EaafBuilderException e) {
      Assert.assertEquals("Wrong errorMsg", "builder.32", e.getErrorId());
      
    }
  }
  
  @Test
  public void decBpkWrongTargetInEncBpk() throws InvalidKeyException, EaafBuilderException {
    String bpk = RandomStringUtils.randomAlphanumeric(25);
    String target = EaafConstants.URN_PREFIX_CDID + "AA";
    
    String encBpk = BpkBuilder.encryptBpk(bpk, target, keyPair.getPublic());
    try {
      BpkBuilder.decryptBpk(encBpk, 
          EaafConstants.URN_PREFIX_CDID + "BB", keyPair.getPrivate());
      Assert.fail("Wrong parameters not detected");
      
    } catch (EaafBuilderException e) {
      Assert.assertEquals("Wrong errorMsg", "builder.30", e.getErrorId());
      
    }
  }
  
  @Test
  public void encBpkSuccess() throws EaafBuilderException, InvalidKeyException {
    String bpk = RandomStringUtils.randomAlphanumeric(25);
    String target = EaafConstants.URN_PREFIX_CDID + "AA";
    
    String encBpk = BpkBuilder.encryptBpk(bpk, target, keyPair.getPublic());
    
    Assert.assertNotNull("encBpk", encBpk);
    
    Pair<String, String> decBpk = BpkBuilder.decryptBpk(encBpk, target, keyPair.getPrivate());
    
    Assert.assertEquals("wrong bBK", bpk, decBpk.getFirst());
    Assert.assertEquals("wrong bBK-Target", target, decBpk.getSecond());
    
  }
  
  @Test
  public void encWbpkSuccess() throws EaafBuilderException, InvalidKeyException {
    String bpk = RandomStringUtils.randomAlphanumeric(25);
    String target = EaafConstants.URN_PREFIX_WBPK + "XFN+123456i";
    
    String encBpk = BpkBuilder.encryptBpk(bpk, target, keyPair.getPublic());
    
    Assert.assertNotNull("encBpk", encBpk);
    
    Pair<String, String> decBpk = BpkBuilder.decryptBpk(encBpk, target, keyPair.getPrivate());
    
    Assert.assertEquals("wrong bBK", bpk, decBpk.getFirst());
    Assert.assertEquals("wrong bBK-Target", target, decBpk.getSecond());
    
  }
  
  @Test
  public void encWbpkSuccessSecond() throws EaafBuilderException, InvalidKeyException {
    String bpk = RandomStringUtils.randomAlphanumeric(25);
    String target = EaafConstants.URN_PREFIX_WBPK + "FN+123456i";
    
    String encBpk = BpkBuilder.encryptBpk(bpk, target, keyPair.getPublic());
    
    Assert.assertNotNull("encBpk", encBpk);
    
    Pair<String, String> decBpk = BpkBuilder.decryptBpk(encBpk, 
        EaafConstants.URN_PREFIX_WBPK + "XFN+123456i", keyPair.getPrivate());
    
    Assert.assertEquals("wrong bBK", bpk, decBpk.getFirst());
    Assert.assertEquals("wrong bBK-Target", 
        EaafConstants.URN_PREFIX_WBPK + "XFN+123456i", decBpk.getSecond());
    
  }
  
  
  @Test
  public void noBaseId() {
    try {
      BpkBuilder.generateAreaSpecificPersonIdentifier(null, EaafConstants.URN_PREFIX_CDID + "AA");
      
    } catch (EaafBuilderException e) {
      Assert.assertEquals("Wrong errorCode", "builder.00", e.getErrorId());
    }
  }
  
  @Test
  public void noTarget() {
    try {
      BpkBuilder.generateAreaSpecificPersonIdentifier(BASEID, null);
      
    } catch (EaafBuilderException e) {
      Assert.assertEquals("Wrong errorCode", "builder.00", e.getErrorId());
    }
  }
  
  @Test
  public void noBaseIdType() {
    try {
      BpkBuilder.generateAreaSpecificPersonIdentifier(BASEID, 
          null, EaafConstants.URN_PREFIX_CDID + "AA");
      
    } catch (EaafBuilderException e) {
      Assert.assertEquals("Wrong errorCode", "builder.00", e.getErrorId());
    }
  }
  
  @Test
  public void wrongBaseIdType() {
    try {
      BpkBuilder.generateAreaSpecificPersonIdentifier(BASEID, 
          EaafConstants.URN_PREFIX_CDID + "BB", EaafConstants.URN_PREFIX_CDID + "AA");
      
    } catch (EaafBuilderException e) {
      Assert.assertEquals("Wrong errorCode", "builder.00", e.getErrorId());
    }
  }
  
  @Test
  public void baseIdTypeEqualsTarget() throws EaafBuilderException {
    Pair<String, String> result1 =  BpkBuilder.generateAreaSpecificPersonIdentifier(BASEID, 
          EaafConstants.URN_PREFIX_CDID + "AA", EaafConstants.URN_PREFIX_CDID + "AA");
    
    Assert.assertEquals("first bPK", BASEID,
        result1.getFirst());
    Assert.assertEquals("first bPK", "urn:publicid:gv.at:cdid+AA",
        result1.getSecond());
  
  }
  
  @Test
  public void buildBpk() throws EaafBuilderException {
    
    Pair<String, String> result1 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_CDID + "AA");
    Pair<String, String> result2 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_CDID + "BB");
    
    Assert.assertEquals("first bPK", "b1Ip610zZq/Or/uCqgb51lnAdZM=",
        result1.getFirst());
    Assert.assertEquals("first bPK", "urn:publicid:gv.at:cdid+AA",
        result1.getSecond());
    
    Assert.assertEquals("second bPK", "uYst6hjKJvyp7s/ezD8zsnkcj9k=",
        result2.getFirst());
    Assert.assertEquals("second bPK", "urn:publicid:gv.at:cdid+BB",
        result2.getSecond());
    
  }
  
  @Test
  public void buildWbpkFn() throws EaafBuilderException {
    
    Pair<String, String> result1 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_WBPK + "FN+123456i");
    
    Assert.assertEquals("wbPK", "k65HRxpVcoZ2OPZHo3j2LEn/JQE=",
        result1.getFirst());
    Assert.assertEquals("wbPK", "urn:publicid:gv.at:wbpk+XFN+123456i",
        result1.getSecond());
    
  }
  
  @Test
  public void buildWbpkZvr() throws EaafBuilderException {
    
    Pair<String, String> result1 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_WBPK + "ZVR+123456");
    
    Assert.assertEquals("wbPK", "g4JRKGS+AJxd9FU8k2tG8Lxrx6M=",
        result1.getFirst());
    Assert.assertEquals("wbPK", "urn:publicid:gv.at:wbpk+XZVR+123456",
        result1.getSecond());
    
  }
  
  @Test
  public void buildWbpkErsb() throws EaafBuilderException {
    
    Pair<String, String> result1 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_WBPK + "ERSB+123456");
    
    Assert.assertEquals("wbPK", "Bjnl0BofeJGgqynJP1r/ff6E1Rk=",
        result1.getFirst());
    Assert.assertEquals("wbPK", "urn:publicid:gv.at:wbpk+XERSB+123456",
        result1.getSecond());
    
  }
  
  @Test
  public void buildWbpkXFn() throws EaafBuilderException {
    
    Pair<String, String> result1 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_WBPK + "XFN+123456i");
    
    Assert.assertEquals("wbPK", "k65HRxpVcoZ2OPZHo3j2LEn/JQE=",
        result1.getFirst());
    Assert.assertEquals("wbPK", "urn:publicid:gv.at:wbpk+XFN+123456i",
        result1.getSecond());
    
  }
  
  @Test
  public void buildWbpkXZvr() throws EaafBuilderException {
    
    Pair<String, String> result1 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_WBPK + "XZVR+123456");
    
    Assert.assertEquals("wbPK", "g4JRKGS+AJxd9FU8k2tG8Lxrx6M=",
        result1.getFirst());
    Assert.assertEquals("wbPK", "urn:publicid:gv.at:wbpk+XZVR+123456",
        result1.getSecond());
    
  }
  
  @Test
  public void buildWbpkXErsb() throws EaafBuilderException {
    
    Pair<String, String> result1 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_WBPK + "XERSB+123456");
    
    Assert.assertEquals("wbPK", "Bjnl0BofeJGgqynJP1r/ff6E1Rk=",
        result1.getFirst());
    Assert.assertEquals("wbPK", "urn:publicid:gv.at:wbpk+XERSB+123456",
        result1.getSecond());
    
  }
  
  @Test
  public void buildWbpkOthers() throws EaafBuilderException {
    
    Pair<String, String> result1 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_WBPK + "XABC+123456");
    
    Assert.assertEquals("wbPK", "wv96/xKUyi6YoYGv7IcIlFTsJIk=",
        result1.getFirst());
    Assert.assertEquals("wbPK", "urn:publicid:gv.at:wbpk+XABC+123456",
        result1.getSecond());
    
  }
  
  @Test
  public void buildEidasId() throws EaafBuilderException {
    
    Pair<String, String> result1 = BpkBuilder.generateAreaSpecificPersonIdentifier(
        BASEID, EaafConstants.URN_PREFIX_EIDAS + "AT+ES");
    
    Assert.assertEquals("eidas", "AT/ES/7AuLZNKsiRr97yvLsQ16SZ6r0q0=",
        result1.getFirst());
    Assert.assertEquals("wbPK", "urn:publicid:gv.at:eidasid+AT+ES",
        result1.getSecond());
    
  }
  
  @Test
  public void normalizeNullTarget() {
    Assert.assertNull("Wrong normalized target",  
        BpkBuilder.normalizeBpkTargetIdentifierToCommonFormat(null));
     
  }
  
  @Test
  public void normalizeBpkTarget() {
    String target = EaafConstants.URN_PREFIX_CDID + RandomStringUtils.randomAlphabetic(2);
    Assert.assertEquals("Wrong normalized target", 
        target, 
        BpkBuilder.normalizeBpkTargetIdentifierToCommonFormat(target));
    
  }
  
  @Test
  public void normalizeWbpkTargetWithX() {
    String target = EaafConstants.URN_PREFIX_WBPK_TARGET_WITH_X + RandomStringUtils.randomAlphabetic(2);
    Assert.assertEquals("Wrong normalized target", 
        target, 
        BpkBuilder.normalizeBpkTargetIdentifierToCommonFormat(target));
    
  }
  
  @Test
  public void normalizeWbpkTargetWithOutXNoMapping() {
    String target = EaafConstants.URN_PREFIX_WBPK + RandomStringUtils.randomAlphabetic(2);
    Assert.assertEquals("Wrong normalized target", 
        target, 
        BpkBuilder.normalizeBpkTargetIdentifierToCommonFormat(target));
    
  }
  
  @Test
  public void normalizeWbpkTargetWithOutXMappingFn() {
    Assert.assertEquals("Wrong normalized target", 
        EaafConstants.URN_PREFIX_WBPK + "XFN+123456i", 
        BpkBuilder.normalizeBpkTargetIdentifierToCommonFormat(EaafConstants.URN_PREFIX_WBPK + "FN+123456i"));
    
  }
  
  @Test
  public void normalizeWbpkTargetWithOutXMappingZvr() {
    Assert.assertEquals("Wrong normalized target", 
        EaafConstants.URN_PREFIX_WBPK + "XZVR+1122334455", 
        BpkBuilder.normalizeBpkTargetIdentifierToCommonFormat(EaafConstants.URN_PREFIX_WBPK + "ZVR+1122334455"));
    
  }
  
  @Test
  public void normalizeWbpkTargetWithOutXMappingErsb() {
    Assert.assertEquals("Wrong normalized target", 
        EaafConstants.URN_PREFIX_WBPK + "XERSB+998877665544", 
        BpkBuilder.normalizeBpkTargetIdentifierToCommonFormat(EaafConstants.URN_PREFIX_WBPK + "ERSB+998877665544"));
    
  }
  
  @Test
  public void normalizeEidasTarget() {
    String target = EaafConstants.URN_PREFIX_EIDAS + RandomStringUtils.randomAlphabetic(2) 
        + "+" + RandomStringUtils.randomAlphabetic(2);
    Assert.assertEquals("Wrong normalized target", 
        target, 
        BpkBuilder.normalizeBpkTargetIdentifierToCommonFormat(target));
    
  }
  
  @Test
  public void calcNormalizeNullTarget() {
    Assert.assertNull("Wrong normalized target",  
        BpkBuilder.normalizeBpkTargetIdentifierToCalculationFormat(null));
     
  }
  
  @Test
  public void calcNormalizeBpkTarget() {
    String target = EaafConstants.URN_PREFIX_CDID + RandomStringUtils.randomAlphabetic(2);
    Assert.assertEquals("Wrong normalized target", 
        target, 
        BpkBuilder.normalizeBpkTargetIdentifierToCalculationFormat(target));
    
  }
  
  @Test
  public void calcNormalizeWbpkTargetWithoutX() {

    Assert.assertEquals("Wrong normalized target", 
        EaafConstants.URN_PREFIX_WBPK + "FN+123456i", 
        BpkBuilder.normalizeBpkTargetIdentifierToCalculationFormat(EaafConstants.URN_PREFIX_WBPK + "FN+123456i"));
    
  }
  
  @Test
  public void calcNormalizeWbpkTargetWithOutXNoMapping() {
    String target = EaafConstants.URN_PREFIX_WBPK + RandomStringUtils.randomAlphabetic(2);
    Assert.assertEquals("Wrong normalized target", 
        target, 
        BpkBuilder.normalizeBpkTargetIdentifierToCalculationFormat(target));
    
  }
  
  @Test
  public void calcNormalizeWbpkTargetWithXMappingFn() {
    Assert.assertEquals("Wrong normalized target", 
        EaafConstants.URN_PREFIX_WBPK + "FN+123456i", 
        BpkBuilder.normalizeBpkTargetIdentifierToCalculationFormat(EaafConstants.URN_PREFIX_WBPK + "XFN+123456i"));
    
  }
  
  @Test
  public void calcNormalizeWbpkTargetWithXMappingZvr() {
    Assert.assertEquals("Wrong normalized target", 
        EaafConstants.URN_PREFIX_WBPK + "ZVR+1122334455", 
        BpkBuilder.normalizeBpkTargetIdentifierToCalculationFormat(EaafConstants.URN_PREFIX_WBPK + "XZVR+1122334455"));
    
  }
  
  @Test
  public void calcNormalizeWbpkTargetWithXMappingErsb() {
    Assert.assertEquals("Wrong normalized target", 
        EaafConstants.URN_PREFIX_WBPK + "ERSB+998877665544", 
        BpkBuilder.normalizeBpkTargetIdentifierToCalculationFormat(
            EaafConstants.URN_PREFIX_WBPK + "XERSB+998877665544"));
    
  }
  
  @Test
  public void calcNormalizeEidasTarget() {
    String target = EaafConstants.URN_PREFIX_EIDAS + RandomStringUtils.randomAlphabetic(2) 
        + "+" + RandomStringUtils.randomAlphabetic(2);
    Assert.assertEquals("Wrong normalized target", 
        target, 
        BpkBuilder.normalizeBpkTargetIdentifierToCalculationFormat(target));
    
  }
  
}