samltest/saml2/core/impl/Response20Test.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/saml2/core/Protocols.h>\r
  19 #include <saml/util/SAMLConstants.h>\r
  20 \r
  21 using namespace opensaml::saml2p;\r
  22 using namespace opensaml::saml2;\r
  23 \r
  24 \r
  25 class Response20Test : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {\r
  26     XMLCh* expectedID; \r
  27     XMLCh* expectedInResponseTo; \r
  28     XMLCh* expectedVersion; \r
  29     XMLCh* expectedConsent; \r
  30     XMLCh* expectedDestination; \r
  31     DateTime* expectedIssueInstant; \r
  32 \r
  33     // Assertion marshaller autogenerates ID, Version and IssueInstant if they are NULL,\r
  34     // so have to agree on something to put in the control XML\r
  35     XMLCh* assertionID1, * assertionID2, * assertionID3;\r
  36 \r
  37 public:\r
  38     void setUp() {\r
  39         expectedID = XMLString::transcode("def456"); \r
  40         expectedInResponseTo = XMLString::transcode("abc123"); \r
  41         expectedVersion = XMLString::transcode("2.0"); \r
  42         expectedConsent = XMLString::transcode("urn:string:consent"); \r
  43         expectedDestination = XMLString::transcode("http://sp.example.org/endpoint"); \r
  44         expectedIssueInstant = new DateTime(XMLString::transcode("2006-02-21T16:40:00.000Z"));\r
  45         expectedIssueInstant->parseDateTime();\r
  46 \r
  47         assertionID1 = XMLString::transcode("test1"); \r
  48         assertionID2= XMLString::transcode("test2"); \r
  49         assertionID3 = XMLString::transcode("test3"); \r
  50 \r
  51         singleElementFile = data_path + "saml2/core/impl/Response.xml";\r
  52         singleElementOptionalAttributesFile = data_path + "saml2/core/impl/ResponseOptionalAttributes.xml";\r
  53         childElementsFile  = data_path + "saml2/core/impl/ResponseChildElements.xml";    \r
  54         SAMLObjectBaseTestCase::setUp();\r
  55     }\r
  56     \r
  57     void tearDown() {\r
  58         XMLString::release(&expectedID);\r
  59         XMLString::release(&expectedInResponseTo);\r
  60         XMLString::release(&expectedVersion);\r
  61         XMLString::release(&expectedConsent);\r
  62         XMLString::release(&expectedDestination);\r
  63         XMLString::release(&assertionID1);\r
  64         XMLString::release(&assertionID2);\r
  65         XMLString::release(&assertionID3);\r
  66         delete expectedIssueInstant;\r
  67         SAMLObjectBaseTestCase::tearDown();\r
  68     }\r
  69 \r
  70     void testSingleElementUnmarshall() {\r
  71         auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));\r
  72         Response* response = dynamic_cast<Response*>(xo.get());\r
  73         TS_ASSERT(response!=NULL);\r
  74 \r
  75         assertEquals("ID attribute", expectedID, response->getID());\r
  76         assertEquals("Version attribute", expectedVersion, response->getVersion());\r
  77         TSM_ASSERT_EQUALS("IssueInstant attribute", expectedIssueInstant->getEpoch(), response->getIssueInstant()->getEpoch());\r
  78 \r
  79         TS_ASSERT(response->getIssuer()==NULL);\r
  80         TS_ASSERT(response->getSignature()==NULL);\r
  81         TS_ASSERT(response->getExtensions()==NULL);\r
  82         TS_ASSERT(response->getStatus()==NULL);\r
  83         TSM_ASSERT_EQUALS("# of Assertion child elements", 0, response->getAssertions().size());\r
  84         TSM_ASSERT_EQUALS("# of EncryptedAssertion child elements", 0, response->getEncryptedAssertions().size());\r
  85     }\r
  86 \r
  87     void testSingleElementOptionalAttributesUnmarshall() {\r
  88         auto_ptr<XMLObject> xo(unmarshallElement(singleElementOptionalAttributesFile));\r
  89         Response* response = dynamic_cast<Response*>(xo.get());\r
  90         TS_ASSERT(response!=NULL);\r
  91 \r
  92         assertEquals("Consent attribute", expectedConsent, response->getConsent());\r
  93         assertEquals("Destination attribute", expectedDestination, response->getDestination());\r
  94         assertEquals("InResponseTo attribute", expectedInResponseTo, response->getInResponseTo());\r
  95 \r
  96         TS_ASSERT(response->getIssuer()==NULL);\r
  97         TS_ASSERT(response->getSignature()==NULL);\r
  98         TS_ASSERT(response->getExtensions()==NULL);\r
  99         TS_ASSERT(response->getStatus()==NULL);\r
 100         TSM_ASSERT_EQUALS("# of Assertion child elements", 0, response->getAssertions().size());\r
 101         TSM_ASSERT_EQUALS("# of EncryptedAssertion child elements", 0, response->getEncryptedAssertions().size());\r
 102     }\r
 103 \r
 104     void testChildElementsUnmarshall() {\r
 105         auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));\r
 106         Response* response= dynamic_cast<Response*>(xo.get());\r
 107         TS_ASSERT(response!=NULL);\r
 108 \r
 109         TS_ASSERT(response->getIssuer()!=NULL);\r
 110         TS_ASSERT(response->getSignature()!=NULL);\r
 111         TS_ASSERT(response->getExtensions()!=NULL);\r
 112         TS_ASSERT(response->getStatus()!=NULL);\r
 113         TSM_ASSERT_EQUALS("# of Assertion child elements", 3, response->getAssertions().size());\r
 114         TSM_ASSERT_EQUALS("# of EncryptedAssertion child elements", 1, response->getEncryptedAssertions().size());\r
 115     }\r
 116 \r
 117     void testSingleElementMarshall() {\r
 118         Response* response = ResponseBuilder::buildResponse();\r
 119         TS_ASSERT(response!=NULL);\r
 120 \r
 121         response->setID(expectedID);\r
 122         response->setIssueInstant(expectedIssueInstant);\r
 123         //response->setVersion(expectedVersion);\r
 124         assertEquals(expectedDOM, response);\r
 125     }\r
 126 \r
 127     void testSingleElementOptionalAttributesMarshall() {\r
 128         Response* response = ResponseBuilder::buildResponse();\r
 129         TS_ASSERT(response!=NULL);\r
 130 \r
 131         response->setID(expectedID);\r
 132         response->setInResponseTo(expectedInResponseTo);\r
 133         response->setIssueInstant(expectedIssueInstant);\r
 134         //response->setVersion(expectedVersion);\r
 135         response->setConsent(expectedConsent);\r
 136         response->setDestination(expectedDestination);\r
 137         response->setInResponseTo(expectedInResponseTo);\r
 138         assertEquals(expectedOptionalAttributesDOM, response);\r
 139     }\r
 140 \r
 141     void testChildElementsMarshall() {\r
 142         Response* response = ResponseBuilder::buildResponse();\r
 143         TS_ASSERT(response!=NULL);\r
 144 \r
 145         response->setID(expectedID);\r
 146         response->setIssueInstant(expectedIssueInstant);\r
 147         // Do this just so don't have to redeclare the saml namespace prefix on every child element in the control XML file\r
 148         Namespace* ns = new Namespace(opensaml::SAMLConstants::SAML20_NS, opensaml::SAMLConstants::SAML20_PREFIX);\r
 149         response->addNamespace(*ns);\r
 150         response->setIssuer(IssuerBuilder::buildIssuer());\r
 151         // If the form of the default, basic, empty signature that is emittted changes wrt whitespace, etc,\r
 152         // this will probably break the test.  In that case need to fix the control XML.\r
 153         response->setSignature(xmlsignature::SignatureBuilder::buildSignature());\r
 154         response->setExtensions(ExtensionsBuilder::buildExtensions());\r
 155         response->setStatus(StatusBuilder::buildStatus());\r
 156 \r
 157         Assertion* assertion=NULL;\r
 158 \r
 159         assertion = AssertionBuilder::buildAssertion();\r
 160         assertion->setIssueInstant(expectedIssueInstant);\r
 161         assertion->setID(assertionID1);\r
 162         response->getAssertions().push_back(assertion);\r
 163 \r
 164         assertion = AssertionBuilder::buildAssertion();\r
 165         assertion->setIssueInstant(expectedIssueInstant);\r
 166         assertion->setID(assertionID2);\r
 167         response->getAssertions().push_back(assertion);\r
 168 \r
 169         response->getEncryptedAssertions().push_back((EncryptedAssertionBuilder::buildEncryptedAssertion()));\r
 170 \r
 171         assertion = AssertionBuilder::buildAssertion();\r
 172         assertion->setIssueInstant(expectedIssueInstant);\r
 173         assertion->setID(assertionID3);\r
 174         response->getAssertions().push_back(assertion);\r
 175 \r
 176 \r
 177         assertEquals(expectedChildElementsDOM, response);\r
 178         delete ns;\r
 179     }\r
 180 \r
 181 };\r