def test_attribute_simple(self): with HTTMock(attribute_uid): res = ldap_brute.brute_nowild(BASE_URL, TRUE_STRING, ldap_brute.common.LDAP_GLOBALS.BRUTE, 3, size_is_exact=True) self.assertEquals(["uid"], res)
def test_nowildcard_simple(self): with HTTMock(nowildcard_admin): res = ldap_brute.brute_nowild(BASE_URL, TRUE_STRING, DEFAULT_ATTRIBUTE, 4, size_is_exact=True) self.assertEquals(['user'], res, "Result should contain user.")
def test_nowildcard_wordlist(self, mocked_method): return_values = ['cn', 'sn', 'uid', 'password'] mocked_method.return_value = iter(return_values) wordlist_custom_set("this file is too good to be true, commas, spaces, we!rd si#ns.") with HTTMock(attribute_uid): res = ldap_brute.brute_nowild(BASE_URL, TRUE_STRING, ldap_brute.common.LDAP_GLOBALS.BRUTE) self.assertEqual(['uid'], res, "Valid results should only be uid.")
def test_nowildcard_wordlist(self, mocked_method): return_values = ['cn', 'sn', 'uid', 'password'] mocked_method.return_value = iter(return_values) wordlist_custom_set( "this file is too good to be true, commas, spaces, we!rd si#ns.") with HTTMock(attribute_uid): res = ldap_brute.brute_nowild(BASE_URL, TRUE_STRING, ldap_brute.common.LDAP_GLOBALS.BRUTE) self.assertEqual(['uid'], res, "Valid results should only be uid.")