def test_given_unsuccessful_fed_id_search_then_error_raised(self, mock_ldap):
fed_id = "not_a_fed123"
ldap_inst = MagicMock()
mock_ldap.initialize.return_value = ldap_inst
ldap_inst.result.return_value = \
(115, [(None, ['ldap://res02.fed.cclrc.ac.uk/DC=res02,DC=fed,DC=cclrc,DC=ac,DC=uk'])])
with self.assertRaises(Exception):
dls_module_contacts.lookup_contact_name(fed_id)
def test_search_parameters_correct(self, mock_ldap):
fed_id = "fed123"
basedn = "dc=fed,dc=cclrc,dc=ac,dc=uk"
search_filter = "(&(cn={}))".format(fed_id)
search_attribute = ["givenName", "sn"]
mock_ldap.SCOPE_SUBTREE.return_value = "test_scope_subtree"
search_scope = mock_ldap.SCOPE_SUBTREE
ldap_inst = MagicMock()
mock_ldap.initialize.return_value = ldap_inst
dls_module_contacts.lookup_contact_name(fed_id)
mock_ldap.initialize.assert_called_once_with("ldap://altfed.cclrc.ac.uk")
ldap_inst.search.assert_called_once_with(basedn, search_scope, search_filter, search_attribute)
def test_given_successful_fed_id_search_then_no_error_raised(self, mock_ldap):
fed_id = "fed123"
ldap_inst = MagicMock()
mock_ldap.initialize.return_value = ldap_inst
ldap_inst.result.return_value = \
(100, [('CN=<FED-ID>,OU=DLS,DC=fed,DC=cclrc,DC=ac,DC=uk',
{'givenName': ['<FirstName>'], 'sn': ['<Surname>']})])
contact_name = dls_module_contacts.lookup_contact_name(fed_id)
self.assertEqual(contact_name, "<FirstName> <Surname>")
