99911f6c4c
Retains python2.7 compatibility for all files. Fixes only syntax errors, tests still fail on python3 for various reasons.
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from saml2.eptid import Eptid, EptidShelve
|
|
|
|
__author__ = 'rolandh'
|
|
|
|
|
|
def test_eptid():
|
|
edb = Eptid("secret")
|
|
e1 = edb.get("idp_entity_id", "sp_entity_id", "user_id", "some other data")
|
|
print(e1)
|
|
assert e1.startswith("idp_entity_id!sp_entity_id!")
|
|
e2 = edb.get("idp_entity_id", "sp_entity_id", "user_id", "some other data")
|
|
assert e1 == e2
|
|
|
|
e3 = edb.get("idp_entity_id", "sp_entity_id", "user_2", "some other data")
|
|
print(e3)
|
|
assert e1 != e3
|
|
|
|
e4 = edb.get("idp_entity_id", "sp_entity_id2", "user_id", "some other data")
|
|
assert e4 != e1
|
|
assert e4 != e3
|
|
|
|
|
|
def test_eptid_shelve():
|
|
edb = EptidShelve("secret", "eptid.db")
|
|
e1 = edb.get("idp_entity_id", "sp_entity_id", "user_id", "some other data")
|
|
print(e1)
|
|
assert e1.startswith("idp_entity_id!sp_entity_id!")
|
|
e2 = edb.get("idp_entity_id", "sp_entity_id", "user_id", "some other data")
|
|
assert e1 == e2
|
|
|
|
e3 = edb.get("idp_entity_id", "sp_entity_id", "user_2", "some other data")
|
|
print(e3)
|
|
assert e1 != e3
|
|
|
|
e4 = edb.get("idp_entity_id", "sp_entity_id2", "user_id", "some other data")
|
|
assert e4 != e1
|
|
assert e4 != e3
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_eptid_shelve()
|