From ffa46ee8ad885a5b09e4abd569cf500bb7b298cb Mon Sep 17 00:00:00 2001
From: Roland Hedberg <roland.hedberg@adm.umu.se>
Date: Thu, 17 Jan 2013 11:45:31 +0100
Subject: [PATCH] Added new test class

---
 src/idp_test/saml2int.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/idp_test/saml2int.py b/src/idp_test/saml2int.py
index 42fd6f8..99b6328 100644
--- a/src/idp_test/saml2int.py
+++ b/src/idp_test/saml2int.py
@@ -90,6 +90,19 @@ class AuthnQuery(Request):
         assertion = resp.assertion[0]
         self.args["subject"] = assertion.subject
 
+class NameIDMappeingRequest(Request):
+    request = "name_id_mapping_request"
+    _args = {"binding": BINDING_HTTP_REDIRECT}
+
+    def __init__(self):
+        Request.__init__(self)
+        self.tests["post"].append(VerifySuccessStatus)
+
+    def setup(self, environ):
+        resp = environ["response"][-1].response
+        assertion = resp.assertion[0]
+        self.args["subject"] = assertion.subject
+
 # -----------------------------------------------------------------------------
 
 OPERATIONS = {