More unit tests.

[shibboleth/opensaml2.git] / samltest / saml1 / core / impl / AdviceTest.h

diff --git a/samltest/saml1/core/impl/AdviceTest.h b/samltest/saml1/core/impl/AdviceTest.h
new file mode 100644 (file)
index 0000000..50d5eed
--- /dev/null
+++ b/samltest/saml1/core/impl/AdviceTest.h
@@ -0,0 +1,73 @@
+/*\r
+ *  Copyright 2001-2006 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 "internal.h"\r
+#include <saml/saml1/core/Assertions.h>\r
+\r
+using namespace opensaml::saml1;\r
+\r
+class AdviceTest : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {\r
+    XMLCh* AssertionID;\r
+    XMLCh* IssueInstant;\r
+\r
+public:\r
+    void setUp() {\r
+        AssertionID=XMLString::transcode("_123456789");\r
+        IssueInstant=XMLString::transcode("1971-03-19T13:23:00Z");\r
+        singleElementFile = data_path + "saml1/core/impl/singleAdvice.xml";\r
+        childElementsFile  = data_path + "saml1/core/impl/AdviceWithChildren.xml";    \r
+        SAMLObjectBaseTestCase::setUp();\r
+    }\r
+    \r
+    void tearDown() {\r
+        XMLString::release(&AssertionID);\r
+        XMLString::release(&IssueInstant);\r
+        SAMLObjectBaseTestCase::tearDown();\r
+    }\r
+\r
+    void testSingleElementUnmarshall() {\r
+        auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));\r
+        Advice* advice = dynamic_cast<Advice*>(xo.get());\r
+        TS_ASSERT(advice!=NULL);\r
+        TSM_ASSERT_EQUALS("Number of child AssertIDReference elements", 0, advice->getAssertionIDReferences().size());\r
+        TSM_ASSERT_EQUALS("Number of child Assertion elements", 0, advice->getAssertions().size());\r
+    }\r
+\r
+    void testChildElementsUnmarshall() {\r
+        auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));\r
+        Advice* advice = dynamic_cast<Advice*>(xo.get());\r
+        TSM_ASSERT_EQUALS("Number of child AssertIDReference elements", 2, advice->getAssertionIDReferences().size());\r
+        TSM_ASSERT_EQUALS("Number of child Assertion elements", 1, advice->getAssertions().size());\r
+    }\r
+\r
+    void testSingleElementMarshall() {\r
+        assertEquals(expectedDOM, AdviceBuilder::buildAdvice());\r
+    }\r
+\r
+    void testChildElementsMarshall() {\r
+        Advice* advice=AdviceBuilder::buildAdvice();\r
+        \r
+        advice->getAssertionIDReferences().push_back(AssertionIDReferenceBuilder::buildAssertionIDReference());\r
+        Assertion* assertion=AssertionBuilder::buildAssertion();\r
+        assertion->setAssertionID(AssertionID);\r
+        assertion->setIssueInstant(IssueInstant);\r
+        advice->getAssertions().push_back(assertion);\r
+        advice->getAssertionIDReferences().push_back(AssertionIDReferenceBuilder::buildAssertionIDReference());\r
+\r
+        assertEquals(expectedChildElementsDOM, advice);\r
+    }\r
+\r
+};\r