package test.at.gv.egovernment.moa.id.auth.parser;

import at.gv.egovernment.moa.id.auth.builder.SAMLArtifactBuilder;
import at.gv.egovernment.moa.id.auth.parser.SAMLArtifactParser;
import at.gv.egovernment.moa.id.util.Random;
import test.at.gv.egovernment.moa.id.UnitTestCase;

/*
 * @author Paul Ivancsics
 * @version $Id$
 */
public class SAMLArtifactParserTest extends UnitTestCase {
  
  private static String URL1 = "http://moa.gv.at/auth";
  private static String URL2 = "https://moa.gv.at/auth";
  
  public SAMLArtifactParserTest(String name) {
    super(name);
  }

  public void testParseTypeCode() throws Exception {
    String sessionID = Random.nextRandom();
    String samlArtifact = new SAMLArtifactBuilder().build(URL1, sessionID); 
    byte[] typeCode = new SAMLArtifactParser(samlArtifact).parseTypeCode();
    assertEquals(typeCode[0], 0);
    assertEquals(typeCode[1], 1);
  }
  public void testParseAssertionHandleSameSessionID() throws Exception {
    // SAML artifacts for different authURL's but same sessionID MUST give same assertion handle
    String sessionID = Random.nextRandom();
    String samlArtifact1 = new SAMLArtifactBuilder().build(URL1, sessionID);
    String samlArtifact2 = new SAMLArtifactBuilder().build(URL2, sessionID);
    String assertionHandle1 = new SAMLArtifactParser(samlArtifact1).parseAssertionHandle();
    String assertionHandle2 = new SAMLArtifactParser(samlArtifact2).parseAssertionHandle();
    assertEquals(assertionHandle1, assertionHandle2);
  }
  public void testParseAssertionHandleSameURL() throws Exception {
    // SAML artifacts for same authURL but different sessionID's MUST give different assertion handles
    String sessionID1 = Random.nextRandom();
    String sessionID2 = Random.nextRandom();
    String samlArtifact1 = new SAMLArtifactBuilder().build(URL1, sessionID1);
    String samlArtifact2 = new SAMLArtifactBuilder().build(URL1, sessionID2);
    String assertionHandle1 = new SAMLArtifactParser(samlArtifact1).parseAssertionHandle();
    String assertionHandle2 = new SAMLArtifactParser(samlArtifact2).parseAssertionHandle();
    assertFalse(assertionHandle1.equals(assertionHandle2));
  }
  public void testParseAssertionHandleSameSAMLArtifact() throws Exception {
    // SAML artifact parsed twice MUST give same assertion handle each time
    String sessionID = Random.nextRandom();
    String samlArtifact = new SAMLArtifactBuilder().build(URL1, sessionID);
    String assertionHandle1 = new SAMLArtifactParser(samlArtifact).parseAssertionHandle();
    String assertionHandle2 = new SAMLArtifactParser(samlArtifact).parseAssertionHandle();
    assertEquals(assertionHandle1, assertionHandle2);
  }
}