diff options
author | Ivan Kanakarakis <ivan.kanak@gmail.com> | 2019-05-14 15:01:35 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-05-14 15:01:35 +0200 |
commit | d655fc924af5ddd56a51d1e6bb41e6bf3d3d1843 (patch) | |
tree | 10ff0de30e1d253be29ce8438b5d69fe610fde16 /tests/test_37_entity_categories.py | |
parent | 1d06338385c2c59964560fd7ec0fc51d5339650b (diff) | |
parent | 15bdc66ac776e04777666fff3d08a38e24f5305a (diff) | |
download | pysaml2-d655fc924af5ddd56a51d1e6bb41e6bf3d3d1843.tar.gz |
Merge pull request #613 from skoranda/more_flexible_entity_category_import
Make entity category imports more flexible
Diffstat (limited to 'tests/test_37_entity_categories.py')
-rw-r--r-- | tests/test_37_entity_categories.py | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/tests/test_37_entity_categories.py b/tests/test_37_entity_categories.py index 625caaa1..839030fd 100644 --- a/tests/test_37_entity_categories.py +++ b/tests/test_37_entity_categories.py @@ -152,5 +152,44 @@ def test_idp_policy_filter(): "eduPersonTargetedID"] # because no entity category +def test_entity_category_import_from_path(): + # The entity category module myentitycategory.py is in the tests + # directory which is on the standard module search path. + # The module uses a custom interpretation of the REFEDs R&S entity category + # by adding eduPersonUniqueId. + policy = Policy({ + "default": { + "lifetime": {"minutes": 15}, + "entity_categories": ["myentitycategory"] + } + }) + + mds = MetadataStore(ATTRCONV, sec_config, + disable_ssl_certificate_validation=True) + + # The file entity_cat_rs.xml contains the SAML metadata for an SP + # tagged with the REFEDs R&S entity category. + mds.imp([{"class": "saml2.mdstore.MetaDataFile", + "metadata": [(full_path("entity_cat_rs.xml"),)]}]) + + ava = {"givenName": ["Derek"], "sn": ["Jeter"], + "displayName": "Derek Jeter", + "mail": ["derek@nyy.mlb.com"], "c": ["USA"], + "eduPersonTargetedID": "foo!bar!xyz", + "eduPersonUniqueId": "R13ET7UD68K0HGR153KE@my.org", + "eduPersonScopedAffiliation": "member@my.org", + "eduPersonPrincipalName": "user01@my.org", + "norEduPersonNIN": "19800101134"} + + ava = policy.filter(ava, "urn:mace:example.com:saml:roland:sp", mds) + + # We expect c and norEduPersonNIN to be filtered out since they are not + # part of the custom entity category. + assert _eq(list(ava.keys()), + ["eduPersonTargetedID", "eduPersonPrincipalName", + "eduPersonUniqueId", "displayName", "givenName", + "eduPersonScopedAffiliation", "mail", "sn"]) + + if __name__ == "__main__": test_filter_ava3() |