X-Git-Url: http://www.project-moonshot.org/gitweb/?a=blobdiff_plain;f=samltest%2Fsaml2%2Fmetadata%2FFilesystemMetadataProviderTest.h;h=3e736baeffe841527fa51b77b1060dcb79ac9158;hb=af8396d073111195d2bfcbb8d7bce2f7611f3595;hp=049ab314a4e13b4839b0cf4de3fde5ecbaa6130d;hpb=e8a002ede2ec05e3d12bdebbf0d8cfc87df8a1ad;p=shibboleth%2Fopensaml2.git

diff --git a/samltest/saml2/metadata/FilesystemMetadataProviderTest.h b/samltest/saml2/metadata/FilesystemMetadataProviderTest.h
index 049ab31..3e736ba 100644
--- a/samltest/saml2/metadata/FilesystemMetadataProviderTest.h
+++ b/samltest/saml2/metadata/FilesystemMetadataProviderTest.h
@@ -15,54 +15,132 @@
  */
 
 #include "internal.h"
+#include <saml/SAMLConfig.h>
+#include <saml/saml2/binding/SAML2ArtifactType0004.h>
+#include <saml/saml2/metadata/Metadata.h>
 #include <saml/saml2/metadata/MetadataProvider.h>
 
 using namespace opensaml::saml2md;
