172 lines
6.5 KiB
Python
172 lines
6.5 KiB
Python
"""Unit tests for the lumi2.ldap module."""
|
|
|
|
import pytest
|
|
|
|
import lumi2.ldap as ll
|
|
import lumi2.exceptions as le
|
|
|
|
|
|
def test_assert_is_valid_base_dn():
|
|
for invalid_type in [None, 0, True, ("x",), ["x"]]:
|
|
with pytest.raises(TypeError):
|
|
ll._assert_is_valid_base_dn(invalid_type)
|
|
|
|
for invalid_string in [
|
|
"", " ", "ou=foo", "abc", "\u1337",
|
|
"a,b", "dc=,dc=", "dc=a,dc=b", "ou=example,dc=com", "dc=example,dc="
|
|
"dc=,dc=com", "dc=EXAMPLE,dc=com", "dc=example,dc=COM",
|
|
"dc=example$,dc=com", "dc=foo,dc=com$", "dc=exa%,dc=com%",
|
|
"dc=ex-ample,dc=com", "dc=f_oo,dc=com",
|
|
]:
|
|
with pytest.raises(ll.InvalidStringFormatException):
|
|
ll._assert_is_valid_base_dn(invalid_string)
|
|
|
|
for valid_string in ["dc=example,dc=com", "dc=a,dc=de"]:
|
|
assert ll._assert_is_valid_base_dn(valid_string) == None
|
|
|
|
|
|
def test_assert_is_valid_ou_dn():
|
|
for invalid_type in [None, 0, True, ("x",), ["x"]]:
|
|
with pytest.raises(TypeError):
|
|
ll._assert_is_valid_ou_dn(invalid_type)
|
|
|
|
for invalid_string in [
|
|
"", " ", "ou=foo", "abc", "\u1337",
|
|
"ou=,dc=example,dc=com", "ou=foo,dc=example,dc=", "ou=foo,dc=example,",
|
|
"ou=foo,dc=example", "ou=foo,ou=bar,dc=example",
|
|
"ou=foo,ou=bar,dc=example,dc=com", "cn=foo,dc=example,dc=com",
|
|
"ou=abc%,dc=example,dc=com",
|
|
]:
|
|
with pytest.raises(ll.InvalidStringFormatException):
|
|
print(invalid_string)
|
|
ll._assert_is_valid_ou_dn(invalid_string)
|
|
|
|
for valid_string in [
|
|
"ou=users,dc=example,dc=com", "ou=users,dc=a,dc=de",
|
|
"ou=Users,dc=example,dc=com", "ou=abcdABCD,dc=example,dc=com",
|
|
]:
|
|
assert ll._assert_is_valid_ou_dn(valid_string) == None
|
|
|
|
|
|
def test_assert_is_valid_user_dn():
|
|
for invalid_type in [None, 0, True, ("x",), ["x"]]:
|
|
with pytest.raises(TypeError):
|
|
ll._assert_is_valid_user_dn(invalid_type)
|
|
|
|
for invalid_string in [
|
|
"", " ", "uid=foo", "abc", "\u1337",
|
|
"uid=,ou=foo,dc=bar,dc=baz", "uid=foo,ou=,dc=bar,dc=baz",
|
|
"uid=foo,ou=bar,dc=,dc=baz", "uid=foo,ou=bar,dc=baz,dc=",
|
|
"uid=foo,dc=example,dc=com", "uid=1foo,ou=bar,dc=example,dc=com",
|
|
"uid=.bob,ou=foo,dc=bar,dc=baz", "uid=_foo,ou=foo,dc=bar,dc=baz",
|
|
"uid=foo,uid=bar,ou=bar,dc=com", "cn=foo,ou=bar,dc=example,dc=com",
|
|
"uid=abc%,ou=foo,dc=example,dc=com",
|
|
]:
|
|
with pytest.raises(ll.InvalidStringFormatException):
|
|
print(invalid_string)
|
|
ll._assert_is_valid_user_dn(invalid_string)
|
|
|
|
for valid_string in [
|
|
"uid=foo,ou=users,dc=example,dc=com", "uid=Foo-bar.baz_,ou=users,dc=a,dc=de",
|
|
"uid=bOb_.-,ou=Users,dc=example,dc=com", "uid=foo,ou=foo,dc=bar,dc=baz",
|
|
]:
|
|
assert ll._assert_is_valid_user_dn(valid_string) == None
|
|
|
|
|
|
def test_assert_is_valid_group_dn():
|
|
for invalid_type in [None, 0, True, ("x",), ["x"]]:
|
|
with pytest.raises(TypeError):
|
|
ll._assert_is_valid_group_dn(invalid_type)
|
|
|
|
for invalid_string in [
|
|
"", " ", "cn=foo", "abc", "\u1337",
|
|
"cn=,ou=foo,dc=bar,dc=baz", "cn=foo,ou=,dc=bar,dc=baz",
|
|
"cn=foo,ou=bar,dc=,dc=baz", "cn=foo,ou=bar,dc=baz,dc=",
|
|
"cn=foo,dc=example,dc=com", "cn=1foo,ou=bar,dc=example,dc=com",
|
|
"cn=.bob,ou=foo,dc=bar,dc=baz", "cn=_foo,ou=foo,dc=bar,dc=baz",
|
|
"cn=foo,cn=bar,ou=bar,dc=com",
|
|
"cn=abc%,ou=foo,dc=bar,dc=baz", "cn=abc1,ou=foo,dc=bar,dc=baz",
|
|
"cn=ab-c,ou=foo,dc=bar,dc=baz",
|
|
]:
|
|
with pytest.raises(ll.InvalidStringFormatException):
|
|
print(invalid_string)
|
|
ll._assert_is_valid_group_dn(invalid_string)
|
|
|
|
for valid_string in [
|
|
"cn=foo,ou=groups,dc=example,dc=com", "cn=foo,ou=bar,dc=a,dc=de",
|
|
"cn=Abc,ou=foo,dc=bar,dc=baz",
|
|
]:
|
|
assert ll._assert_is_valid_group_dn(valid_string) == None
|
|
|
|
|
|
def test_assert_is_valid_user_object_class():
|
|
for invalid_type in [None, 0, True, ("x",), ["x"]]:
|
|
with pytest.raises(TypeError):
|
|
ll._assert_is_valid_user_object_class(invalid_type)
|
|
|
|
for invalid_string in [
|
|
"", " ", "\u1337",
|
|
"organizationalUnit", "INETORGPERSON", "inetorgperson",
|
|
]:
|
|
with pytest.raises(ll.InvalidStringFormatException):
|
|
ll._assert_is_valid_user_object_class(invalid_string)
|
|
|
|
assert ll._assert_is_valid_user_object_class("inetOrgPerson") == None
|
|
|
|
|
|
def test_assert_is_valid_group_object_class():
|
|
for invalid_type in [None, 0, True, ("x",), ["x"]]:
|
|
with pytest.raises(TypeError):
|
|
ll._assert_is_valid_group_object_class(invalid_type)
|
|
|
|
for invalid_string in [
|
|
"", " ", "\u1337",
|
|
"organizationalUnit", "inetOrgPerson",
|
|
"groupofuniquenames", "GROUPOFUNIQUENAMES",
|
|
]:
|
|
with pytest.raises(ll.InvalidStringFormatException):
|
|
ll._assert_is_valid_group_object_class(invalid_string)
|
|
|
|
assert ll._assert_is_valid_group_object_class("groupOfUniqueNames") == None
|
|
|
|
|
|
def test_assert_is_valid_connection(connection):
|
|
for invalid_type in [1, "", None, (connection,)]:
|
|
with pytest.raises(TypeError):
|
|
ll._assert_is_valid_connection(invalid_type)
|
|
|
|
assert ll._assert_is_valid_connection(connection) == None
|
|
|
|
with pytest.raises(ll.InvalidConnectionException):
|
|
connection.unbind()
|
|
ll._assert_is_valid_connection(connection)
|
|
|
|
|
|
def test_ou_exists(app, connection):
|
|
with app.app_context():
|
|
for invalid_type in [1, True, None]:
|
|
with pytest.raises(TypeError):
|
|
ll.ou_exists(connection, invalid_type)
|
|
|
|
with pytest.raises(ll.InvalidStringFormatException):
|
|
ll.ou_exists(connection, "cn=foo,dc=example,dc=com")
|
|
|
|
assert ll.ou_exists(connection, "ou=users,dc=example,dc=com")
|
|
assert ll.ou_exists(connection, "ou=groups,dc=example,dc=com")
|
|
assert not ll.ou_exists(connection, "ou=nonexsistent,dc=example,dc=com")
|
|
|
|
|
|
def test_ou_exists_unpopulated(app, connection_empty_dit):
|
|
with app.app_context():
|
|
assert not ll.ou_exists(connection_empty_dit, "ou=users,dc=example,dc=com")
|
|
assert not ll.ou_exists(connection_empty_dit, "ou=groups,dc=example,dc=com")
|
|
|
|
def test_create_ou(app, connection):
|
|
with app.app_context():
|
|
with pytest.raises(ll.EntryExistsException):
|
|
ll.create_ou(connection, "ou=users,dc=example,dc=com")
|
|
|
|
assert not ll.ou_exists(connection, "ou=foo,dc=example,dc=com")
|
|
ll.create_ou(connection, "ou=foo,dc=example,dc=com")
|
|
assert ll.ou_exists(connection, "ou=foo,dc=example,dc=com")
|