Files
deb-python-pysaml2/tests/test_metadata.py
Roland Hedberg af3c668e85 More tests
2009-10-14 05:48:29 +02:00

35 lines
1.3 KiB
Python

from saml2 import metadata
SWAMI_METADATA = "tests/urn-mace-swami.se-swamid-test-1.0-metadata.xml"
INCOMMON_METADATA = "tests/InCommon-metadata.xml"
def test_swami_1():
md = metadata.MetaData()
md.import_metadata(open(SWAMI_METADATA).read())
print len(md)
assert len(md) == 1
print md.keys()
assert md.keys() == ['https://idp.umu.se/saml2/idp/metadata.php']
idp_sso = md.single_sign_on_services('https://idp.umu.se/saml2/idp/metadata.php')
assert len(idp_sso) == 1
assert idp_sso == ['https://idp.umu.se/saml2/idp/SSOService.php']
ssocerts = md.certs('https://idp.umu.se/saml2/idp/SSOService.php')
print ssocerts
assert len(ssocerts) == 1
def test_incommon_1():
md = metadata.MetaData()
md.import_metadata(open(INCOMMON_METADATA).read())
print len(md)
assert len(md) == 35
print md.keys()
idp_sso = md.single_sign_on_services('urn:mace:incommon:uiuc.edu')
assert idp_sso == []
idp_sso = md.single_sign_on_services('urn:mace:incommon:alaska.edu')
assert len(idp_sso) == 1
print idp_sso
assert idp_sso == ['https://idp.alaska.edu/idp/profile/SAML2/Redirect/SSO']
redirect_idps = [eid for eid in md.keys() if len( md.single_sign_on_services(eid))]
print redirect_idps
assert len(redirect_idps) == 8 # !!!!????