Tests for full coverage of async_model.

This commit is contained in:
sfigato 2019-05-01 14:21:30 +02:00
parent b466bf8ed2
commit 8dce6566ee
Signed by: blallo
GPG Key ID: 0CBE577C9B72DC3F

View File

@ -9,20 +9,28 @@ from phi.ldap.async_model import (
recall, recall,
call_if_callable, call_if_callable,
inheritance, inheritance,
iter_children,
Entry, Entry,
build_heritage,
Hackers, Hackers,
Robots, Robots,
Congregations,
Service, Service,
User, User,
Group,
) )
BASE_DN = "dc=test,dc=abbiamoundominio,dc=org" BASE_DN = "dc=test,dc=abbiamoundominio,dc=org"
USER_LIST = [{"uid": ["conte_mascetti"]}, {"uid": ["perozzi"]}, {"uid": ["necchi"]}]
SERVICE_LIST = [{"uid": ["phi"]}, {"uid": ["irc"]}]
GROUP_LIST = [{"cn": ["amici_miei"]}, {"cn": ["antani"]}]
class MockClient: class MockClient:
def __init__(self, base_dn): def __init__(self, base_dn, *args):
self._base_dn = base_dn self._base_dn = base_dn
self.args = args
@property @property
def base_dn(self): def base_dn(self):
@ -31,9 +39,14 @@ class MockClient:
@asynccontextmanager @asynccontextmanager
async def connect(self, *args, **kwargs): async def connect(self, *args, **kwargs):
conn = Namespace() conn = Namespace()
async def _search(*args, **kwargs): async def _search(*args, **kwargs):
await asyncio.sleep(0.01) if "Services" in self.args:
return "Fake description" return SERVICE_LIST
elif "Groups" in self.args:
return GROUP_LIST
return USER_LIST
conn.search = _search conn.search = _search
yield conn yield conn
@ -43,6 +56,15 @@ def client_fixture():
return MockClient(BASE_DN) return MockClient(BASE_DN)
@pytest.fixture
def client_fixture_multi():
res = Namespace()
res.users = MockClient(BASE_DN)
res.services = MockClient(BASE_DN, "Services")
res.groups = MockClient(BASE_DN, "Groups")
return res
@pytest.fixture @pytest.fixture
def lineage_fixture(): def lineage_fixture():
class Grand: class Grand:
@ -113,6 +135,19 @@ def test_inheritance(lineage_fixture):
assert inheritance(Child, "name", Ma) == "Child" assert inheritance(Child, "name", Ma) == "Child"
@pytest.mark.asyncio
async def test_iter_children():
LIST = [1, 2, 3, 4]
async def _async_gen():
for i in LIST:
yield i
ALIST = _async_gen()
assert LIST == await iter_children(ALIST)
def test_Entry(client_fixture): def test_Entry(client_fixture):
e = Entry(client_fixture) e = Entry(client_fixture)
@ -142,21 +177,115 @@ def test_Entry_dn(client_fixture):
async def test_Entry_describe(client_fixture): async def test_Entry_describe(client_fixture):
e = Entry(client_fixture) e = Entry(client_fixture)
assert "Fake description" == await e.describe() assert USER_LIST == await e.describe()
def test_Hackers(client_fixture): @pytest.mark.asyncio
h = Hackers(client_fixture) async def test_build_heritage(client_fixture):
class MockAIterable(object):
client = None
def __init__(self, elements):
self.elements = elements
async def get_children(self):
for el in self.elements:
yield el
class Dummy(object):
def __init__(self, *args, **kwargs):
self.id = sum([id(el) for el in args]) + sum(
[id(v) for _, v in kwargs.items()]
)
def __repr__(self):
return f"<Dummy({self.id})>"
def __eq__(self, other):
return self.id == other.id
m = MockAIterable(USER_LIST)
res = [el async for el in build_heritage(m, Dummy, attribute_id="uid")]
exp_res = [Dummy(el["uid"][0], None) for el in USER_LIST]
assert res == exp_res
def test_Hackers(client_fixture_multi):
h = Hackers(client_fixture_multi.users)
assert h.kind == "ou" assert h.kind == "ou"
assert h.dn == "ou={},{}".format(h.name(), BASE_DN) assert h.dn == "ou={},{}".format(h.name(), BASE_DN)
assert repr(h) == f"<Hackers {h.kind}=Hackers,{BASE_DN}>"
def test_Robots(client_fixture): @pytest.mark.asyncio
r = Robots(client_fixture) async def test_Entry_get_children(client_fixture_multi):
h = Hackers(client_fixture_multi.users)
assert USER_LIST == [el async for el in h.get_children()]
@pytest.mark.asyncio
async def test_Hackers_anext(client_fixture_multi):
h = Hackers(client_fixture_multi.users)
exp_res = [User(el["uid"][0], client_fixture_multi.users) for el in USER_LIST]
assert exp_res == [el async for el in h]
def test_Hackers_children(client_fixture_multi):
h = Hackers(client_fixture_multi.users)
assert USER_LIST == h.children
def test_Robots(client_fixture_multi):
r = Robots(client_fixture_multi.services)
assert r.kind == "ou" assert r.kind == "ou"
assert r.dn == "ou={},{}".format(r.name(), BASE_DN) assert r.dn == "ou={},{}".format(r.name(), BASE_DN)
assert repr(r) == f"<Services {r.kind}=Services,{BASE_DN}>"
@pytest.mark.asyncio
async def test_Robots_anext(client_fixture_multi):
r = Robots(client_fixture_multi.services)
exp_res = [Service(el["uid"][0], client_fixture_multi.services) for el in SERVICE_LIST]
assert exp_res == [el async for el in r]
def test_Robots_children(client_fixture_multi):
r = Robots(client_fixture_multi.services)
assert SERVICE_LIST == r.children
def test_Congregations(client_fixture_multi):
g = Congregations(client_fixture_multi.groups)
assert g.kind == "ou"
assert g.dn == "ou={},{}".format(g.name(), BASE_DN)
assert repr(g) == f"<Groups {g.kind}=Groups,{BASE_DN}>"
@pytest.mark.asyncio
async def test_Congregations_anext(client_fixture_multi):
c = Congregations(client_fixture_multi.groups)
exp_res = [Group(el["cn"][0], client_fixture_multi.groups) for el in GROUP_LIST]
assert exp_res == [el async for el in c]
def test_Congregations_children(client_fixture_multi):
c = Congregations(client_fixture_multi.groups)
assert GROUP_LIST == c.children
def test_User(client_fixture): def test_User(client_fixture):
@ -165,6 +294,20 @@ def test_User(client_fixture):
assert c.kind == "uid" assert c.kind == "uid"
assert c.name == "conte_mascetti" assert c.name == "conte_mascetti"
assert c.dn == "uid=conte_mascetti,ou=Hackers,{}".format(BASE_DN) assert c.dn == "uid=conte_mascetti,ou=Hackers,{}".format(BASE_DN)
assert repr(c) == f"<User({c.name}) {c.dn}>"
def test_User_singleton(client_fixture):
other_client = MockClient(BASE_DN)
c1 = User("conte_mascetti", client_fixture)
c2 = User("perozzi", client_fixture)
c3 = User("conte_mascetti", client_fixture)
c4 = User("conte_mascetti", other_client)
assert client_fixture is not other_client
assert c1 is c3
assert c2 is not c1
assert c4 is not c1
def test_Service(client_fixture): def test_Service(client_fixture):
@ -173,3 +316,39 @@ def test_Service(client_fixture):
assert c.kind == "uid" assert c.kind == "uid"
assert c.name == "phi" assert c.name == "phi"
assert c.dn == "uid=phi,ou=Services,{}".format(BASE_DN) assert c.dn == "uid=phi,ou=Services,{}".format(BASE_DN)
assert repr(c) == f"<Service({c.name}) {c.dn}>"
def test_Service_singleton(client_fixture):
other_client = MockClient(BASE_DN)
c1 = Service("phi", client_fixture)
c2 = Service("irc", client_fixture)
c3 = Service("phi", client_fixture)
c4 = Service("phi", other_client)
assert client_fixture is not other_client
assert c1 is c3
assert c2 is not c1
assert c4 is not c1
def test_Group(client_fixture):
c = Group("amici_miei", client_fixture)
assert c.kind == "cn"
assert c.name == "amici_miei"
assert c.dn == "cn=amici_miei,ou=Groups,{}".format(BASE_DN)
assert repr(c) == f"<Group({c.name}) {c.dn}>"
def test_Group_singleton(client_fixture):
other_client = MockClient(BASE_DN)
c1 = Group("amici_miei", client_fixture)
c2 = Group("antani", client_fixture)
c3 = Group("amici_miei", client_fixture)
c4 = Group("amici_miei", other_client)
assert client_fixture is not other_client
assert c1 is c3
assert c2 is not c1
assert c4 is not c1