Amend test case adjustments

Signed-off-by: Ivan Kanakarakis <[email protected]>

Author: c00kiemon5ter

Diff for: tests/fakeIDP.py

@@ -43,8 +43,8 @@ def unpack_form(_str, ver="SAMLRequest"):
 
 
 class DummyResponse(object):
-    def __init__(self, code, data, headers=None):
-        self.status_code = code
+    def __init__(self, status, data, headers=None):
+        self.status_code = status
         self.text = data
         self.headers = headers or []
         self.content = data
@@ -130,8 +130,7 @@ def authn_request_endpoint(self, req, binding, relay_state):
         _dict = pack.factory(_binding, response,
                              resp_args["destination"], relay_state,
                              "SAMLResponse")
-        _dict.pop('status')
-        return DummyResponse(200, **_dict)
+        return DummyResponse(**_dict)
 
     def attribute_query_endpoint(self, xml_str, binding):
         if binding == BINDING_SOAP:
@@ -161,7 +160,7 @@ def attribute_query_endpoint(self, xml_str, binding):
         else:  # Just POST
             response = "%s" % attr_resp
 
-        return DummyResponse(200, response)
+        return DummyResponse(status=200, data=response)
 
     def logout_endpoint(self, xml_str, binding):
         if binding == BINDING_SOAP:
@@ -186,4 +185,4 @@ def logout_endpoint(self, xml_str, binding):
         else: # Just POST
             response = "%s" % _resp
 
-        return DummyResponse(200, response)
+        return DummyResponse(status=200, data=response)

Diff for: tests/test_51_client.py

@@ -3249,7 +3249,6 @@ def test_post_sso(self):
         # Here I fake what the client will do
         # create the form post
 
-        http_args.pop('status')
         http_args["data"] = parse.urlencode(_dic)
         http_args["method"] = "POST"
         http_args["dummy"] = _dic["SAMLRequest"]
@@ -3286,7 +3285,6 @@ def test_negotiated_post_sso(self):
         # Here I fake what the client will do
         # create the form post
 
-        http_args.pop('status')
         http_args["data"] = parse.urlencode(_dic)
         http_args["method"] = "POST"
         http_args["dummy"] = _dic["SAMLRequest"]