99 lines
3.3 KiB
Python
99 lines
3.3 KiB
Python
|
import sys
|
||
|
|
import unittest
|
||
|
|
from unittest.mock import MagicMock, patch
|
||
|
|
|
||
|
|
import ldapsm.__main__ as manage_schema
|
||
|
|
|
||
|
|
|
||
|
|
class TestManageSchema(unittest.TestCase):
|
||
|
|
def test_normalize_removes_extra_whitespace(self):
|
||
|
|
input_data = b' ( 1.2.3.4 NAME \'foo\' \n DESC \t "bar" ) '
|
||
|
|
expected = b'( 1.2.3.4 NAME \'foo\' DESC "bar" )'
|
||
|
|
self.assertEqual(manage_schema.normalize(input_data), expected)
|
||
|
|
|
||
|
|
def test_extract_oid_from_definition(self):
|
||
|
|
ldif = "( 1.3.6.1.4.1.99999.1 NAME 'example' DESC 'something' )"
|
||
|
|
oid = manage_schema.extract_oid(ldif)
|
||
|
|
self.assertEqual(oid, "1.3.6.1.4.1.99999.1")
|
||
|
|
|
||
|
|
@patch("ldapsm.__main__.ldap.initialize")
|
||
|
|
def test_add_new_object_class_no_double_add(self, mock_ldap_init):
|
||
|
|
mock_conn = MagicMock()
|
||
|
|
mock_ldap_init.return_value = mock_conn
|
||
|
|
|
||
|
|
schema_name = "nextcloud"
|
||
|
|
ocdef = "( 1.3.6.1.4.1.99999.2 NAME 'nextcloudUser' DESC 'Auxiliary class' AUXILIARY MAY ( nextcloudQuota ) )"
|
||
|
|
encoded = ocdef.encode()
|
||
|
|
|
||
|
|
mock_conn.search_s.side_effect = [
|
||
|
|
[(f"cn={{0}}{schema_name},cn=schema,cn=config", {"dn": [b"dummy"]})],
|
||
|
|
[(f"cn={{0}}{schema_name},cn=schema,cn=config", {"olcObjectClasses": [encoded]})],
|
||
|
|
]
|
||
|
|
|
||
|
|
with patch.object(
|
||
|
|
sys,
|
||
|
|
"argv",
|
||
|
|
[
|
||
|
|
"ldapsm",
|
||
|
|
"-s",
|
||
|
|
"ldap://localhost",
|
||
|
|
"-D",
|
||
|
|
"cn=admin,dc=example,dc=org",
|
||
|
|
"-W",
|
||
|
|
"secret",
|
||
|
|
"-n",
|
||
|
|
schema_name,
|
||
|
|
"-c",
|
||
|
|
ocdef,
|
||
|
|
],
|
||
|
|
):
|
||
|
|
manage_schema.main()
|
||
|
|
|
||
|
|
mock_conn.modify_s.assert_not_called()
|
||
|
|
|
||
|
|
@patch("ldapsm.__main__.ldap.initialize")
|
||
|
|
def test_replace_attribute_type_same_oid(self, mock_ldap_init):
|
||
|
|
mock_conn = MagicMock()
|
||
|
|
mock_ldap_init.return_value = mock_conn
|
||
|
|
|
||
|
|
schema_name = "nextcloud"
|
||
|
|
atdef = "( 1.3.6.1.4.1.99999.1 NAME 'nextcloudQuota' DESC 'Quota for Nextcloud' SYNTAX 1.3.6.1.4.1.1466.115.121.1.27 SINGLE-VALUE )"
|
||
|
|
encoded = atdef.encode()
|
||
|
|
|
||
|
|
existing_attr_old = b"( 1.3.6.1.4.1.99999.1 NAME 'nextcloudQuota' DESC 'Old desc' SYNTAX 1.3.6.1.4.1.1466.115.121.1.27 )"
|
||
|
|
|
||
|
|
mock_conn.search_s.side_effect = [
|
||
|
|
[(f"cn={{0}}{schema_name},cn=schema,cn=config", {"dn": [b"dummy"]})],
|
||
|
|
[(f"cn={{0}}{schema_name},cn=schema,cn=config", {"olcAttributeTypes": [existing_attr_old]})],
|
||
|
|
]
|
||
|
|
|
||
|
|
with patch.object(
|
||
|
|
sys,
|
||
|
|
"argv",
|
||
|
|
[
|
||
|
|
"ldapsm",
|
||
|
|
"-s",
|
||
|
|
"ldap://localhost",
|
||
|
|
"-D",
|
||
|
|
"cn=admin,dc=example,dc=org",
|
||
|
|
"-W",
|
||
|
|
"secret",
|
||
|
|
"-n",
|
||
|
|
schema_name,
|
||
|
|
"-a",
|
||
|
|
atdef,
|
||
|
|
],
|
||
|
|
):
|
||
|
|
manage_schema.main()
|
||
|
|
|
||
|
|
mock_conn.modify_s.assert_called_once()
|
||
|
|
args, _kwargs = mock_conn.modify_s.call_args
|
||
|
|
self.assertIn("cn={0}nextcloud,cn=schema,cn=config", args[0])
|
||
|
|
self.assertEqual(args[1][0][0], manage_schema.ldap.MOD_REPLACE)
|
||
|
|
self.assertEqual(args[1][0][1], "olcAttributeTypes")
|
||
|
|
self.assertEqual(args[1][0][2], [encoded])
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|