132 lines
2.6 KiB
Python
132 lines
2.6 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
from async_generator import asynccontextmanager
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from phi.async_ldap.new_model import (
|
||
|
|
Hackers,
|
||
|
|
User,
|
||
|
|
Robots,
|
||
|
|
Service,
|
||
|
|
Group,
|
||
|
|
Congregations,
|
||
|
|
)
|
||
|
|
from phi.async_ldap.mixins import build_heritage
|
||
|
|
from phi.async_ldap.client import AsyncClient
|
||
|
|
import phi.exceptions as e
|
||
|
|
|
||
|
|
BASE_DN = "dc=unit,dc=macaomilano,dc=org"
|
||
|
|
|
||
|
|
cl = AsyncClient(
|
||
|
|
"ldap://localhost",
|
||
|
|
port=389,
|
||
|
|
encryption=True,
|
||
|
|
# validate=True,
|
||
|
|
ca_cert="../openldap/cert.pem",
|
||
|
|
username="root",
|
||
|
|
password="root",
|
||
|
|
base_dn=BASE_DN,
|
||
|
|
attribute_id="cn",
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@asynccontextmanager
|
||
|
|
async def clean_db():
|
||
|
|
h = Hackers(cl)
|
||
|
|
r = Robots(cl)
|
||
|
|
c = Congregations(cl)
|
||
|
|
h.delete_cascade = True
|
||
|
|
r.delete_cascade = True
|
||
|
|
c.delete_cascade = True
|
||
|
|
await h.delete()
|
||
|
|
await r.delete()
|
||
|
|
await c.delete()
|
||
|
|
yield
|
||
|
|
await h.delete()
|
||
|
|
await r.delete()
|
||
|
|
await c.delete()
|
||
|
|
|
||
|
|
|
||
|
|
async def init_achilles():
|
||
|
|
u = User(cl, "achilles")
|
||
|
|
u["cn"] = "Achilles"
|
||
|
|
u["sn"] = "achilles"
|
||
|
|
u["mail"] = "achilles@phthia.gr"
|
||
|
|
u["userPassword"] = "Patroclus123"
|
||
|
|
|
||
|
|
await u.save()
|
||
|
|
|
||
|
|
return u
|
||
|
|
|
||
|
|
|
||
|
|
async def init_group(group_name, members):
|
||
|
|
g = Group(cl, group_name, member=members)
|
||
|
|
|
||
|
|
await g.save()
|
||
|
|
|
||
|
|
return g
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
@pytest.mark.integration_test
|
||
|
|
async def test_User_init():
|
||
|
|
async with clean_db():
|
||
|
|
u = await init_achilles()
|
||
|
|
|
||
|
|
h = Hackers(cl)
|
||
|
|
|
||
|
|
res = await h.search("achilles")
|
||
|
|
|
||
|
|
assert u == res
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
@pytest.mark.integration_test
|
||
|
|
async def test_User_modify():
|
||
|
|
async with clean_db():
|
||
|
|
u = await init_achilles()
|
||
|
|
NEW_EMAIL = "a@myrmidons.mil"
|
||
|
|
u["mail"] = NEW_EMAIL
|
||
|
|
await u.modify()
|
||
|
|
|
||
|
|
h = Hackers(cl)
|
||
|
|
|
||
|
|
res = await h.search("achilles")
|
||
|
|
await u.sync()
|
||
|
|
|
||
|
|
assert u["mail"] == res["mail"] == NEW_EMAIL
|
||
|
|
for attr in u.ldap_attributes:
|
||
|
|
assert u[attr] == res[attr]
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
@pytest.mark.integration_test
|
||
|
|
async def test_User_delete():
|
||
|
|
async with clean_db():
|
||
|
|
u = await init_achilles()
|
||
|
|
await u.delete()
|
||
|
|
|
||
|
|
h = Hackers(cl)
|
||
|
|
|
||
|
|
with pytest.raises(e.PhiEntryDoesNotExist) as ex:
|
||
|
|
await h.search("achilles")
|
||
|
|
|
||
|
|
assert u.dn in str(ex.value)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
@pytest.mark.integration_test
|
||
|
|
async def test_Group_init():
|
||
|
|
async with clean_db():
|
||
|
|
u = await init_achilles()
|
||
|
|
g = await init_group("achaeans", [u])
|
||
|
|
|
||
|
|
c = Congregations(cl)
|
||
|
|
|
||
|
|
res = await c.search("achaeans")
|
||
|
|
|
||
|
|
assert g == res
|
||
|
|
assert [u] == [a for a in g.get_members()]
|