import pytest

from preference_center.profile.utils import mask_email, mask_phone_number, mask_str


@pytest.mark.parametrize(
    "value, expected",
    [
        ("", "*"),
        ("h", "*"),
        ("he", "h*"),
        ("hello", "h****"),
        ("Hello", "H****"),
    ],
)
def test_mask_str(value: str, expected: str) -> None:
    assert mask_str(value) == expected


@pytest.mark.parametrize(
    "value, expected",
    [
        ("hello", "h****"),
        ("test@mail.com", "t***@m***.com"),
        ("test.user@mail.com", "t***.u***@m***.com"),
        ("test@london.gov.uk", "t***@l*****.gov.uk"),
        ("test.user@london.gov.uk", "t***.u***@l*****.gov.uk"),
    ],
)
def test_mask_email(value: str, expected: str) -> None:
    assert mask_email(value) == expected


@pytest.mark.parametrize(
    "value, expected",
    [
        ("", ""),
        ("h", "*"),
        ("he", "**"),
        ("hello", "*****"),
        ("Hello", "*****"),
    ],
)
def test_mask_phone_number(value: str, expected: str) -> None:
    assert mask_phone_number(value) == expected