def test_create_light_request_without_extensions(self): root = Element(Q_NAMES['saml2p:AuthnRequest'], nsmap=EIDAS_NAMESPACES) saml_request = SAMLRequest(ElementTree(root), 'CZ', 'relay123') expected = LightRequest(citizen_country_code='CZ', relay_state='relay123', requested_attributes=OrderedDict()) self.assertEqual(saml_request.create_light_request(), expected)
def test_create_light_request_success(self): self.maxDiff = None with cast(TextIO, (DATA_DIR / 'saml_request.xml').open('r')) as f: data = f.read() saml_request = SAMLRequest(parse_xml(data), 'CA', 'relay123') self.assertEqual( saml_request.create_light_request().get_data_as_dict(), LIGHT_REQUEST_DICT)
def test_create_light_request_extra_elements(self): self.maxDiff = None with cast(TextIO, (DATA_DIR / 'saml_request.xml').open('r')) as f: document = parse_xml(f.read()) SubElement(document.getroot(), 'extra').text = 'extra' SubElement( document.find(".//{}".format( Q_NAMES['eidas:RequestedAttributes'])), 'extra').text = 'extra' saml_request = SAMLRequest(document, 'CA', 'relay123') self.assertEqual( saml_request.create_light_request().get_data_as_dict(), LIGHT_REQUEST_DICT)