1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
import configparser
import os.path
import sys
import pytest
import bonsai
from bonsai import LDAPReference
from bonsai import LDAPClient
from bonsai import LDAPURL
@pytest.fixture(scope="module")
def host_url():
""" Set host url for connection. """
curdir = os.path.abspath(os.path.dirname(__file__))
cfg = configparser.ConfigParser()
cfg.read(os.path.join(curdir, "test.ini"))
url = "ldap://%s:%s" % (cfg["SERVER"]["hostname"], cfg["SERVER"]["port"])
return url
def test_init_errors(host_url):
""" Testing errors during initialization of LDAPReference. """
client = LDAPClient(host_url)
with pytest.raises(TypeError):
_ = LDAPReference(None, ["ldap://a"])
with pytest.raises(TypeError):
_ = LDAPReference(client, [0])
with pytest.raises(ValueError):
_ = LDAPReference(client, ["asd", LDAPURL("ldap://b")])
def test_client_prop(host_url):
""" Testing client property. """
client = LDAPClient(host_url)
ref = LDAPReference(client, [])
assert ref.client == client
with pytest.raises(TypeError):
ref.client = "b"
ref.client = LDAPClient()
assert ref.client != client
def test_references_prop(host_url):
""" Testing references property. """
client = LDAPClient(host_url)
reflist = [LDAPURL("ldap://localhost"), host_url]
ref = LDAPReference(client, reflist)
assert ref.references == reflist
with pytest.raises(AttributeError):
ref.references = None
@pytest.mark.skipif(
sys.platform.startswith("win"), reason="Referrals are not set in AD."
)
def test_referral_chasing(host_url):
""" Testing referral chasing option. """
refdn = "o=admin,ou=nerdherd-refs,dc=bonsai,dc=test"
client = LDAPClient(host_url)
client.server_chase_referrals = True
with client.connect() as conn:
res = conn.search(refdn, 0)
assert isinstance(res[0], bonsai.LDAPEntry)
client.server_chase_referrals = False
client.ignore_referrals = False
with client.connect() as conn:
res = conn.search(refdn, 0)
assert len(res) == 0
res = conn.search("ou=nerdherd-refs,dc=bonsai,dc=test", 1)
refs = [item for item in res if isinstance(item, LDAPReference)]
assert any(refs)
@pytest.mark.skipif(
sys.platform.startswith("win"), reason="Referrals are not set in AD."
)
def test_ignore_referrals(host_url):
""" Testing ignore referrals option. """
client = LDAPClient(host_url)
client.ignore_referrals = True
with client.connect() as conn:
res = conn.search("ou=nerdherd-refs,dc=bonsai,dc=test", 1)
assert res == []
client.ignore_referrals = False
with client.connect() as conn:
res = conn.search("ou=nerdherd-refs,dc=bonsai,dc=test", 1)
assert len(res) != 0
refs = [item for item in res if isinstance(item, LDAPReference)]
assert any(refs)
|
