import base64 import unittest from app import reconcile_shares as rs ROOT_DN = "CN=FS_Data,OU=Groups,DC=example,DC=com" GROUP_A_DN = "CN=GroupA,OU=Groups,DC=example,DC=com" GROUP_B_DN = "CN=GroupB,OU=Groups,DC=example,DC=com" GROUP_C_DN = "CN=GroupC,OU=Groups,DC=example,DC=com" USER_1_DN = "CN=Alice,OU=Users,DC=example,DC=com" USER_2_DN = "CN=Bob,OU=Users,DC=example,DC=com" USER_3_DN = "CN=Carol,OU=Users,DC=example,DC=com" COMPUTER_DN = "CN=PC01,OU=Computers,DC=example,DC=com" def principal(dn, sam, classes, members=()): return { "dn": dn, "objectGUID": "", "samAccountName": sam, "objectClasses": set(classes), "memberDns": list(members), } class GroupFolderNameTests(unittest.TestCase): def test_sanitizer_preserves_leading_dot(self): self.assertEqual(rs.sanitize_group_folder_name(".Finance"), ".Finance") self.assertEqual(rs.sanitize_group_folder_name("..Finance"), "..Finance") def test_sanitizer_still_rejects_dot_only_names(self): self.assertEqual(rs.sanitize_group_folder_name("."), "") self.assertEqual(rs.sanitize_group_folder_name(".."), "") self.assertEqual(rs.sanitize_group_folder_name("Finance."), "Finance") class LdapParsingTests(unittest.TestCase): def test_parser_keeps_repeated_members_and_unfolds_lines(self): display_name = base64.b64encode(b"Data Folder").decode("ascii") output = f""" dn: {ROOT_DN} objectGUID: 550e8400-e29b-41d4-a716-446655440000 objectClass: top objectClass: group sAMAccountName: FS_Data displayName:: {display_name} member: {USER_1_DN} member: CN=Nested,OU=Groups,DC=example, DC=com member;range=2-2: CN=Ranged,OU=Groups,DC=example,DC=com """ entries = rs.parse_ldap_entries(output) self.assertEqual(len(entries), 1) self.assertEqual(rs.ldap_first(entries[0], "displayName"), "Data Folder") self.assertEqual( rs.ldap_values(entries[0], "member"), [ USER_1_DN, "CN=Nested,OU=Groups,DC=example,DC=com", "CN=Ranged,OU=Groups,DC=example,DC=com", ], ) self.assertEqual(rs.ldap_values(entries[0], "objectClass"), ["top", "group"]) def test_parse_groups_includes_dn_members_and_classes(self): output = f""" dn: {ROOT_DN} objectGUID: 550e8400-e29b-41d4-a716-446655440000 objectClass: group sAMAccountName: FS_Data displayName: Data Folder distinguishedName: {ROOT_DN} member: {GROUP_A_DN} """ groups = rs.parse_groups_from_ldap_output(output) self.assertEqual(len(groups), 1) self.assertEqual(groups[0]["samAccountName"], "FS_Data") self.assertEqual(groups[0]["shareName"], "Data Folder") self.assertEqual(groups[0]["distinguishedName"], ROOT_DN) self.assertEqual(groups[0]["memberDns"], [GROUP_A_DN]) self.assertEqual(groups[0]["objectClasses"], {"group"}) def test_distinguished_name_filter_escapes_rfc4515_specials(self): dn = r"CN=A*B(C)\\Name,DC=example,DC=com" self.assertEqual( rs.build_distinguished_name_filter([dn]), r"(distinguishedName=CN=A\2aB\28C\29\5c\5cName,DC=example,DC=com)", ) class MembershipExpansionTests(unittest.TestCase): def lookup_from(self, principals): indexed = {rs.normalize_dn(value["dn"]): value for value in principals} def lookup(dns): return { rs.normalize_dn(dn): indexed[rs.normalize_dn(dn)] for dn in dns if rs.normalize_dn(dn) in indexed } return lookup def test_recursive_expansion_dedupes_groups_users_and_detects_cycle(self): root_group = { "objectGUID": "root-guid", "samAccountName": "FS_Data", "distinguishedName": ROOT_DN, "memberDns": [USER_1_DN, GROUP_A_DN, GROUP_B_DN, COMPUTER_DN], "objectClasses": {"group"}, } lookup = self.lookup_from( [ principal(USER_1_DN, "alice", {"user"}), principal(USER_2_DN, "bob", {"user"}), principal(USER_3_DN, "carol", {"user"}), principal(COMPUTER_DN, "PC01$", {"user", "computer"}), principal(GROUP_A_DN, "GroupA", {"group"}, [USER_2_DN, GROUP_B_DN]), principal(GROUP_B_DN, "GroupB", {"group"}, [GROUP_A_DN, USER_3_DN]), ] ) expansion = rs.expand_group_membership(root_group, lookup_func=lookup) self.assertEqual(expansion.group_sams, ["GroupA", "GroupB"]) self.assertEqual(expansion.terminal_user_count, 3) self.assertEqual(expansion.unresolved_dns, []) self.assertEqual(expansion.ignored_dns, [COMPUTER_DN]) self.assertEqual(expansion.cycle_paths, [["FS_Data", "GroupA", "GroupB", "GroupA"]]) def test_self_cycle_is_reported_once(self): root_group = { "objectGUID": "root-guid", "samAccountName": "FS_Data", "distinguishedName": ROOT_DN, "memberDns": [GROUP_C_DN], "objectClasses": {"group"}, } lookup = self.lookup_from( [principal(GROUP_C_DN, "GroupC", {"group"}, [GROUP_C_DN])] ) expansion = rs.expand_group_membership(root_group, lookup_func=lookup) self.assertEqual(expansion.group_sams, ["GroupC"]) self.assertEqual(expansion.cycle_paths, [["FS_Data", "GroupC", "GroupC"]]) def test_unresolved_member_is_reported(self): missing_dn = "CN=Missing,OU=Groups,DC=example,DC=com" root_group = { "objectGUID": "root-guid", "samAccountName": "FS_Data", "distinguishedName": ROOT_DN, "memberDns": [missing_dn], "objectClasses": {"group"}, } expansion = rs.expand_group_membership(root_group, lookup_func=lambda dns: {}) self.assertEqual(expansion.unresolved_dns, [missing_dn]) self.assertEqual(expansion.group_sams, []) class AclResolutionTests(unittest.TestCase): def test_gid_resolution_preserves_order_dedupes_and_reports_missing(self): mapping = {"FS_Data": 1001, "Nested": 1002} def resolver(workgroup, group_name): self.assertEqual(workgroup, "EXAMPLE") return mapping.get(group_name) gids, unresolved = rs.resolve_group_gids_for_acl( "EXAMPLE", ["FS_Data", "Nested", "nested", "Missing"], resolver ) self.assertEqual(gids, [1001, 1002]) self.assertEqual(unresolved, ["Missing"]) if __name__ == "__main__": unittest.main()