"""Test for Country model."""

import pytest

from carveouts.models import country


@pytest.mark.parametrize(
    "country_codes, expected_result",
    [
        (["US", "GB"], [1, 3]),
        (["XX", "YY"], []),
        ([], []),
        (["US", "XX", "GB"], [1, 3]),
    ],
)
def test_for_valid_country_code(
    country_codes: list[str], expected_result: list[int], db_fixture: None
) -> None:
    """Test to check valid country code."""
    result = country.get_country_ids(country_codes)

    assert result == expected_result