+using namespace opensaml::saml2p;
 
 class FilesystemMetadataProviderTest : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {
     XMLCh* entityID;
+    XMLCh* entityID2;
     XMLCh* supportedProtocol;
     XMLCh* supportedProtocol2;
-    MetadataProvider* metadataProvider;
 
 public:
     void setUp() {
         entityID=XMLString::transcode("urn:mace:incommon:washington.edu");
+        entityID2=XMLString::transcode("urn:mace:incommon:rochester.edu");
         supportedProtocol=XMLString::transcode("urn:oasis:names:tc:SAML:1.1:protocol");
         supportedProtocol2=XMLString::transcode("urn:mace:shibboleth:1.0");
-        
-        auto_ptr_XMLCh MP("MetadataProvider");
-        auto_ptr_XMLCh path("path");
-        auto_ptr_XMLCh validate("validate");
-        string s=data_path + "saml2/metadata/InCommon-metadata.xml";
-        auto_ptr_XMLCh file(s.c_str());
-        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().newDocument();
-        XercesJanitor<DOMDocument> janitor(doc);
-        DOMElement* root=doc->createElementNS(NULL,MP.get());
-        root->setAttributeNS(NULL,path.get(),file.get());
-        root->setAttributeNS(NULL,validate.get(),XMLConstants::XML_ZERO);
-        metadataProvider = NULL;
-        metadataProvider = SAMLConfig::getConfig().MetadataProviderManager.newPlugin(FILESYSTEM_METADATA_PROVIDER,root);
-        metadataProvider->init();
-        
         SAMLObjectBaseTestCase::setUp();
     }
     
     void tearDown() {
         XMLString::release(&entityID);
+        XMLString::release(&entityID2);
         XMLString::release(&supportedProtocol);
-        delete metadataProvider;
+        XMLString::release(&supportedProtocol2);
         SAMLObjectBaseTestCase::tearDown();
     }
 
-    void testEntityDescriptor() {
-        Locker locker(metadataProvider);
+    void testFilesystemProvider() {
+        string config = data_path + "saml2/metadata/FilesystemMetadataProvider.xml";
+        ifstream in(config.c_str());
+        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);
+        XercesJanitor<DOMDocument> janitor(doc);
+
+        auto_ptr_XMLCh path("path");
+        string s = data_path + "saml2/metadata/InCommon-metadata.xml";
+        auto_ptr_XMLCh file(s.c_str());
+        doc->getDocumentElement()->setAttributeNS(NULL,path.get(),file.get());
+
+        auto_ptr<MetadataProvider> metadataProvider(
+            SAMLConfig::getConfig().MetadataProviderManager.newPlugin(FILESYSTEM_METADATA_PROVIDER,doc->getDocumentElement())
+            );
+        try {
+            metadataProvider->init();
+        }
+        catch (XMLToolingException& ex) {
+            TS_TRACE(ex.what());
+            throw;
+        }
+        
+        Locker locker(metadataProvider.get());
         const EntityDescriptor* descriptor = metadataProvider->getEntityDescriptor(entityID);
         TSM_ASSERT("Retrieved entity descriptor was null", descriptor!=NULL);
         assertEquals("Entity's ID does not match requested ID", entityID, descriptor->getEntityID());
         TSM_ASSERT_EQUALS("Unexpected number of roles", 1, descriptor->getIDPSSODescriptors().size());
         TSM_ASSERT("Role lookup failed", descriptor->getIDPSSODescriptor(supportedProtocol)!=NULL);
         TSM_ASSERT("Role lookup failed", descriptor->getIDPSSODescriptor(supportedProtocol2)!=NULL);
+
+        auto_ptr<SAML2ArtifactType0004> artifact(
+            new SAML2ArtifactType0004(SAMLConfig::getConfig().hashSHA1("urn:mace:incommon:washington.edu"),1)
+            );
+        descriptor = metadataProvider->getEntityDescriptor(artifact.get());
+        TSM_ASSERT("Retrieved entity descriptor was null", descriptor!=NULL);
+        assertEquals("Entity's ID does not match requested ID", entityID, descriptor->getEntityID());
     }
 
+    void testFilesystemWithBlacklists() {
+        string config = data_path + "saml2/metadata/FilesystemWithBlacklists.xml";
+        ifstream in(config.c_str());
+        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);
+        XercesJanitor<DOMDocument> janitor(doc);
+
+        auto_ptr_XMLCh path("path");
+        string s = data_path + "saml2/metadata/InCommon-metadata.xml";
+        auto_ptr_XMLCh file(s.c_str());
+        doc->getDocumentElement()->setAttributeNS(NULL,path.get(),file.get());
+
+        auto_ptr<MetadataProvider> metadataProvider(
+            SAMLConfig::getConfig().MetadataProviderManager.newPlugin(FILESYSTEM_METADATA_PROVIDER,doc->getDocumentElement())
+            );
+        try {
+            metadataProvider->init();
+        }
+        catch (XMLToolingException& ex) {
+            TS_TRACE(ex.what());
+            throw;
+        }
+
+        Locker locker(metadataProvider.get());
+        const EntityDescriptor* descriptor = metadataProvider->getEntityDescriptor(entityID);
+        TSM_ASSERT("Retrieved entity descriptor was not null", descriptor==NULL);
+        descriptor = metadataProvider->getEntityDescriptor(entityID2);
+        TSM_ASSERT("Retrieved entity descriptor was null", descriptor!=NULL);
+        assertEquals("Entity's ID does not match requested ID", entityID2, descriptor->getEntityID());
+    }
+
+    void testFilesystemWithWhitelists() {
+        string config = data_path + "saml2/metadata/FilesystemWithWhitelists.xml";
+        ifstream in(config.c_str());
+        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);
+        XercesJanitor<DOMDocument> janitor(doc);
+
+        auto_ptr_XMLCh path("path");
+        string s = data_path + "saml2/metadata/InCommon-metadata.xml";
+        auto_ptr_XMLCh file(s.c_str());
+        doc->getDocumentElement()->setAttributeNS(NULL,path.get(),file.get());
+
+        auto_ptr<MetadataProvider> metadataProvider(
+            SAMLConfig::getConfig().MetadataProviderManager.newPlugin(FILESYSTEM_METADATA_PROVIDER,doc->getDocumentElement())
+            );
+        try {
+            metadataProvider->init();
+        }
+        catch (XMLToolingException& ex) {
+            TS_TRACE(ex.what());
+            throw;
+        }
+
+        Locker locker(metadataProvider.get());
+        const EntityDescriptor* descriptor = metadataProvider->getEntityDescriptor(entityID2);
+        TSM_ASSERT("Retrieved entity descriptor was not null", descriptor==NULL);
+        descriptor = metadataProvider->getEntityDescriptor(entityID);
+        TSM_ASSERT("Retrieved entity descriptor was null", descriptor!=NULL);
+        assertEquals("Entity's ID does not match requested ID", entityID, descriptor->getEntityID());
+    }
 };