"""Test brand utils file."""

import pytest

from permissions.constants import constants
from permissions.utils.brand_utils import default_brand_from_brand


@pytest.mark.parametrize(
    ('brand', 'expected'),
    [
        (constants.THEORCHARD_BRAND, constants.THEORCHARD_BRAND),
        (constants.SONY_BRAND, constants.SONY_BRAND),
        (constants.AWAL_BRAND, constants.AWAL_BRAND),
        (constants.KNR_BRAND, constants.KNR_BRAND),
        (constants.MASS_APPEAL_BRAND, constants.THEORCHARD_BRAND),
        (constants.DRM_BRAND, constants.THEORCHARD_BRAND),
        (constants.FOUNDATION_BRAND, constants.SONY_BRAND),
        ('', constants.THEORCHARD_BRAND),
    ],
)
def test_default_brand_from_brand(brand, expected):
    """Test default_brand_from_brand converts brands to auth0 org."""
    actual = default_brand_from_brand(brand)
    assert actual == expected