samltest/saml2/core/impl/AttributeStatement20Test.h

   1 /*
   2  *  Copyright 2001-2010 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/Assertions.h>
  19 #include <saml/util/SAMLConstants.h>
  20
  21 using namespace opensaml::saml2;
  22
  23 class AttributeStatement20Test : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {
  24
  25 public:
  26     void setUp() {
  27         singleElementFile = data_path + "saml2/core/impl/AttributeStatement.xml";
  28         childElementsFile  = data_path + "saml2/core/impl/AttributeStatementChildElements.xml";
  29         SAMLObjectBaseTestCase::setUp();
  30     }
  31
  32     void tearDown() {
  33         SAMLObjectBaseTestCase::tearDown();
  34     }
  35
  36     void testSingleElementUnmarshall() {
  37         auto_ptr<XMLObject> xo(unmarshallElement(singleElementFile));
  38         AttributeStatement* statement = dynamic_cast<AttributeStatement*>(xo.get());
  39         TS_ASSERT(statement!=nullptr);
  40
  41         TSM_ASSERT_EQUALS("# of Attribute child elements", 0, statement->getAttributes().size());
  42         TSM_ASSERT_EQUALS("# of EncryptedAttribute child elements", 0, statement->getEncryptedAttributes().size());
  43     }
  44
  45     void testChildElementsUnmarshall() {
  46         auto_ptr<XMLObject> xo(unmarshallElement(childElementsFile));
  47         AttributeStatement* statement= dynamic_cast<AttributeStatement*>(xo.get());
  48         TS_ASSERT(statement!=nullptr);
  49
  50         TSM_ASSERT_EQUALS("# of Attribute child elements", 3, statement->getAttributes().size());
  51         TSM_ASSERT_EQUALS("# of EncryptedAttribute child elements", 1, statement->getEncryptedAttributes().size());
  52     }
  53
  54     void testSingleElementMarshall() {
  55         AttributeStatement* statement=AttributeStatementBuilder::buildAttributeStatement();
  56         assertEquals(expectedDOM, statement);
  57     }
  58
  59     void testChildElementsMarshall() {
  60         AttributeStatement* statement=AttributeStatementBuilder::buildAttributeStatement();
  61
  62         statement->getAttributes().push_back(AttributeBuilder::buildAttribute());
  63         statement->getAttributes().push_back(AttributeBuilder::buildAttribute());
  64         statement->getEncryptedAttributes().push_back(EncryptedAttributeBuilder::buildEncryptedAttribute());
  65         statement->getAttributes().push_back(AttributeBuilder::buildAttribute());
  66         assertEquals(expectedChildElementsDOM, statement);
  67     }
  68
  69 };