def test_09_ldaps(self):
# Check that use_ssl and tls are actually passed to the Connection
ldap3mock.setLDAPDirectory(LDAPDirectory)
config = MYCONFIG.copy()
config['LDAPURI'] = 'ldaps://1.2.3.4'
config['TLS_VERIFY'] = '1'
ldaps_resolver = LdapMachineResolver("myResolver", config=config)
machines = ldaps_resolver.get_machines()
self.assertEqual(len(machines), 3)
# We check that all Server objects were constructed with a non-None TLS context and use_ssl=True
for _, kwargs in ldap3mock.get_server_mock().call_args_list:
self.assertIsNotNone(kwargs['tls'])
self.assertTrue(kwargs['use_ssl'])
def test_09_ldaps(self):
# Check that use_ssl and tls are actually passed to the Connection
ldap3mock.setLDAPDirectory(LDAPDirectory)
config = MYCONFIG.copy()
config['LDAPURI'] = 'ldaps://1.2.3.4'
config['TLS_VERIFY'] = '1'
ldaps_resolver = LdapMachineResolver("myResolver", config=config)
machines = ldaps_resolver.get_machines()
self.assertEqual(len(machines), 3)
# We check that all Server objects were constructed with a non-None TLS context and use_ssl=True
for _, kwargs in ldap3mock.get_server_mock().call_args_list:
self.assertIsNotNone(kwargs['tls'])
self.assertTrue(kwargs['use_ssl'])
def test_08_start_tls(self):
# Check that START_TLS and TLS_VERIFY are actually passed to the ldap3 Connection
ldap3mock.setLDAPDirectory(LDAPDirectory)
config = MYCONFIG.copy()
config['START_TLS'] = '1'
config['TLS_VERIFY'] = '1'
start_tls_resolver = LdapMachineResolver("myResolver", config=config)
machines = start_tls_resolver.get_machines()
self.assertEqual(len(machines), 3)
# We check two things:
# 1) start_tls has actually been called!
self.assertTrue(start_tls_resolver.l.start_tls_called)
# 2) All Server objects were constructed with a non-None TLS context, but use_ssl=False
for _, kwargs in ldap3mock.get_server_mock().call_args_list:
self.assertIsNotNone(kwargs['tls'])
self.assertFalse(kwargs['use_ssl'])
def test_08_start_tls(self):
# Check that START_TLS and TLS_VERIFY are actually passed to the ldap3 Connection
ldap3mock.setLDAPDirectory(LDAPDirectory)
config = MYCONFIG.copy()
config['START_TLS'] = '1'
config['TLS_VERIFY'] = '1'
start_tls_resolver = LdapMachineResolver("myResolver", config=config)
machines = start_tls_resolver.get_machines()
self.assertEqual(len(machines), 3)
# We check two things:
# 1) start_tls has actually been called!
self.assertTrue(start_tls_resolver.l.start_tls_called)
# 2) All Server objects were constructed with a non-None TLS context, but use_ssl=False
for _, kwargs in ldap3mock.get_server_mock().call_args_list:
self.assertIsNotNone(kwargs['tls'])
self.assertFalse(kwargs['use_ssl'])
def test_07_testconnection(self):
ldap3mock.setLDAPDirectory(LDAPDirectory)
(success, desc) = LdapMachineResolver.testconnection(MYCONFIG)
self.assertTrue(success)
self.assertEqual(
desc, "Your LDAP config seems to be OK, 3 "
"machine objects found.")
def test_07_testconnection(self):
ldap3mock.setLDAPDirectory(LDAPDirectory)
(success, desc) = LdapMachineResolver.testconnection(MYCONFIG)
self.assertTrue(success)
self.assertTrue("3" in desc)
class LdapMachineTestCase(MyTestCase):
"""
Test the LDAP Resolver
"""
mreso = LdapMachineResolver("myResolver", config=MYCONFIG)
mrAD = LdapMachineResolver("myreso",
config={
"LDAPURI": "ldap://172.16.200.202",
"LDAPBASE": "dc=privacyidea,"
"dc=test",
"BINDDN": "*****@*****.**",
"BINDPW": "Test1234!",
"HOSTNAMEATTRIBUTE": "dNSHostName"
})
def test_01_get_config_description(self):
# Missing LDAPURI
self.assertRaises(MachineResolverError,
LdapMachineResolver,
"reso2",
config={"IDATTRIBUTE": "na"})
# Missing LDAPBASE
self.assertRaises(MachineResolverError,
LdapMachineResolver,
"reso2",
config={"LDAPURI": "ldap://1.2.3.4"})
desc = self.mreso.get_config_description()
self.assertEqual(
desc.get("ldap").get("config").get("LDAPURI"), "string")
"""
def test_01_get_machines(self):
machines = self.mrAD.get_machines()
self.assertEqual(len(machines), 1)
self.assertEqual(machines[0].hostname, "dc01.privacyidea.test")
def test_02_get_machine_id(self):
id = self.mrAD.get_machine_id(hostname="dc01.privacyidea.test")
self.assertEqual(id,
"CN=DC01,OU=Domain Controllers,DC=privacyidea,DC=test")
"""
@ldap3mock.activate
def test_03_get_machines(self):
ldap3mock.setLDAPDirectory(LDAPDirectory)
machines = self.mreso.get_machines()
self.assertEqual(len(machines), 3)
machines = self.mreso.get_machines(hostname="machine1.example.test")
self.assertEqual(len(machines), 1)
# THere is one machine, that contains "e2"
machines = self.mreso.get_machines(hostname="e2", substring=True)
self.assertEqual(len(machines), 1)
def test_04_get_machines_any(self):
# The MockLdap module can not handle complicates ldap searches like
# (&(&( )( )( ))(|( )( )( )) )
# so we only test the creation of the ldapsearch string
filter = LdapMachineResolver.\
_create_ldap_filter("(""objectClass=computer)",
"objectSid", "",
"dNSHostName", "",
"", "",
substring=True, any="substr")
self.assertEqual(
filter, "(&(&(objectClass=computer))"
"(|(objectSid=*substr*)("
"dNSHostName=*substr*)))")
def test_05_get_single_machine(self):
machines = self.mreso.get_machines(
machine_id="cn=machine1,ou=example,o=test")
self.assertEqual(len(machines), 1)
machine = machines[0]
self.assertEqual(machine.id, "cn=machine1,ou=example,o=test")
self.assertTrue(machine.has_hostname("machine1.example.test"))
def test_06_get_machine_id(self):
id = self.mreso.get_machine_id(hostname="not existing")
self.assertTrue(id is None)
id = self.mreso.get_machine_id(hostname="machine1.example.test")
self.assertEqual(id, "cn=machine1,ou=example,o=test")
@ldap3mock.activate
def test_07_testconnection(self):
ldap3mock.setLDAPDirectory(LDAPDirectory)
(success, desc) = LdapMachineResolver.testconnection(MYCONFIG)
self.assertTrue(success)
self.assertTrue("3" in desc)
# "Your LDAP config seems to be OK, 3 machine objects found."
@ldap3mock.activate
def test_08_start_tls(self):
# Check that START_TLS and TLS_VERIFY are actually passed to the ldap3 Connection
ldap3mock.setLDAPDirectory(LDAPDirectory)
config = MYCONFIG.copy()
config['START_TLS'] = '1'
config['TLS_VERIFY'] = '1'
start_tls_resolver = LdapMachineResolver("myResolver", config=config)
machines = start_tls_resolver.get_machines()
self.assertEqual(len(machines), 3)
# We check two things:
# 1) start_tls has actually been called!
self.assertTrue(start_tls_resolver.l.start_tls_called)
# 2) All Server objects were constructed with a non-None TLS context, but use_ssl=False
for _, kwargs in ldap3mock.get_server_mock().call_args_list:
self.assertIsNotNone(kwargs['tls'])
self.assertFalse(kwargs['use_ssl'])
@ldap3mock.activate
def test_09_ldaps(self):
# Check that use_ssl and tls are actually passed to the Connection
ldap3mock.setLDAPDirectory(LDAPDirectory)
config = MYCONFIG.copy()
config['LDAPURI'] = 'ldaps://1.2.3.4'
config['TLS_VERIFY'] = '1'
ldaps_resolver = LdapMachineResolver("myResolver", config=config)
machines = ldaps_resolver.get_machines()
self.assertEqual(len(machines), 3)
# We check that all Server objects were constructed with a non-None TLS context and use_ssl=True
for _, kwargs in ldap3mock.get_server_mock().call_args_list:
self.assertIsNotNone(kwargs['tls'])
self.assertTrue(kwargs['use_ssl'])
def test_07_testconnection(self):
ldap3mock.setLDAPDirectory(LDAPDirectory)
(success, desc) = LdapMachineResolver.testconnection(MYCONFIG)
self.assertTrue(success)
self.assertTrue("3" in desc)
class LdapMachineTestCase(MyTestCase):
"""
Test the LDAP Resolver
"""
mreso = LdapMachineResolver("myResolver", config=MYCONFIG)
mrAD = LdapMachineResolver("myreso",
config={
"LDAPURI": "ldap://172.16.200.202",
"LDAPBASE": "dc=privacyidea,"
"dc=test",
"BINDDN": "*****@*****.**",
"BINDPW": "Test1234!",
"HOSTNAMEATTRIBUTE": "dNSHostName"
})
def test_01_get_config_description(self):
# Missing LDAPURI
self.assertRaises(MachineResolverError,
LdapMachineResolver,
"reso2",
config={"IDATTRIBUTE": "na"})
# Missing LDAPBASE
self.assertRaises(MachineResolverError,
LdapMachineResolver,
"reso2",
config={"LDAPURI": "ldap://1.2.3.4"})
desc = self.mreso.get_config_description()
self.assertEqual(
desc.get("ldap").get("config").get("LDAPURI"), "string")
"""
def test_01_get_machines(self):
machines = self.mrAD.get_machines()
self.assertEqual(len(machines), 1)
self.assertEqual(machines[0].hostname, "dc01.privacyidea.test")
def test_02_get_machine_id(self):
id = self.mrAD.get_machine_id(hostname="dc01.privacyidea.test")
self.assertEqual(id,
"CN=DC01,OU=Domain Controllers,DC=privacyidea,DC=test")
"""
@ldap3mock.activate
def test_03_get_machines(self):
ldap3mock.setLDAPDirectory(LDAPDirectory)
machines = self.mreso.get_machines()
self.assertEqual(len(machines), 3)
machines = self.mreso.get_machines(hostname="machine1.example.test")
self.assertEqual(len(machines), 1)
# THere is one machine, that contains "e2"
machines = self.mreso.get_machines(hostname="e2", substring=True)
self.assertEqual(len(machines), 1)
def test_04_get_machines_any(self):
# The MockLdap module can not handle complicates ldap searches like
# (&(&( )( )( ))(|( )( )( )) )
# so we only test the creation of the ldapsearch string
filter = LdapMachineResolver.\
_create_ldap_filter("(""objectClass=computer)",
"objectSid", "",
"dNSHostName", "",
"", "",
substring=True, any="substr")
self.assertEqual(
filter, "(&(&(objectClass=computer))"
"(|(objectSid=*substr*)("
"dNSHostName=*substr*)))")
def test_05_get_single_machine(self):
machines = self.mreso.get_machines(
machine_id="cn=machine1,ou=example,o=test")
self.assertEqual(len(machines), 1)
machine = machines[0]
self.assertEqual(machine.id, "cn=machine1,ou=example,o=test")
self.assertTrue(machine.has_hostname("machine1.example.test"))
def test_06_get_machine_id(self):
id = self.mreso.get_machine_id(hostname="not existing")
self.assertTrue(id is None)
id = self.mreso.get_machine_id(hostname="machine1.example.test")
self.assertEqual(id, "cn=machine1,ou=example,o=test")
@ldap3mock.activate
def test_07_testconnection(self):
ldap3mock.setLDAPDirectory(LDAPDirectory)
(success, desc) = LdapMachineResolver.testconnection(MYCONFIG)
self.assertTrue(success)
self.assertTrue("3" in desc)
def test_07_testconnection(self):
ldap3mock.setLDAPDirectory(LDAPDirectory)
(success, desc) = LdapMachineResolver.testconnection(MYCONFIG)
self.assertTrue(success)
self.assertEqual(desc, "Your LDAP config seems to be OK, 3 "
"machine objects found.")