samltest/ArtifactMapTest.h

   1 /*\r
   2  *  Copyright 2001-2005 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/SAMLConfig.h>\r
  19 #include <saml/saml2/binding/SAML2ArtifactType0004.h>\r
  20 #include <saml/saml2/core/Protocols.h>\r
  21 \r
  22 using namespace opensaml::saml2p;\r
  23 using namespace opensaml;\r
  24 using namespace std;\r
  25 \r
  26 class ArtifactMapTest : public CxxTest::TestSuite\r
  27 {\r
  28 public:\r
  29     string providerIdStr;\r
  30     string handle;\r
  31     ArtifactMap* artifactMap;\r
  32     void setUp() {\r
  33         if (handle.empty()) {\r
  34             providerIdStr = "https://idp.org/SAML";\r
  35             SAMLConfig::getConfig().generateRandomBytes(handle,SAML2ArtifactType0004::HANDLE_LENGTH);\r
  36         }\r
  37         artifactMap = new ArtifactMap();\r
  38     }\r
  39     void tearDown() {\r
  40         delete artifactMap;\r
  41         artifactMap=NULL;\r
  42     }\r
  43     void testArtifactMap(void) {\r
  44         auto_ptr<Response> response(ResponseBuilder::buildResponse());\r
  45 \r
  46         SAML2ArtifactType0004 artifact(SAMLConfig::getConfig().hashSHA1(providerIdStr.c_str()),666,handle);\r
  47 \r
  48         artifactMap->storeContent(response.get(), &artifact, providerIdStr.c_str());\r
  49         response.release();\r
  50 \r
  51         auto_ptr<XMLObject> xmlObject(artifactMap->retrieveContent(&artifact, providerIdStr.c_str()));\r
  52         TSM_ASSERT_THROWS("Artifact resolution improperly succeeded.", artifactMap->retrieveContent(&artifact), BindingException);\r
  53         TSM_ASSERT("Mapped content was not a Response.", dynamic_cast<Response*>(xmlObject.get())!=NULL);\r
  54     }\r
  55 };\r