diff --git a/tests/test_client.py b/tests/test_client.py index d5ed85d..6d04bae 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -88,7 +88,7 @@ def test_parse_2(): # assert False REQ1 = """ -http://vo.example.com/sp1E8042FB4-4D5B-48C3-8E14-8EDD852790DD""" +http://vo.example.com/sp1E8042FB4-4D5B-48C3-8E14-8EDD852790DD""" class TestClient: def setup_class(self): @@ -107,8 +107,10 @@ class TestClient: assert req.version == "2.0" subject = req.subject name_id = subject.name_id - assert name_id.format == saml.NAMEID_FORMAT_PERSISTENT + assert name_id.name_format == saml.NAMEID_FORMAT_PERSISTENT assert name_id.text == "E8042FB4-4D5B-48C3-8E14-8EDD852790DD" + issuer = req.issuer + assert issuer.text == "http://vo.example.com/sp1" def test_create_attribute_query2(self): req = self.client.create_attribute_request("1", @@ -130,7 +132,7 @@ class TestClient: assert req.version == "2.0" subject = req.subject name_id = subject.name_id - assert name_id.format == saml.NAMEID_FORMAT_PERSISTENT + assert name_id.name_format == saml.NAMEID_FORMAT_PERSISTENT assert name_id.text == "E8042FB4-4D5B-48C3-8E14-8EDD852790DD" assert len(req.attribute) == 3 # one is givenName @@ -149,4 +151,21 @@ class TestClient: if getattr(attribute,"friendly_name"): assert False seen.append("email") - assert set(seen) == set(["givenName","surname","email"]) \ No newline at end of file + assert set(seen) == set(["givenName","surname","email"]) + + def test_create_attribute_query_3(self): + req = self.client.create_attribute_request("1", + "_e7b68a04488f715cda642fbdd90099f5", + "urn:mace:umu.se:saml/rolandsp", + "https://aai-demo-idp.switch.ch/idp/shibboleth", + format=saml.NAMEID_FORMAT_TRANSIENT ) + + assert isinstance(req, samlp.AttributeQuery) + assert req.destination == "https://aai-demo-idp.switch.ch/idp/shibboleth" + assert req.id == "1" + assert req.version == "2.0" + assert req.issue_instant + assert req.issuer.text == "urn:mace:umu.se:saml/rolandsp" + nameid = req.subject.name_id + assert nameid.name_format == saml.NAMEID_FORMAT_TRANSIENT + assert nameid.text == "_e7b68a04488f715cda642fbdd90099f5"