Extend KeyResolver to include certificate resolution, add inline resolver.

[shibboleth/cpp-xmltooling.git] / xmltoolingtest / InlineKeyResolverTest.h

diff --git a/xmltoolingtest/InlineKeyResolverTest.h b/xmltoolingtest/InlineKeyResolverTest.h
new file mode 100644 (file)
index 0000000..1c9e39e
--- /dev/null
+++ b/xmltoolingtest/InlineKeyResolverTest.h
@@ -0,0 +1,59 @@
+/*\r
+ *  Copyright 2001-2005 Internet2\r
+ * \r
+ * Licensed under the Apache License, Version 2.0 (the "License");\r
+ * you may not use this file except in compliance with the License.\r
+ * You may obtain a copy of the License at\r
+ *\r
+ *     http://www.apache.org/licenses/LICENSE-2.0\r
+ *\r
+ * Unless required by applicable law or agreed to in writing, software\r
+ * distributed under the License is distributed on an "AS IS" BASIS,\r
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+ * See the License for the specific language governing permissions and\r
+ * limitations under the License.\r
+ */\r
+\r
+#include "XMLObjectBaseTestCase.h"\r
+\r
+#include <fstream>\r
+#include <xmltooling/signature/KeyResolver.h>\r
+\r
+using namespace xmlsignature;\r
+\r
+class InlineKeyResolverTest : public CxxTest::TestSuite {\r
+    KeyResolver* m_resolver;\r
+public:\r
+    InlineKeyResolverTest() : m_resolver(NULL) {}\r
+\r
+    void setUp() {\r
+        string config = data_path + "InlineKeyResolver.xml";\r
+        ifstream in(config.c_str());\r
+        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);\r
+        XercesJanitor<DOMDocument> janitor(doc);\r
+        m_resolver=XMLToolingConfig::getConfig().KeyResolverManager.newPlugin(INLINE_KEY_RESOLVER,doc->getDocumentElement());\r
+    }\r
+\r
+    void tearDown() {\r
+        delete m_resolver;\r
+        m_resolver=NULL;\r
+    }\r
+\r
+    void testResolver() {\r
+        string path=data_path + "KeyInfo1.xml";\r
+        ifstream fs(path.c_str());\r
+        DOMDocument* doc=XMLToolingConfig::getConfig().getValidatingParser().parse(fs);\r
+        TS_ASSERT(doc!=NULL);\r
+        const XMLObjectBuilder* b = XMLObjectBuilder::getBuilder(doc->getDocumentElement());\r
+        TS_ASSERT(b!=NULL);\r
+        auto_ptr<KeyInfo> kiObject(dynamic_cast<KeyInfo*>(b->buildFromDocument(doc)));\r
+        TS_ASSERT(kiObject.get()!=NULL);\r
+\r
+        auto_ptr<XSECCryptoKey> key(m_resolver->resolveKey(kiObject.get()));\r
+        TSM_ASSERT("Unable to resolve public key.", key.get()!=NULL);\r
+        TSM_ASSERT_EQUALS("Unexpected key type.", key->getKeyType(), XSECCryptoKey::KEY_RSA_PUBLIC);\r
+\r
+        vector<XSECCryptoX509*> certs;\r
+        TSM_ASSERT_EQUALS("Wrong certificate count.", m_resolver->resolveCertificates(kiObject.get(), certs), 1);\r
+    }\r
+};\r