2 * Copyright 2001-2007 Internet2
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
18 #include <saml/saml2/core/Protocols.h>
19 #include <saml/util/SAMLConstants.h>
21 using namespace opensaml::saml2p;
22 using namespace opensaml::saml2;
25 class Response20Test : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {
27 XMLCh* expectedInResponseTo;
28 XMLCh* expectedVersion;
29 XMLCh* expectedConsent;
30 XMLCh* expectedDestination;
31 DateTime* expectedIssueInstant;
33 // Assertion marshaller autogenerates ID, Version and IssueInstant if they are NULL,
34 // so have to agree on something to put in the control XML
35 XMLCh* assertionID1, * assertionID2, * assertionID3;
39 expectedID = XMLString::transcode("def456");
40 expectedInResponseTo = XMLString::transcode("abc123");
41 expectedVersion = XMLString::transcode("2.0");
42 expectedConsent = XMLString::transcode("urn:string:consent");
43 expectedDestination = XMLString::transcode("http://sp.example.org/endpoint");
44 expectedIssueInstant = new DateTime(XMLString::transcode("2006-02-21T16:40:00.000Z"));
45 expectedIssueInstant->parseDateTime();
47 assertionID1 = XMLString::transcode("test1");
48 assertionID2= XMLString::transcode("test2");
49 assertionID3 = XMLString::transcode("test3");
51 singleElementFile = data_path + "saml2/core/impl/Response.xml";
52 singleElementOptionalAttributesFile = data_path + "saml2/core/impl/ResponseOptionalAttributes.xml";
53 childElementsFile = data_path + "saml2/core/impl/ResponseChildElements.xml";
54 SAMLObjectBaseTestCase::setUp();
58 XMLString::release(&expectedID);
59 XMLString::release(&expectedInResponseTo);
60 XMLString::release(&expectedVersion);
61 XMLString::release(&expectedConsent);
62 XMLString::release(&expectedDestination);
63 XMLString::release(&assertionID1);
64 XMLString::release(&assertionID2);
65 XMLString::release(&assertionID3);
66 delete expectedIssueInstant;
67 SAMLObjectBaseTestCase::tearDown();
70 void testSingleElementUnmarshall() {
71 auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));
72 Response* response = dynamic_cast<Response*>(xo.get());
73 TS_ASSERT(response!=NULL);
75 assertEquals("ID attribute", expectedID, response->getID());
76 assertEquals("Version attribute", expectedVersion, response->getVersion());
77 TSM_ASSERT_EQUALS("IssueInstant attribute", expectedIssueInstant->getEpoch(), response->getIssueInstant()->getEpoch());
79 TS_ASSERT(response->getIssuer()==NULL);
80 TS_ASSERT(response->getSignature()==NULL);
81 TS_ASSERT(response->getExtensions()==NULL);
82 TS_ASSERT(response->getStatus()==NULL);
83 TSM_ASSERT_EQUALS("# of Assertion child elements", 0, response->getAssertions().size());
84 TSM_ASSERT_EQUALS("# of EncryptedAssertion child elements", 0, response->getEncryptedAssertions().size());
87 void testSingleElementOptionalAttributesUnmarshall() {
88 auto_ptr<XMLObject> xo(unmarshallElement(singleElementOptionalAttributesFile));
89 Response* response = dynamic_cast<Response*>(xo.get());
90 TS_ASSERT(response!=NULL);
92 assertEquals("Consent attribute", expectedConsent, response->getConsent());
93 assertEquals("Destination attribute", expectedDestination, response->getDestination());
94 assertEquals("InResponseTo attribute", expectedInResponseTo, response->getInResponseTo());
96 TS_ASSERT(response->getIssuer()==NULL);
97 TS_ASSERT(response->getSignature()==NULL);
98 TS_ASSERT(response->getExtensions()==NULL);
99 TS_ASSERT(response->getStatus()==NULL);
100 TSM_ASSERT_EQUALS("# of Assertion child elements", 0, response->getAssertions().size());
101 TSM_ASSERT_EQUALS("# of EncryptedAssertion child elements", 0, response->getEncryptedAssertions().size());
104 void testChildElementsUnmarshall() {
105 auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));
106 Response* response= dynamic_cast<Response*>(xo.get());
107 TS_ASSERT(response!=NULL);
109 TS_ASSERT(response->getIssuer()!=NULL);
110 TS_ASSERT(response->getSignature()!=NULL);
111 TS_ASSERT(response->getExtensions()!=NULL);
112 TS_ASSERT(response->getStatus()!=NULL);
113 TSM_ASSERT_EQUALS("# of Assertion child elements", 3, response->getAssertions().size());
114 TSM_ASSERT_EQUALS("# of EncryptedAssertion child elements", 1, response->getEncryptedAssertions().size());
117 void testSingleElementMarshall() {
118 Response* response = ResponseBuilder::buildResponse();
119 TS_ASSERT(response!=NULL);
121 response->setID(expectedID);
122 response->setIssueInstant(expectedIssueInstant);
123 //response->setVersion(expectedVersion);
124 assertEquals(expectedDOM, response);
127 void testSingleElementOptionalAttributesMarshall() {
128 Response* response = ResponseBuilder::buildResponse();
129 TS_ASSERT(response!=NULL);
131 response->setID(expectedID);
132 response->setInResponseTo(expectedInResponseTo);
133 response->setIssueInstant(expectedIssueInstant);
134 //response->setVersion(expectedVersion);
135 response->setConsent(expectedConsent);
136 response->setDestination(expectedDestination);
137 response->setInResponseTo(expectedInResponseTo);
138 assertEquals(expectedOptionalAttributesDOM, response);
141 void testChildElementsMarshall() {
142 Response* response = ResponseBuilder::buildResponse();
143 TS_ASSERT(response!=NULL);
145 response->setID(expectedID);
146 response->setIssueInstant(expectedIssueInstant);
147 // Do this just so don't have to redeclare the saml namespace prefix on every child element in the control XML file
148 Namespace* ns = new Namespace(samlconstants::SAML20_NS, samlconstants::SAML20_PREFIX);
149 response->addNamespace(*ns);
150 response->setIssuer(IssuerBuilder::buildIssuer());
151 // If the form of the default, basic, empty signature that is emittted changes wrt whitespace, etc,
152 // this will probably break the test. In that case need to fix the control XML.
153 response->setSignature(xmlsignature::SignatureBuilder::buildSignature());
154 response->setExtensions(ExtensionsBuilder::buildExtensions());
155 response->setStatus(StatusBuilder::buildStatus());
157 Assertion* assertion=NULL;
159 assertion = AssertionBuilder::buildAssertion();
160 assertion->setIssueInstant(expectedIssueInstant);
161 assertion->setID(assertionID1);
162 response->getAssertions().push_back(assertion);
164 assertion = AssertionBuilder::buildAssertion();
165 assertion->setIssueInstant(expectedIssueInstant);
166 assertion->setID(assertionID2);
167 response->getAssertions().push_back(assertion);
169 response->getEncryptedAssertions().push_back((EncryptedAssertionBuilder::buildEncryptedAssertion()));
171 assertion = AssertionBuilder::buildAssertion();
172 assertion->setIssueInstant(expectedIssueInstant);
173 assertion->setID(assertionID3);
174 response->getAssertions().push_back(assertion);
177 assertEquals(expectedChildElementsDOM, response);