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

import pytest

from src.datahub.ingestion.source.ldap import parse_from_attrs, strip_ldap_info


def test_strip_ldap_info():
    assert (
        strip_ldap_info(b"uid=firstname.surname,ou=People,dc=internal,dc=machines")
        == "firstname.surname"
    )


@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_from_attrs(input, expected):
    assert (
        parse_from_attrs(
            input,
            "admins",
        )
        == expected
    )