datahub/metadata-ingestion/tests/unit/test_ldap_source.py

import pytest

from datahub.ingestion.source.ldap import parse_groups, parse_ldap_dn, parse_users


@pytest.mark.parametrize(
    "input, expected",
    [
        (
            b"uid=firstname.surname,ou=People,dc=internal,dc=machines",
            "firstname.surname",
        ),
        (
            b"cn=group_name,ou=Groups,dc=internal,dc=machines",
            "group_name",
        ),
        (
            b"cn=comma group (one\\, two\\, three),ou=Groups,dc=internal,dc=machines",
            "comma group (one, two, three)",
        ),
    ],
)
def test_parse_ldap_dn(input, expected):
    assert parse_ldap_dn(input) == expected


@pytest.mark.parametrize(
    "input, expected",
    [
        (
            {
                "admins": [
                    b"uid=A.B,ou=People,dc=internal,dc=machines",
                    b"uid=C.D,ou=People,dc=internal,dc=machines",
                ]
            },
            ["urn:li:corpuser:A.B", "urn:li:corpuser:C.D"],
        ),
        (
            {
                "not_admins": [
                    b"doesntmatter",
                ]
            },
            [],
        ),
    ],
)
def test_parse_users(input, expected):
    assert (
        parse_users(
            input,
            "admins",
        )
        == expected
    )


@pytest.mark.parametrize(
    "input, expected",
    [
        (
            {
                "memberOf": [
                    b"cn=group1,ou=Groups,dc=internal,dc=machines",
                    b"cn=group2,ou=Groups,dc=internal,dc=machines",
                ]
            },
            ["urn:li:corpGroup:group1", "urn:li:corpGroup:group2"],
        ),
        (
            {
                "not_member": [
                    b"doesntmatter",
                ]
            },
            [],
        ),
    ],
)
def test_parse_groups(input, expected):
    assert (
        parse_groups(
            input,
            "memberOf",
        )
        == expected
    )