50d5eed4c488a8d55ed0042c7f8a66af659b0b51
[shibboleth/cpp-opensaml.git] / samltest / saml1 / core / impl / AdviceTest.h
1 /*\r
2  *  Copyright 2001-2006 Internet2\r
3  * \r
4  * Licensed under the Apache License, Version 2.0 (the "License");\r
5  * you may not use this file except in compliance with the License.\r
6  * You may obtain a copy of the License at\r
7  *\r
8  *     http://www.apache.org/licenses/LICENSE-2.0\r
9  *\r
10  * Unless required by applicable law or agreed to in writing, software\r
11  * distributed under the License is distributed on an "AS IS" BASIS,\r
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
13  * See the License for the specific language governing permissions and\r
14  * limitations under the License.\r
15  */\r
16 \r
17 #include "internal.h"\r
18 #include <saml/saml1/core/Assertions.h>\r
19 \r
20 using namespace opensaml::saml1;\r
21 \r
22 class AdviceTest : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {\r
23     XMLCh* AssertionID;\r
24     XMLCh* IssueInstant;\r
25 \r
26 public:\r
27     void setUp() {\r
28         AssertionID=XMLString::transcode("_123456789");\r
29         IssueInstant=XMLString::transcode("1971-03-19T13:23:00Z");\r
30         singleElementFile = data_path + "saml1/core/impl/singleAdvice.xml";\r
31         childElementsFile  = data_path + "saml1/core/impl/AdviceWithChildren.xml";    \r
32         SAMLObjectBaseTestCase::setUp();\r
33     }\r
34     \r
35     void tearDown() {\r
36         XMLString::release(&AssertionID);\r
37         XMLString::release(&IssueInstant);\r
38         SAMLObjectBaseTestCase::tearDown();\r
39     }\r
40 \r
41     void testSingleElementUnmarshall() {\r
42         auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));\r
43         Advice* advice = dynamic_cast<Advice*>(xo.get());\r
44         TS_ASSERT(advice!=NULL);\r
45         TSM_ASSERT_EQUALS("Number of child AssertIDReference elements", 0, advice->getAssertionIDReferences().size());\r
46         TSM_ASSERT_EQUALS("Number of child Assertion elements", 0, advice->getAssertions().size());\r
47     }\r
48 \r
49     void testChildElementsUnmarshall() {\r
50         auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));\r
51         Advice* advice = dynamic_cast<Advice*>(xo.get());\r
52         TSM_ASSERT_EQUALS("Number of child AssertIDReference elements", 2, advice->getAssertionIDReferences().size());\r
53         TSM_ASSERT_EQUALS("Number of child Assertion elements", 1, advice->getAssertions().size());\r
54     }\r
55 \r
56     void testSingleElementMarshall() {\r
57         assertEquals(expectedDOM, AdviceBuilder::buildAdvice());\r
58     }\r
59 \r
60     void testChildElementsMarshall() {\r
61         Advice* advice=AdviceBuilder::buildAdvice();\r
62         \r
63         advice->getAssertionIDReferences().push_back(AssertionIDReferenceBuilder::buildAssertionIDReference());\r
64         Assertion* assertion=AssertionBuilder::buildAssertion();\r
65         assertion->setAssertionID(AssertionID);\r
66         assertion->setIssueInstant(IssueInstant);\r
67         advice->getAssertions().push_back(assertion);\r
68         advice->getAssertionIDReferences().push_back(AssertionIDReferenceBuilder::buildAssertionIDReference());\r
69 \r
70         assertEquals(expectedChildElementsDOM, advice);\r
71     }\r
72 \r
73 };\r