samltest/saml2/core/impl/AttributeStatement20Test.h

   1 /**
   2  * Licensed to the University Corporation for Advanced Internet
   3  * Development, Inc. (UCAID) under one or more contributor license
   4  * agreements. See the NOTICE file distributed with this work for
   5  * additional information regarding copyright ownership.
   6  *
   7  * UCAID licenses this file to you under the Apache License,
   8  * Version 2.0 (the "License"); you may not use this file except
   9  * in compliance with the License. You may obtain a copy of the
  10  * License at
  11  *
  12  * http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing,
  15  * software distributed under the License is distributed on an
  16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
  17  * either express or implied. See the License for the specific
  18  * language governing permissions and limitations under the License.
  19  */
  20
  21 #include "internal.h"
  22 #include <saml/saml2/core/Assertions.h>
  23 #include <saml/util/SAMLConstants.h>
  24
  25 using namespace opensaml::saml2;
  26
  27 class AttributeStatement20Test : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {
  28
  29 public:
  30     void setUp() {
  31         singleElementFile = data_path + "saml2/core/impl/AttributeStatement.xml";
  32         childElementsFile  = data_path + "saml2/core/impl/AttributeStatementChildElements.xml";
  33         SAMLObjectBaseTestCase::setUp();
  34     }
  35
  36     void tearDown() {
  37         SAMLObjectBaseTestCase::tearDown();
  38     }
  39
  40     void testSingleElementUnmarshall() {
  41         auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));
  42         AttributeStatement* statement = dynamic_cast<AttributeStatement*>(xo.get());
  43         TS_ASSERT(statement!=nullptr);
  44
  45         TSM_ASSERT_EQUALS("# of Attribute child elements", 0, statement->getAttributes().size());
  46         TSM_ASSERT_EQUALS("# of EncryptedAttribute child elements", 0, statement->getEncryptedAttributes().size());
  47     }
  48
  49     void testChildElementsUnmarshall() {
  50         auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));
  51         AttributeStatement* statement= dynamic_cast<AttributeStatement*>(xo.get());
  52         TS_ASSERT(statement!=nullptr);
  53
  54         TSM_ASSERT_EQUALS("# of Attribute child elements", 3, statement->getAttributes().size());
  55         TSM_ASSERT_EQUALS("# of EncryptedAttribute child elements", 1, statement->getEncryptedAttributes().size());
  56     }
  57
  58     void testSingleElementMarshall() {
  59         AttributeStatement* statement=AttributeStatementBuilder::buildAttributeStatement();
  60         assertEquals(expectedDOM, statement);
  61     }
  62
  63     void testChildElementsMarshall() {
  64         AttributeStatement* statement=AttributeStatementBuilder::buildAttributeStatement();
  65
  66         statement->getAttributes().push_back(AttributeBuilder::buildAttribute());
  67         statement->getAttributes().push_back(AttributeBuilder::buildAttribute());
  68         statement->getEncryptedAttributes().push_back(EncryptedAttributeBuilder::buildEncryptedAttribute());
  69         statement->getAttributes().push_back(AttributeBuilder::buildAttribute());
  70         assertEquals(expectedChildElementsDOM, statement);
  71     }
  72
  73 };