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