samltest/saml2/core/impl/AssertionIDRequest20Test.h

   1 /*
   2  *  Copyright 2001-2006 Internet2
   3  *
   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
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  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.
  15  */
  16
  17 #include "internal.h"
  18 #include <saml/saml2/core/Protocols.h>
  19 #include <saml/util/SAMLConstants.h>
  20
  21 using namespace opensaml::saml2p;
  22 using namespace opensaml::saml2;
  23
  24 class AssertionIDRequest20Test : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {
  25     XMLCh* expectedID;
  26     XMLCh* expectedVersion;
  27     XMLCh* expectedConsent;
  28     XMLCh* expectedDestination;
  29     DateTime* expectedIssueInstant;
  30
  31 public:
  32     void setUp() {
  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();
  39
  40         singleElementFile = data_path + "saml2/core/impl/AssertionIDRequest.xml";
  41         singleElementOptionalAttributesFile = data_path + "saml2/core/impl/AssertionIDRequestOptionalAttributes.xml";
  42         childElementsFile  = data_path + "saml2/core/impl/AssertionIDRequestChildElements.xml";
  43         SAMLObjectBaseTestCase::setUp();
  44     }
  45
  46     void tearDown() {
  47         XMLString::release(&expectedID);
  48         XMLString::release(&expectedVersion);
  49         XMLString::release(&expectedConsent);
  50         XMLString::release(&expectedDestination);
  51         delete expectedIssueInstant;
  52         SAMLObjectBaseTestCase::tearDown();
  53     }
  54
  55     void testSingleElementUnmarshall() {
  56         auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));
  57         AssertionIDRequest* request = dynamic_cast<AssertionIDRequest*>(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());
  62
  63         TS_ASSERT(request->getIssuer()==NULL);
  64         TS_ASSERT(request->getSignature()==NULL);
  65         TS_ASSERT(request->getExtensions()==NULL);
  66         TSM_ASSERT_EQUALS("# of AssertionIDRef child elements", 0, request->getAssertionIDRefs().size());
  67     }
  68
  69     void testSingleElementOptionalAttributesUnmarshall() {
  70         auto_ptr<XMLObject> xo(unmarshallElement(singleElementOptionalAttributesFile));
  71         AssertionIDRequest* request = dynamic_cast<AssertionIDRequest*>(xo.get());
  72         TS_ASSERT(request!=NULL);
  73         assertEquals("Consent attribute", expectedConsent, request->getConsent());
  74         assertEquals("Destination attribute", expectedDestination, request->getDestination());
  75
  76         TS_ASSERT(request->getIssuer()==NULL);
  77         TS_ASSERT(request->getSignature()==NULL);
  78         TS_ASSERT(request->getExtensions()==NULL);
  79         TSM_ASSERT_EQUALS("# of AssertionIDRef child elements", 0, request->getAssertionIDRefs().size());
  80     }
  81
  82     void testChildElementsUnmarshall() {
  83         auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));
  84         AssertionIDRequest* request= dynamic_cast<AssertionIDRequest*>(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         TSM_ASSERT_EQUALS("# of AssertionIDRef child elements", 3, request->getAssertionIDRefs().size());
  90     }
  91
  92     void testSingleElementMarshall() {
  93         AssertionIDRequest* request=AssertionIDRequestBuilder::buildAssertionIDRequest();
  94         request->setID(expectedID);
  95         request->setIssueInstant(expectedIssueInstant);
  96         //request->setVersion(expectedVersion);
  97         assertEquals(expectedDOM, request);
  98     }
  99
 100     void testSingleElementOptionalAttributesMarshall() {
 101         AssertionIDRequest* request=AssertionIDRequestBuilder::buildAssertionIDRequest();
 102         request->setID(expectedID);
 103         request->setIssueInstant(expectedIssueInstant);
 104         //request->setVersion(expectedVersion);
 105         request->setConsent(expectedConsent);
 106         request->setDestination(expectedDestination);
 107         assertEquals(expectedOptionalAttributesDOM, request);
 108     }
 109
 110     void testChildElementsMarshall() {
 111         AssertionIDRequest* request=AssertionIDRequestBuilder::buildAssertionIDRequest();
 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(opensaml::SAMLConstants::SAML20_NS, opensaml::SAMLConstants::SAML20_PREFIX);
 116         request->addNamespace(*ns);
 117         request->setIssuer(IssuerBuilder::buildIssuer());
 118         request->getAssertionIDRefs().push_back(AssertionIDRefBuilder::buildAssertionIDRef());
 119         request->getAssertionIDRefs().push_back(AssertionIDRefBuilder::buildAssertionIDRef());
 120         request->getAssertionIDRefs().push_back(AssertionIDRefBuilder::buildAssertionIDRef());
 121         assertEquals(expectedChildElementsDOM, request);
 122         delete ns;
 123     }
 124
 125 };