class TestVirtualOrg(): def setup_class(self): conf = config.SPConfig() conf.load_file("server_conf") self.sp = Saml2Client(conf) vo_name = conf.virtual_organization.keys()[0] self.vo = VirtualOrg(self.sp, vo_name) add_derek_info(self.sp) def test_mta(self): aas = self.vo.members_to_ask("abcdefgh") print aas assert len(aas) == 2 assert 'urn:mace:example.com:saml:aa' in aas assert 'urn:mace:example.com:saml:idp' in aas def test_unknown_subject(self): aas = self.vo.members_to_ask("01234567") print aas assert len(aas) == 0 def test_id(self): id = self.vo.get_common_identifier("abcdefgh") print id assert id == "deje0001" def test_id_unknown(self): id = self.vo.get_common_identifier("01234567") assert id is None