97 lines
2.4 KiB
Python
97 lines
2.4 KiB
Python
|
# -*- encoding: utf-8 -*-
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
from async_generator import asynccontextmanager
|
||
|
|
from bonsai import LDAPEntry
|
||
|
|
import mock
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from phi.async_ldap.new_model import get_dn, User
|
||
|
|
|
||
|
|
BASE_DN = "dc=test,dc=domain,dc=tld"
|
||
|
|
|
||
|
|
|
||
|
|
class MockClient(object):
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
self.return_value = kwargs.get("return_value")
|
||
|
|
self.connect_called = False
|
||
|
|
self.conn = mock.MagicMock()
|
||
|
|
|
||
|
|
def connect_called_with_search(self):
|
||
|
|
self.conn.search.assert_called()
|
||
|
|
|
||
|
|
def connect_called_with_add(self):
|
||
|
|
self.conn.add.assert_called()
|
||
|
|
|
||
|
|
def connect_called_with_modify(self):
|
||
|
|
self.conn.modify.assert_called()
|
||
|
|
|
||
|
|
def connect_called_with_delete(self):
|
||
|
|
self.conn.delete.assert_called()
|
||
|
|
|
||
|
|
@property
|
||
|
|
def base_dn(self):
|
||
|
|
return BASE_DN
|
||
|
|
|
||
|
|
@asynccontextmanager
|
||
|
|
async def connect(self, *args, **kwargs):
|
||
|
|
self.connect_called = True
|
||
|
|
|
||
|
|
async def _search(*a, **kw):
|
||
|
|
return self.return_value
|
||
|
|
|
||
|
|
async def _add(*a, **kw):
|
||
|
|
return self.return_value
|
||
|
|
|
||
|
|
async def _modify(*a, **kw):
|
||
|
|
return self.return_value
|
||
|
|
|
||
|
|
async def _delete(*a, **kw):
|
||
|
|
return self.return_value
|
||
|
|
|
||
|
|
self.conn.search = mock.MagicMock(side_effect=_search)
|
||
|
|
self.conn.add = mock.MagicMock(side_effect=_add)
|
||
|
|
self.conn.modify = mock.MagicMock(side_effect=_modify)
|
||
|
|
self.conn.delete = mock.MagicMock(side_effect=_delete)
|
||
|
|
|
||
|
|
yield self.conn
|
||
|
|
|
||
|
|
|
||
|
|
cl = mock.MagicMock()
|
||
|
|
cl.base_dn = BASE_DN
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"input_obj, expected_result",
|
||
|
|
[
|
||
|
|
(User(cl, "test_user"), f"uid=test_user,ou=Hackers,{BASE_DN}"),
|
||
|
|
(
|
||
|
|
LDAPEntry(f"uid=test_user,ou=Hackers,{BASE_DN}"),
|
||
|
|
f"uid=test_user,ou=Hackers,{BASE_DN}",
|
||
|
|
),
|
||
|
|
(f"uid=test_user,ou=Hackers,{BASE_DN}", f"uid=test_user,ou=Hackers,{BASE_DN}"),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_get_dn(input_obj, expected_result):
|
||
|
|
assert get_dn(input_obj) == expected_result
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_dn_raises():
|
||
|
|
with pytest.raises(ValueError) as e:
|
||
|
|
_ = get_dn(object)
|
||
|
|
|
||
|
|
assert "Unacceptable input:" in str(e.value)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_User_add():
|
||
|
|
_cl = MockClient(return_value=None)
|
||
|
|
u = User(_cl, "test_user")
|
||
|
|
|
||
|
|
assert u.dn == f"uid=test_user,ou=Hackers,{BASE_DN}"
|
||
|
|
|
||
|
|
_ = await u.save()
|
||
|
|
|
||
|
|
assert _cl.connect_called
|
||
|
|
_cl.connect_called_with_add()
|