samltest/saml2/core/impl/AuthnQuery20Test.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 class AuthnQuery20Test : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {\r
  25     XMLCh* expectedID; \r
  26     XMLCh* expectedVersion; \r
  27     XMLCh* expectedConsent; \r
  28     XMLCh* expectedDestination; \r
  29     DateTime* expectedIssueInstant; \r
  30     XMLCh* expectedSessionIndex; \r
  31 \r
  32 public:\r
  33     void setUp() {\r
  34         expectedID = XMLString::transcode("abc123");; \r
  35         expectedVersion = XMLString::transcode("2.0"); \r
  36         expectedConsent = XMLString::transcode("urn:string:consent"); \r
  37         expectedDestination = XMLString::transcode("http://idp.example.org/endpoint"); \r
  38         expectedIssueInstant = new DateTime(XMLString::transcode("2006-02-21T16:40:00.000Z"));\r
  39         expectedSessionIndex = XMLString::transcode("session12345"); \r
  40 \r
  41         singleElementFile = data_path + "saml2/core/impl/AuthnQuery.xml";\r
  42         singleElementOptionalAttributesFile = data_path + "saml2/core/impl/AuthnQueryOptionalAttributes.xml";\r
  43         childElementsFile  = data_path + "saml2/core/impl/AuthnQueryChildElements.xml";    \r
  44         SAMLObjectBaseTestCase::setUp();\r
  45     }\r
  46     \r
  47     void tearDown() {\r
  48         XMLString::release(&expectedID);\r
  49         XMLString::release(&expectedVersion);\r
  50         XMLString::release(&expectedConsent);\r
  51         XMLString::release(&expectedDestination);\r
  52         XMLString::release(&expectedSessionIndex);\r
  53         delete expectedIssueInstant;\r
  54         SAMLObjectBaseTestCase::tearDown();\r
  55     }\r
  56 \r
  57     void testSingleElementUnmarshall() {\r
  58         auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));\r
  59         AuthnQuery* query = dynamic_cast<AuthnQuery*>(xo.get());\r
  60         TS_ASSERT(query!=NULL);\r
  61         assertEquals("ID attribute", expectedID, query->getID());\r
  62         assertEquals("Version attribute", expectedVersion, query->getVersion());\r
  63         assertEquals("IssueInstant attribute", expectedIssueInstant->getFormattedString(), query->getIssueInstant()->getFormattedString());\r
  64 \r
  65         TS_ASSERT(query->getIssuer()==NULL);\r
  66         TS_ASSERT(query->getSignature()==NULL);\r
  67         TS_ASSERT(query->getExtensions()==NULL);\r
  68         TS_ASSERT(query->getSubject()==NULL);\r
  69         TS_ASSERT(query->getRequestedAuthnContext()==NULL);\r
  70     }\r
  71 \r
  72     void testSingleElementOptionalAttributesUnmarshall() {\r
  73         auto_ptr<XMLObject> xo(unmarshallElement(singleElementOptionalAttributesFile));\r
  74         AuthnQuery* query = dynamic_cast<AuthnQuery*>(xo.get());\r
  75         TS_ASSERT(query!=NULL);\r
  76         assertEquals("Consent attribute", expectedConsent, query->getConsent());\r
  77         assertEquals("Destination attribute", expectedDestination, query->getDestination());\r
  78         assertEquals("SessionIndex attribute", expectedSessionIndex, query->getSessionIndex());\r
  79 \r
  80         TS_ASSERT(query->getIssuer()==NULL);\r
  81         TS_ASSERT(query->getSignature()==NULL);\r
  82         TS_ASSERT(query->getExtensions()==NULL);\r
  83         TS_ASSERT(query->getSubject()==NULL);\r
  84         TS_ASSERT(query->getRequestedAuthnContext()==NULL);\r
  85     }\r
  86 \r
  87     void testChildElementsUnmarshall() {\r
  88         auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));\r
  89         AuthnQuery* query= dynamic_cast<AuthnQuery*>(xo.get());\r
  90         TS_ASSERT(query!=NULL);\r
  91         TS_ASSERT(query->getIssuer()!=NULL);\r
  92         TS_ASSERT(query->getSignature()==NULL);\r
  93         TS_ASSERT(query->getExtensions()==NULL);\r
  94         TS_ASSERT(query->getSubject()!=NULL);\r
  95         TS_ASSERT(query->getRequestedAuthnContext()!=NULL);\r
  96     }\r
  97 \r
  98     void testSingleElementMarshall() {\r
  99         AuthnQuery* query=AuthnQueryBuilder::buildAuthnQuery();\r
 100         query->setID(expectedID);\r
 101         query->setIssueInstant(expectedIssueInstant);\r
 102         //query->setVersion(expectedVersion);\r
 103         assertEquals(expectedDOM, query);\r
 104     }\r
 105 \r
 106     void testSingleElementOptionalAttributesMarshall() {\r
 107         AuthnQuery* query=AuthnQueryBuilder::buildAuthnQuery();\r
 108         query->setID(expectedID);\r
 109         query->setIssueInstant(expectedIssueInstant);\r
 110         //query->setVersion(expectedVersion);\r
 111         query->setConsent(expectedConsent);\r
 112         query->setDestination(expectedDestination);\r
 113         query->setSessionIndex(expectedSessionIndex);\r
 114         assertEquals(expectedOptionalAttributesDOM, query);\r
 115     }\r
 116 \r
 117     void testChildElementsMarshall() {\r
 118         AuthnQuery* query=AuthnQueryBuilder::buildAuthnQuery();\r
 119         query->setID(expectedID);\r
 120         query->setIssueInstant(expectedIssueInstant);\r
 121         // Do this just so don't have to redeclare the saml namespace prefix on every child element in the control XML file\r
 122         Namespace* ns = new Namespace(opensaml::SAMLConstants::SAML20_NS, opensaml::SAMLConstants::SAML20_PREFIX);\r
 123         query->addNamespace(*ns);\r
 124         query->setIssuer(IssuerBuilder::buildIssuer());\r
 125         query->setSubject(SubjectBuilder::buildSubject());\r
 126         query->setRequestedAuthnContext(RequestedAuthnContextBuilder::buildRequestedAuthnContext());\r
 127         assertEquals(expectedChildElementsDOM, query);\r
 128         delete ns;\r
 129     }\r
 130 \r
 131 };\r