[jboss-identity-commits] JBoss Identity SVN: r207 - identity-federation/trunk/identity-fed-model/src/test/java/org/jboss/test/identity/federation/ws/trust.

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

Author: sguilhen at redhat.com
Date: 2009-01-08 08:17:48 -0500 (Thu, 08 Jan 2009)
New Revision: 207

Modified:
   identity-federation/trunk/identity-fed-model/src/test/java/org/jboss/test/identity/federation/ws/trust/UnmarshallingTestCase.java
Log:
Included a test scenario for security token request collection

Modified: identity-federation/trunk/identity-fed-model/src/test/java/org/jboss/test/identity/federation/ws/trust/UnmarshallingTestCase.java
===================================================================
--- identity-federation/trunk/identity-fed-model/src/test/java/org/jboss/test/identity/federation/ws/trust/UnmarshallingTestCase.java	2009-01-07 22:07:09 UTC (rev 206)
+++ identity-federation/trunk/identity-fed-model/src/test/java/org/jboss/test/identity/federation/ws/trust/UnmarshallingTestCase.java	2009-01-08 13:17:48 UTC (rev 207)
@@ -37,6 +37,7 @@
 
 import junit.framework.TestCase;
 
+import org.jboss.identity.federation.ws.trust.RequestSecurityTokenCollectionType;
 import org.jboss.identity.federation.ws.trust.RequestSecurityTokenType;
 
 /**
@@ -76,8 +77,8 @@
       // validate the request contents.
       RequestSecurityTokenType request = (RequestSecurityTokenType) element.getValue();
       List<Object> contents = request.getAny();
-      assertNotNull("Unexpected null request contents", contents);
-      assertEquals("Unexpected number of elements", 2, contents.size());
+      assertNotNull("Unexpected null value for the request contents", contents);
+      assertEquals("Unexpected number of contents", 2, contents.size());
 
       // first element should be TokenType.
       JAXBElement<String> tokenType = (JAXBElement<String>) contents.get(0);
@@ -90,18 +91,48 @@
       assertEquals("http://docs.oasis-open.org/ws-sx/ws-trust/200512/Issue", requestType.getValue());
    }
 
+   /**
+    * <p>
+    * Tests unmarshalling a ws-trust request for a collection of security tokens.
+    * </p>
+    * 
+    * @throws Exception if an error occurs while running the test.
+    */
+   @SuppressWarnings("unchecked")
    public void testUnmarshallTokenCollectionRequest() throws Exception
    {
-      JAXBContext context = JAXBContext
-            .newInstance("org.jboss.identity.federation.ws.trust");
+      JAXBContext context = JAXBContext.newInstance("org.jboss.identity.federation.ws.trust");
       Unmarshaller unmarshaller = context.createUnmarshaller();
       // this.setValidatingSchema("/schema/wstrust/v1_3/ws-trust-1.3.xsd", unmarshaller);
 
       // unmarshall the sample security token request.
       URI sampleURI = this.getClass().getResource("/wstrust/collection-request.xml").toURI();
       Object object = unmarshaller.unmarshal(new File(sampleURI));
+      assertNotNull("Unexpected null object", object);
+      assertTrue("Unexpected object type", object instanceof JAXBElement);
 
       JAXBElement element = (JAXBElement) object;
+      assertEquals("Invalid element name", "RequestSecurityTokenCollection", element.getName().getLocalPart());
+      assertEquals("Invalid element type", RequestSecurityTokenCollectionType.class, element.getDeclaredType());
+
+      List<RequestSecurityTokenType> requests = ((RequestSecurityTokenCollectionType) element.getValue())
+            .getRequestSecurityToken();
+      assertNotNull("Unexpected null request list", requests);
+      assertEquals("Unexpected number of requests", 2, requests.size());
+      
+      // first request must have the http://www.example.com/1 context.
+      RequestSecurityTokenType request = requests.get(0);
+      assertEquals("Invalid context id", "http://www.example.com/1", request.getContext());
+      List<Object> contents = request.getAny();
+      assertNotNull("Unexpected null value for the request contents", contents);
+      assertEquals("Unexpected number of contents", 4, contents.size());
+
+      // second request must have the http://www.example.com/2 context.
+      request = requests.get(1);
+      assertEquals("Invalid context id", "http://www.example.com/2", request.getContext());
+      contents = request.getAny();
+      assertNotNull("Unexpected null value for the request contents", contents);
+      assertEquals("Unexpected number of contents", 4, contents.size());
    }
 
    /**