def test_default_with_delay(self):
server = LdapServer(java_delay=1)
server.start()
dn = server.config['bind_dn']
pw = server.config['password']
srv = ldap3.Server('localhost', port=server.config['port'])
conn = ldap3.Connection(srv, user=dn, password=pw, auto_bind=True)
base_dn = server.config['base']['dn']
search_filter = '(objectclass=domain)'
attrs = ['dc']
conn.search(base_dn, search_filter, attributes=attrs)
self.assertEqual(conn.response, [{
'dn': 'dc=example,dc=com',
'raw_attributes': {
'dc': [b'example']
},
'attributes': {
'dc': ['example']
},
'type': 'searchResEntry'
}])
conn.unbind()
server.stop()
def test_default_with_delay(self):
server = LdapServer(java_delay=1)
server.start()
dn = server.config['bind_dn']
pw = server.config['password']
srv = ldap3.Server('localhost', port=server.config['port'])
conn = ldap3.Connection(srv, user=dn, password=pw, auto_bind=True)
base_dn = server.config['base']['dn']
search_filter = '(objectclass=domain)'
attrs = ['dc']
conn.search(base_dn, search_filter, attributes=attrs)
self.assertEqual(conn.response, [{
'dn': 'dc=example,dc=com',
'raw_attributes': {'dc': [b'example']},
'attributes': {'dc': ['example']},
'type': 'searchResEntry'
}])
conn.unbind()
server.stop()
def test_config(self):
server = LdapServer({
'port':
3333,
'bind_dn':
'cn=admin,dc=zoldar,dc=net',
'password':
'******',
'base': {
'objectclass': ['domain'],
'dn': 'dc=zoldar,dc=net',
'attributes': {
'dc': 'zoldar'
}
},
'entries': [{
'objectclass': ['domain'],
'dn': 'dc=users,dc=zoldar,dc=net',
'attributes': {
'dc': 'users'
}
}, {
'objectclass': ['organization'],
'dn': 'o=foocompany,dc=users,dc=zoldar,dc=net',
'attributes': {
'o': 'foocompany'
}
}],
})
server.start()
dn = "cn=admin,dc=zoldar,dc=net"
pw = "pass1"
srv = ldap3.Server('localhost', port=3333)
conn = ldap3.Connection(srv, user=dn, password=pw, auto_bind=True)
base_dn = 'dc=zoldar,dc=net'
search_filter = '(objectclass=organization)'
attrs = ['o']
conn.search(base_dn, search_filter, attributes=attrs)
self.assertEqual(conn.response, [{
'dn': 'o=foocompany,dc=users,dc=zoldar,dc=net',
'raw_attributes': {
'o': [b'foocompany']
},
'attributes': {
'o': ['foocompany']
},
'type': 'searchResEntry'
}])
conn.unbind()
server.stop()
def test_config(self):
server = LdapServer({
'port': 3333,
'bind_dn': 'cn=admin,dc=zoldar,dc=net',
'password': '******',
'base': {
'objectclass': ['domain'],
'dn': 'dc=zoldar,dc=net',
'attributes': {'dc': 'zoldar'}
},
'entries': [{
'objectclass': ['domain'],
'dn': 'dc=users,dc=zoldar,dc=net',
'attributes': {'dc': 'users'}
}, {
'objectclass': ['organization'],
'dn': 'o=foocompany,dc=users,dc=zoldar,dc=net',
'attributes': {'o': 'foocompany'}
}],
})
server.start()
dn = "cn=admin,dc=zoldar,dc=net"
pw = "pass1"
srv = ldap3.Server('localhost', port=3333)
conn = ldap3.Connection(srv, user=dn, password=pw, auto_bind=True)
base_dn = 'dc=zoldar,dc=net'
search_filter = '(objectclass=organization)'
attrs = ['o']
conn.search(base_dn, search_filter, attributes=attrs)
self.assertEqual(conn.response, [{
'dn': 'o=foocompany,dc=users,dc=zoldar,dc=net',
'raw_attributes': {'o': [b'foocompany']},
'attributes': {'o': ['foocompany']},
'type': 'searchResEntry'
}])
conn.unbind()
server.stop()
class LDAPTest(object):
"""
Server fixtures, see https://github.com/zoldar/python-ldap-test
"""
ldap_server_data = NotImplemented
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._ldap_test_server = None
self._ldap_connection = None
def setUp(self):
super().setUp()
self._ldap_test_server = LdapServer(self.ldap_server_data)
self._ldap_test_server.start()
ldap3_server = ldap3.Server('localhost', port=self._ldap_test_server.config['port'])
self._ldap_connection = ldap3.Connection(
ldap3_server,
user=self._ldap_test_server.config['bind_dn'],
password=self._ldap_test_server.config['password'],
auto_bind=True
)
def tearDown(self):
super().tearDown()
self._ldap_test_server.stop()
def test_ldap_connectivity(self):
with make_connection(
'ldap://%s:%d' % ('localhost', self._ldap_test_server.config['port']),
self._ldap_test_server.config['bind_dn'],
'toor'
) as conn:
if not conn.bind():
ok_(False, "Cannot establish connection with LDAP test server")
else:
ok_(True)
class LDAPTest(object):
"""
Server fixtures, see https://github.com/zoldar/python-ldap-test
"""
ldap_server_data = NotImplemented
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._ldap_test_server = None
self._ldap_connection = None
def setUp(self):
super().setUp()
self._ldap_test_server = LdapServer(self.ldap_server_data)
self._ldap_test_server.start()
ldap3_server = ldap3.Server('localhost', port=self._ldap_test_server.config['port'])
self._ldap_connection = ldap3.Connection(
ldap3_server,
user=self._ldap_test_server.config['bind_dn'],
password=self._ldap_test_server.config['password'],
auto_bind=True
)
def tearDown(self):
super().tearDown()
self._ldap_test_server.stop()
def test_ldap_connectivity(self):
with make_connection(
'ldap://%s:%d' % ('localhost', self._ldap_test_server.config['port']),
self._ldap_test_server.config['bind_dn'],
'toor'
) as conn:
if not conn.bind():
ok_(False, "Cannot establish connection with LDAP test server")
else:
ok_(True)
class FilterTests(unittest.TestCase):
def __init__(self, methodName):
self.server = LdapServer({
'port': 3333,
'bind_dn': 'cn=admin,dc=cust,dc=local',
'password': '******',
'base': {
'objectclass': ['domain', 'top'],
'dn': 'dc=cust,dc=local',
'attributes': {'dc': 'cust'}
},
'ldifs': [
'tests/user-container.ldif',
'tests/enabled-exchange-user.ldif',
# 'tests/disabled-exchange-user.ldif',
'tests/enabled-non-exchange-user.ldif',
'tests/disabled-non-exchange-user.ldif',
]
})
self.server.start()
super(FilterTests, self).__init__(methodName)
def teardown_func(self):
self.server.stop()
def test_connectivity(self):
dn = self.server.config['bind_dn']
pw = self.server.config['password']
con = ldap.initialize('ldap://localhost:%s' % (self.server.config['port'],))
con.simple_bind_s(dn, pw)
base_dn = self.server.config['base']['dn']
filter = '(objectclass=*)'
attrs = ['dc']
print con.search_s(base_dn, ldap.SCOPE_SUBTREE, filter, attrs)
class MockLdapServer(object):
ldap_url = 'ldap://ldap.python-hvac.org'
ldap_group_name = 'vault-users'
ldap_user_name = 'somedude'
ldap_user_password = '******'
ldap_base_dc = 'hvac'
ldap_base_dn = 'dc={dc},dc=network'.format(dc=ldap_base_dc)
ldap_bind_dn = 'cn=admin,{base_dn}'.format(base_dn=ldap_base_dn)
ldap_bind_password = '******'
ldap_users_dn = 'dc=users,{base_dn}'.format(base_dn=ldap_base_dn)
ldap_groups_ou = 'groups'
ldap_groups_dn = 'ou={ou},{base_dn}'.format(ou=ldap_groups_ou,
base_dn=ldap_base_dn)
ldap_login_user_dn = 'uid={username},{users_dn}'.format(
username=ldap_user_name, users_dn=ldap_users_dn)
ldap_entries = [
{
'objectclass': 'domain',
'dn': ldap_users_dn,
'attributes': {
'dc': 'users'
}
},
{
'objectclass': ['inetorgperson', 'posixgroup', 'top'],
'dn': ldap_login_user_dn,
'attributes': {
'uid': ldap_user_name,
'userpassword': ldap_user_password
}
},
{
'objectclass': 'organizationalunit',
'dn': ldap_groups_dn,
'attributes': {
'ou': 'groups',
}
},
{
'objectclass':
'groupofnames',
'dn':
'cn={cn},{groups_dn}'.format(cn=ldap_group_name,
groups_dn=ldap_groups_dn),
'attributes': {
'cn': ldap_group_name,
'member': ldap_login_user_dn,
}
},
]
def __init__(self):
self.server_port = get_free_port()
self.ldap_server = LdapServer({
'port': self.server_port,
'bind_dn': self.ldap_bind_dn,
'password': self.ldap_bind_password,
'base': {
'objectclass': ['domain'],
'dn': self.ldap_base_dn,
'attributes': {
'dc': self.ldap_base_dc
}
},
'entries': self.ldap_entries,
})
@property
def url(self):
return 'ldap://localhost:{port}'.format(port=self.server_port)
def start(self):
self.ldap_server.start()
def stop(self):
self.ldap_server.stop()
class MockLdapServer(object):
ldap_url = 'ldap://ldap.python-hvac.org'
ldap_group_name = 'vault-users'
ldap_user_name = 'somedude'
ldap_user_password = '******'
ldap_base_dc = 'hvac'
ldap_base_dn = 'dc={dc},dc=network'.format(dc=ldap_base_dc)
ldap_bind_dn = 'cn=admin,{base_dn}'.format(base_dn=ldap_base_dn)
ldap_bind_password = '******'
ldap_users_dn = 'dc=users,{base_dn}'.format(base_dn=ldap_base_dn)
ldap_groups_ou = 'groups'
ldap_groups_dn = 'ou={ou},{base_dn}'.format(ou=ldap_groups_ou, base_dn=ldap_base_dn)
ldap_login_user_dn = 'uid={username},{users_dn}'.format(username=ldap_user_name, users_dn=ldap_users_dn)
ldap_entries = [
{
'objectclass': 'domain',
'dn': ldap_users_dn,
'attributes': {
'dc': 'users'
}
},
{
'objectclass': ['inetorgperson', 'posixgroup', 'top'],
'dn': ldap_login_user_dn,
'attributes': {
'uid': ldap_user_name,
'userpassword': ldap_user_password
}
},
{
'objectclass': 'organizationalunit',
'dn': ldap_groups_dn,
'attributes': {
'ou': 'groups',
}
},
{
'objectclass': 'groupofnames',
'dn': 'cn={cn},{groups_dn}'.format(cn=ldap_group_name, groups_dn=ldap_groups_dn),
'attributes': {
'cn': ldap_group_name,
'member': ldap_login_user_dn,
}
},
]
def __init__(self):
self.server_port = get_free_port()
self.ldap_server = LdapServer({
'port': self.server_port,
'bind_dn': self.ldap_bind_dn,
'password': self.ldap_bind_password,
'base': {
'objectclass': ['domain'],
'dn': self.ldap_base_dn,
'attributes': {'dc': self.ldap_base_dc}
},
'entries': self.ldap_entries,
})
@property
def url(self):
return 'ldap://localhost:{port}'.format(port=self.server_port)
def start(self):
self.ldap_server.start()
def stop(self):
self.ldap_server.stop()
'o': 'foocompany'
}
},
]
})
try:
server.start()
dn = "cn=admin,dc=luochunhui,dc=com"
pw = "pass1"
srv = ldap3.Server('localhost', port=3333)
conn = ldap3.Connection(srv, user=dn, password=pw, auto_bind=True)
base_dn = 'dc=luochunhui,dc=com'
search_filter = '(objectclass=organization)'
attrs = ['o']
conn.search(base_dn, search_filter, attributes=attrs)
print conn.response
# [{
# 'dn': 'o=foocompany,dc=users,dc=luochunhui,dc=com',
# 'raw_attributes': {'o': [b'foocompany']},
# 'attributes': {'o': ['foocompany']},
# 'type': 'searchResEntry'
# }]
finally:
server.stop()
server.start()
dn = server.config['bind_dn']
pw = server.config['password']
con = ldap.initialize('ldap://localhost:%s' % (server.config['port'],))
con.simple_bind_s(dn, pw)
base_dn = server.config['base']['dn']
filter = '(objectclass=domain)'
attrs = ['dc']
print con.search_s(base_dn, ldap.SCOPE_SUBTREE, filter, attrs)
finally:
server.stop()
server = LdapServer({
'port': 3333,
'bind_dn': 'cn=admin,dc=zoldar,dc=net',
'password': '******',
'base': {'objectclass': ['domain'],
'dn': 'dc=zoldar,dc=net',
'attributes': {'dc': 'zoldar'}},
'entries': [
{'objectclass': 'domain',
'dn': 'dc=users,dc=zoldar,dc=net',
'attributes': {'dc': 'users'}},
{'objectclass': 'organization',
'dn': 'o=foocompany,dc=users,dc=zoldar,dc=net',
'attributes': {'o': 'foocompany'}},
