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)