def test_xivo_ldap_init(self, ldap_initialize): ldapobj = ldap_initialize.return_value = Mock() _XivoLDAP(self.config['uri']) ldap_initialize.assert_called_once_with(self.config['uri']) ldapobj.set_option.assert_any_call(ldap.OPT_REFERRALS, 0) ldapobj.set_option.assert_any_call(ldap.OPT_NETWORK_TIMEOUT, 2) ldapobj.set_option.assert_any_call(ldap.OPT_TIMEOUT, 2)
def test_that_perform_search(self, ldap_initialize): ldapobj = ldap_initialize.return_value = Mock() xivo_ldap = _XivoLDAP(self.config) ldapobj.search_ext_s.return_value = ['result1'] result = xivo_ldap.perform_search('base', 'scope') self.assertEqual(result, 'result1')
def test_that_perform_bind_return_false_when_no_wrong_credential(self, ldap_initialize): ldapobj = ldap_initialize.return_value = Mock() xivo_ldap = _XivoLDAP(self.config) ldapobj.simple_bind_s.side_effect = ldap.INVALID_CREDENTIALS() result = xivo_ldap.perform_bind('username', 'password') self.assertEqual(result, False)
def test_that_perform_search_return_none_when_no_result(self, ldap_initialize): ldapobj = ldap_initialize.return_value = Mock() xivo_ldap = _XivoLDAP(self.config) ldapobj.search_ext_s.return_value = [] result_dn, result_attr = xivo_ldap.perform_search('base', 'scope') self.assertEqual(result_dn, None) self.assertEqual(result_attr, None)
def test_that_perform_search_return_none_when_multiple_result(self, ldap_initialize): ldapobj = ldap_initialize.return_value = Mock() xivo_ldap = _XivoLDAP(self.config) ldapobj.search_ext_s.side_effect = ldap.SIZELIMIT_EXCEEDED() result_dn, result_attr = xivo_ldap.perform_search('base', 'scope') self.assertEqual(result_dn, None) self.assertEqual(result_attr, None)
def test_that_perform_bind(self): xivo_ldap = _XivoLDAP(self.config) result = xivo_ldap.perform_bind('username', 'password') self.assertEqual(result, True)