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

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

Author: anil.saldhana at jboss.com
Date: 2012-02-27 14:10:45 -0500 (Mon, 27 Feb 2012)
New Revision: 1446

Modified:
   federation/trunk/picketlink-fed-core/src/test/java/org/picketlink/test/identity/federation/core/saml/v2/metadata/FileBasedMetadataConfigurationStoreUnitTestCase.java
Log:
hack for virtual images

Modified: federation/trunk/picketlink-fed-core/src/test/java/org/picketlink/test/identity/federation/core/saml/v2/metadata/FileBasedMetadataConfigurationStoreUnitTestCase.java
===================================================================
--- federation/trunk/picketlink-fed-core/src/test/java/org/picketlink/test/identity/federation/core/saml/v2/metadata/FileBasedMetadataConfigurationStoreUnitTestCase.java	2012-02-27 18:28:39 UTC (rev 1445)
+++ federation/trunk/picketlink-fed-core/src/test/java/org/picketlink/test/identity/federation/core/saml/v2/metadata/FileBasedMetadataConfigurationStoreUnitTestCase.java	2012-02-27 19:10:45 UTC (rev 1446)
@@ -29,58 +29,61 @@
 import java.util.HashMap;
 import java.util.Map;
 
+import org.junit.Before;
 import org.junit.Test;
 import org.picketlink.identity.federation.core.parsers.saml.SAMLParser;
 import org.picketlink.identity.federation.core.saml.v2.metadata.store.FileBasedMetadataConfigurationStore;
+import org.picketlink.identity.federation.core.util.StringUtil;
 import org.picketlink.identity.federation.saml.v2.metadata.EntityDescriptorType;
 
-
 /**
  * Unit test the FileBasedMetadataConfigurationStore
  * @author Anil.Saldhana at redhat.com
  * @since Apr 28, 2009
  */
-public class FileBasedMetadataConfigurationStoreUnitTestCase 
+public class FileBasedMetadataConfigurationStoreUnitTestCase
 {
    String pkgName = "org.picketlink.identity.federation.saml.v2.metadata";
+
    String id = "test";
-   
+
+   @Before
+   public void setup() throws Exception
+   {
+      String userHome = System.getProperty("user.home");
+      if (StringUtil.isNotNull(userHome) && "?".equals(userHome))
+         System.setProperty("user.home", System.getProperty("user.dir"));
+   }
+
    @Test
    public void testStore() throws Exception
    {
       SAMLParser parser = new SAMLParser();
-      
+
       ClassLoader tcl = Thread.currentThread().getContextClassLoader();
-      InputStream is = 
-         tcl.getResourceAsStream("saml2/metadata/idp-entitydescriptor.xml");
-      assertNotNull("Inputstream not null", is); 
-      
+      InputStream is = tcl.getResourceAsStream("saml2/metadata/idp-entitydescriptor.xml");
+      assertNotNull("Inputstream not null", is);
+
       EntityDescriptorType edt = (EntityDescriptorType) parser.parse(is);
-      assertNotNull( edt );
-      /*
-      Unmarshaller un = JAXBUtil.getUnmarshaller(pkgName);
-      JAXBElement<EntityDescriptorType> je = (JAXBElement<EntityDescriptorType>) un.unmarshal(is);
-      EntityDescriptorType edt = je.getValue();
-      assertNotNull("EntityDescriptorType not null", edt);  
-      */
+      assertNotNull(edt);
       FileBasedMetadataConfigurationStore fbd = new FileBasedMetadataConfigurationStore();
       fbd.persist(edt, id);
-      
+
       EntityDescriptorType loaded = fbd.load(id);
       assertNotNull("loaded EntityDescriptorType not null", loaded);
       fbd.delete(id);
-     
+
       try
       {
          fbd.load(id);
          fail("Did not delete the metadata persistent file");
       }
-      catch(Exception t)
+      catch (Exception t)
       {
          //pass
       }
    }
-   
+
    @Test
    public void testTrustedProviders() throws Exception
    {
@@ -89,22 +92,22 @@
       trustedProviders.put("idp1", "http://localhost:8080/idp1/metadata");
       trustedProviders.put("idp2", "http://localhost:8080/idp2/metadata");
       fbd.persistTrustedProviders(id, trustedProviders);
-      
+
       //Lets get back
       Map<String, String> loadTP = fbd.loadTrustedProviders(id);
       assertNotNull("Loaded Trusted Providers not null", loadTP);
-      
+
       assertTrue("idp1", loadTP.containsKey("idp1"));
       assertTrue("idp2", loadTP.containsKey("idp2"));
       assertTrue("size 2", loadTP.size() == 2);
-      
+
       fbd.deleteTrustedProviders(id);
       try
       {
          fbd.loadTrustedProviders(id);
          fail("Did not delete the trusted providers file");
       }
-      catch(Exception t)
+      catch (Exception t)
       {
          //pass
       }