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;
24 class ArtifactResolve20Test : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {
26 XMLCh* expectedVersion;
27 XMLCh* expectedConsent;
28 XMLCh* expectedDestination;
29 DateTime* expectedIssueInstant;
33 expectedID = XMLString::transcode("abc123");;
34 expectedVersion = XMLString::transcode("2.0");
35 expectedConsent = XMLString::transcode("urn:string:consent");
36 expectedDestination = XMLString::transcode("http://idp.example.org/endpoint");
37 expectedIssueInstant = new DateTime(XMLString::transcode("2006-02-21T16:40:00.000Z"));
38 expectedIssueInstant->parseDateTime();
40 singleElementFile = data_path + "saml2/core/impl/ArtifactResolve.xml";
41 singleElementOptionalAttributesFile = data_path + "saml2/core/impl/ArtifactResolveOptionalAttributes.xml";
42 childElementsFile = data_path + "saml2/core/impl/ArtifactResolveChildElements.xml";
43 SAMLObjectBaseTestCase::setUp();
47 XMLString::release(&expectedID);
48 XMLString::release(&expectedVersion);
49 XMLString::release(&expectedConsent);
50 XMLString::release(&expectedDestination);
51 delete expectedIssueInstant;
52 SAMLObjectBaseTestCase::tearDown();
55 void testSingleElementUnmarshall() {
56 auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));
57 ArtifactResolve* request = dynamic_cast<ArtifactResolve*>(xo.get());
58 TS_ASSERT(request!=NULL);
59 assertEquals("ID attribute", expectedID, request->getID());
60 assertEquals("Version attribute", expectedVersion, request->getVersion());
61 TSM_ASSERT_EQUALS("IssueInstant attribute", expectedIssueInstant->getEpoch(), request->getIssueInstant()->getEpoch());
63 TS_ASSERT(request->getIssuer()==NULL);
64 TS_ASSERT(request->getSignature()==NULL);
65 TS_ASSERT(request->getExtensions()==NULL);
66 TS_ASSERT(request->getArtifact()==NULL);
69 void testSingleElementOptionalAttributesUnmarshall() {
70 auto_ptr<XMLObject> xo(unmarshallElement(singleElementOptionalAttributesFile));
71 ArtifactResolve* request = dynamic_cast<ArtifactResolve*>(xo.get());
72 TS_ASSERT(request!=NULL);
73 assertEquals("Consent attribute", expectedConsent, request->getConsent());
74 assertEquals("Destination attribute", expectedDestination, request->getDestination());
76 TS_ASSERT(request->getIssuer()==NULL);
77 TS_ASSERT(request->getSignature()==NULL);
78 TS_ASSERT(request->getExtensions()==NULL);
79 TS_ASSERT(request->getArtifact()==NULL);
82 void testChildElementsUnmarshall() {
83 auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));
84 ArtifactResolve* request= dynamic_cast<ArtifactResolve*>(xo.get());
85 TS_ASSERT(request!=NULL);
86 TS_ASSERT(request->getIssuer()!=NULL);
87 TS_ASSERT(request->getSignature()==NULL);
88 TS_ASSERT(request->getExtensions()==NULL);
89 TS_ASSERT(request->getArtifact()!=NULL);
92 void testSingleElementMarshall() {
93 ArtifactResolve* request=ArtifactResolveBuilder::buildArtifactResolve();
94 request->setID(expectedID);
95 request->setIssueInstant(expectedIssueInstant);
96 //request->setVersion(expectedVersion);
97 assertEquals(expectedDOM, request);
100 void testSingleElementOptionalAttributesMarshall() {
101 ArtifactResolve* request=ArtifactResolveBuilder::buildArtifactResolve();
102 request->setID(expectedID);
103 request->setIssueInstant(expectedIssueInstant);
104 //request->setVersion(expectedVersion);
105 request->setConsent(expectedConsent);
106 request->setDestination(expectedDestination);
107 assertEquals(expectedOptionalAttributesDOM, request);
110 void testChildElementsMarshall() {
111 ArtifactResolve* request=ArtifactResolveBuilder::buildArtifactResolve();
112 request->setID(expectedID);
113 request->setIssueInstant(expectedIssueInstant);
114 // Do this just so don't have to redeclare the saml namespace prefix on every child element in the control XML file
115 Namespace* ns = new Namespace(samlconstants::SAML20_NS, samlconstants::SAML20_PREFIX);
116 request->addNamespace(*ns);
117 request->setIssuer(IssuerBuilder::buildIssuer());
118 request->setArtifact(ArtifactBuilder::buildArtifact());
119 assertEquals(expectedChildElementsDOM, request);