2 * Licensed to the University Corporation for Advanced Internet
3 * Development, Inc. (UCAID) under one or more contributor license
4 * agreements. See the NOTICE file distributed with this work for
5 * additional information regarding copyright ownership.
7 * UCAID licenses this file to you under the Apache License,
8 * Version 2.0 (the "License"); you may not use this file except
9 * in compliance with the License. You may obtain a copy of the
12 * http://www.apache.org/licenses/LICENSE-2.0
14 * Unless required by applicable law or agreed to in writing,
15 * software distributed under the License is distributed on an
16 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
17 * either express or implied. See the License for the specific
18 * language governing permissions and limitations under the License.
22 #include <saml/saml2/core/Protocols.h>
23 #include <saml/util/SAMLConstants.h>
25 using namespace opensaml::saml2p;
26 using namespace opensaml::saml2;
28 class AssertionIDRequest20Test : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {
30 XMLCh* expectedVersion;
31 XMLCh* expectedConsent;
32 XMLCh* expectedDestination;
33 DateTime* expectedIssueInstant;
37 expectedID = XMLString::transcode("abc123");;
38 expectedVersion = XMLString::transcode("2.0");
39 expectedConsent = XMLString::transcode("urn:string:consent");
40 expectedDestination = XMLString::transcode("http://idp.example.org/endpoint");
41 expectedIssueInstant = new DateTime(XMLString::transcode("2006-02-21T16:40:00.000Z"));
42 expectedIssueInstant->parseDateTime();
44 singleElementFile = data_path + "saml2/core/impl/AssertionIDRequest.xml";
45 singleElementOptionalAttributesFile = data_path + "saml2/core/impl/AssertionIDRequestOptionalAttributes.xml";
46 childElementsFile = data_path + "saml2/core/impl/AssertionIDRequestChildElements.xml";
47 SAMLObjectBaseTestCase::setUp();
51 XMLString::release(&expectedID);
52 XMLString::release(&expectedVersion);
53 XMLString::release(&expectedConsent);
54 XMLString::release(&expectedDestination);
55 delete expectedIssueInstant;
56 SAMLObjectBaseTestCase::tearDown();
59 void testSingleElementUnmarshall() {
60 auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));
61 AssertionIDRequest* request = dynamic_cast<AssertionIDRequest*>(xo.get());
62 TS_ASSERT(request!=nullptr);
63 assertEquals("ID attribute", expectedID, request->getID());
64 assertEquals("Version attribute", expectedVersion, request->getVersion());
65 TSM_ASSERT_EQUALS("IssueInstant attribute", expectedIssueInstant->getEpoch(), request->getIssueInstant()->getEpoch());
67 TS_ASSERT(request->getIssuer()==nullptr);
68 TS_ASSERT(request->getSignature()==nullptr);
69 TS_ASSERT(request->getExtensions()==nullptr);
70 TSM_ASSERT_EQUALS("# of AssertionIDRef child elements", 0, request->getAssertionIDRefs().size());
73 void testSingleElementOptionalAttributesUnmarshall() {
74 auto_ptr<XMLObject> xo(unmarshallElement(singleElementOptionalAttributesFile));
75 AssertionIDRequest* request = dynamic_cast<AssertionIDRequest*>(xo.get());
76 TS_ASSERT(request!=nullptr);
77 assertEquals("Consent attribute", expectedConsent, request->getConsent());
78 assertEquals("Destination attribute", expectedDestination, request->getDestination());
80 TS_ASSERT(request->getIssuer()==nullptr);
81 TS_ASSERT(request->getSignature()==nullptr);
82 TS_ASSERT(request->getExtensions()==nullptr);
83 TSM_ASSERT_EQUALS("# of AssertionIDRef child elements", 0, request->getAssertionIDRefs().size());
86 void testChildElementsUnmarshall() {
87 auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));
88 AssertionIDRequest* request= dynamic_cast<AssertionIDRequest*>(xo.get());
89 TS_ASSERT(request!=nullptr);
90 TS_ASSERT(request->getIssuer()!=nullptr);
91 TS_ASSERT(request->getSignature()==nullptr);
92 TS_ASSERT(request->getExtensions()==nullptr);
93 TSM_ASSERT_EQUALS("# of AssertionIDRef child elements", 3, request->getAssertionIDRefs().size());
96 void testSingleElementMarshall() {
97 AssertionIDRequest* request=AssertionIDRequestBuilder::buildAssertionIDRequest();
98 request->setID(expectedID);
99 request->setIssueInstant(expectedIssueInstant);
100 //request->setVersion(expectedVersion);
101 assertEquals(expectedDOM, request);
104 void testSingleElementOptionalAttributesMarshall() {
105 AssertionIDRequest* request=AssertionIDRequestBuilder::buildAssertionIDRequest();
106 request->setID(expectedID);
107 request->setIssueInstant(expectedIssueInstant);
108 //request->setVersion(expectedVersion);
109 request->setConsent(expectedConsent);
110 request->setDestination(expectedDestination);
111 assertEquals(expectedOptionalAttributesDOM, request);
114 void testChildElementsMarshall() {
115 AssertionIDRequest* request=AssertionIDRequestBuilder::buildAssertionIDRequest();
116 request->setID(expectedID);
117 request->setIssueInstant(expectedIssueInstant);
118 // Do this just so don't have to redeclare the saml namespace prefix on every child element in the control XML file
119 Namespace* ns = new Namespace(samlconstants::SAML20_NS, samlconstants::SAML20_PREFIX);
120 request->addNamespace(*ns);
121 request->setIssuer(IssuerBuilder::buildIssuer());
122 request->getAssertionIDRefs().push_back(AssertionIDRefBuilder::buildAssertionIDRef());
123 request->getAssertionIDRefs().push_back(AssertionIDRefBuilder::buildAssertionIDRef());
124 request->getAssertionIDRefs().push_back(AssertionIDRefBuilder::buildAssertionIDRef());
125 assertEquals(expectedChildElementsDOM, request);