diff options
| author | Erick Tryzelaar <erickt@cloudera.com> | 2014-10-24 14:57:34 -0700 |
|---|---|---|
| committer | Erick Tryzelaar <erickt@cloudera.com> | 2014-10-24 16:50:09 -0700 |
| commit | ff5cb7d8ee3d405471c71d19bcfe3bcdf27de848 (patch) | |
| tree | 9226084c3c55e243dd7ea62523f6708476b58e3a /tests/test_37_entity_categories.py | |
| parent | bf9747cd97e364a7dc745693e86177b20716c4d4 (diff) | |
| download | pysaml2-ff5cb7d8ee3d405471c71d19bcfe3bcdf27de848.tar.gz | |
add closing method to make sure to close ident db connection
Diffstat (limited to 'tests/test_37_entity_categories.py')
| -rw-r--r-- | tests/test_37_entity_categories.py | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/tests/test_37_entity_categories.py b/tests/test_37_entity_categories.py index ab6fa539..63fdefa4 100644 --- a/tests/test_37_entity_categories.py +++ b/tests/test_37_entity_categories.py @@ -1,3 +1,4 @@ +from contextlib import closing from saml2 import saml, sigver from saml2 import md from saml2 import config @@ -150,18 +151,17 @@ def test_filter_ava5(): def test_idp_policy_filter(): - idp = Server("idp_conf_ec") + with closing(Server("idp_conf_ec")) as idp: + ava = {"givenName": ["Derek"], "sn": ["Jeter"], + "mail": ["derek@nyy.mlb.com"], "c": ["USA"], + "eduPersonTargetedID": "foo!bar!xyz", + "norEduPersonNIN": "19800101134"} - ava = {"givenName": ["Derek"], "sn": ["Jeter"], - "mail": ["derek@nyy.mlb.com"], "c": ["USA"], - "eduPersonTargetedID": "foo!bar!xyz", - "norEduPersonNIN": "19800101134"} - - policy = idp.config.getattr("policy", "idp") - ava = policy.filter(ava, "urn:mace:example.com:saml:roland:sp", idp.metadata) + policy = idp.config.getattr("policy", "idp") + ava = policy.filter(ava, "urn:mace:example.com:saml:roland:sp", idp.metadata) - print ava - assert ava.keys() == ["eduPersonTargetedID"] # because no entity category + print ava + assert ava.keys() == ["eduPersonTargetedID"] # because no entity category if __name__ == "__main__": test_idp_policy_filter() |