[jboss-cvs] Picketlink SVN: r1131 - federation/trunk/picketlink-fed-api/src/test/java/org/picketlink/test/identity/federation/api/saml/v2.

[jboss-cvs-commits at lists.jboss.org]

Author: anil.saldhana at jboss.com
Date: 2011-07-26 14:56:56 -0400 (Tue, 26 Jul 2011)
New Revision: 1131

Modified:
   federation/trunk/picketlink-fed-api/src/test/java/org/picketlink/test/identity/federation/api/saml/v2/SAML2ResponseUnitTestCase.java
Log:
test signing the response type

Modified: federation/trunk/picketlink-fed-api/src/test/java/org/picketlink/test/identity/federation/api/saml/v2/SAML2ResponseUnitTestCase.java
===================================================================
--- federation/trunk/picketlink-fed-api/src/test/java/org/picketlink/test/identity/federation/api/saml/v2/SAML2ResponseUnitTestCase.java	2011-07-26 17:28:17 UTC (rev 1130)
+++ federation/trunk/picketlink-fed-api/src/test/java/org/picketlink/test/identity/federation/api/saml/v2/SAML2ResponseUnitTestCase.java	2011-07-26 18:56:56 UTC (rev 1131)
@@ -24,17 +24,42 @@
 import static org.junit.Assert.assertNotNull;
 
 import java.io.InputStream;
+import java.security.Key;
+import java.security.KeyPair;
+import java.security.KeyStore;
+import java.security.PrivateKey;
+import java.security.cert.Certificate;
 
 import org.junit.Test;
 import org.picketlink.identity.federation.api.saml.v2.response.SAML2Response;
+import org.picketlink.identity.federation.api.saml.v2.sig.SAML2Signature;
+import org.picketlink.identity.federation.core.saml.v2.util.DocumentUtil;
 import org.picketlink.identity.federation.saml.v2.SAML2Object;
+import org.picketlink.identity.federation.saml.v2.protocol.ResponseType;
+import org.picketlink.test.identity.federation.api.util.KeyUtilUnitTestCase;
+import org.w3c.dom.Document;
 
 /**
+ * Parse a {@link ResponseType} that contains A
  * @author Anil.Saldhana at redhat.com
  * @since Jul 21, 2011
  */
 public class SAML2ResponseUnitTestCase
 {
+
+   private final String keystoreLocation = "keystore/jbid_test_keystore.jks";
+
+   private final String keystorePass = "store123";
+
+   private final String keyPass = "test123";
+
+   private final String alias = "servercert";
+
+   /**
+    * Parse a {@link ResponseType} that contains ADFS Claims
+    * and then try to sign
+    * @throws Exception
+    */
    @Test
    public void parseADFSClaims() throws Exception
    {
@@ -43,6 +68,34 @@
       SAML2Response samlResponse = new SAML2Response();
       SAML2Object samlObject = samlResponse.getSAML2ObjectFromStream(configStream);
       assertNotNull(samlObject);
+
+      SAML2Signature sig = new SAML2Signature();
+      Document signedDoc = sig.sign((ResponseType) samlObject, getKeyPair());
+      assertNotNull(signedDoc);
+
+      System.out.println("Signed Response=" + DocumentUtil.asString(signedDoc));
    }
 
+   /**
+    * @see {@link KeyUtilUnitTestCase}
+    * @return
+    * @throws Exception
+    */
+   private KeyPair getKeyPair() throws Exception
+   {
+      ClassLoader tcl = Thread.currentThread().getContextClassLoader();
+      InputStream ksStream = tcl.getResourceAsStream(keystoreLocation);
+      assertNotNull("Input keystore stream is not null", ksStream);
+
+      KeyStore ks = KeyStore.getInstance(KeyStore.getDefaultType());
+      ks.load(ksStream, keystorePass.toCharArray());
+      assertNotNull("KeyStore is not null", ks);
+
+      Certificate cert = ks.getCertificate(alias);
+      assertNotNull("Cert not null", cert);
+
+      // Get private key 
+      Key key = ks.getKey(alias, keyPass.toCharArray());
+      return new KeyPair(cert.getPublicKey(), (PrivateKey) key);
+   }
 }
\ No newline at end of file