1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
from saml2.saml import NAMEID_FORMAT_TRANSIENT
from saml2.saml import NameID
__author__ = "rolandh"
from saml2 import config
from saml2.client import Saml2Client
from saml2.time_util import in_a_while
from saml2.time_util import str_to_time
SESSION_INFO_PATTERN = {"ava": {}, "came from": "", "not_on_or_after": 0, "issuer": "", "session_id": -1}
nid = NameID(name_qualifier="foo", format=NAMEID_FORMAT_TRANSIENT, text="abcdefgh")
nid0 = NameID(name_qualifier="foo", format=NAMEID_FORMAT_TRANSIENT, text="01234567")
def add_derek_info(sp):
not_on_or_after = str_to_time(in_a_while(days=1))
session_info = SESSION_INFO_PATTERN.copy()
session_info["ava"] = {"givenName": ["Derek"], "umuselin": ["deje0001"]}
session_info["issuer"] = "urn:mace:example.com:saml:idp"
session_info["name_id"] = nid
session_info["not_on_or_after"] = not_on_or_after
# subject_id, entity_id, info, timestamp
sp.users.add_information_about_person(session_info)
class TestVirtualOrg:
def setup_class(self):
conf = config.SPConfig()
conf.load_file("server_conf")
self.sp = Saml2Client(conf)
vo_name = list(conf.vorg.keys())[0]
self.vo = conf.vorg[vo_name]
add_derek_info(self.sp)
def test_mta(self):
aas = self.vo.members_to_ask(nid)
print(aas)
assert len(aas) == 1
assert "urn:mace:example.com:saml:aa" in aas
def test_unknown_subject(self):
aas = self.vo.members_to_ask(nid0)
print(aas)
assert len(aas) == 2
def test_id(self):
cid = self.vo.get_common_identifier(nid)
print(cid)
assert cid == "deje0001"
def test_id_unknown(self):
cid = self.vo.get_common_identifier(nid0)
assert cid is None
class TestVirtualOrg_2:
def setup_class(self):
conf = config.SPConfig()
conf.load_file("server_conf")
vo_name = list(conf.vorg.keys())[0]
self.sp = Saml2Client(conf, virtual_organization=vo_name)
add_derek_info(self.sp)
def test_mta(self):
aas = self.sp.vorg.members_to_ask(nid)
print(aas)
assert len(aas) == 1
assert "urn:mace:example.com:saml:aa" in aas
def test_unknown_subject(self):
aas = self.sp.vorg.members_to_ask(nid0)
print(aas)
assert len(aas) == 2
def test_id(self):
cid = self.sp.vorg.get_common_identifier(nid)
print(cid)
assert cid == "deje0001"
def test_id_unknown(self):
cid = self.sp.vorg.get_common_identifier(nid0)
assert cid is None
|
