import enum
from dataclasses import dataclass
from typing import List
from typing import Optional
from typing import Union

import pytest

from codemagic.apple.app_store_connect.resource_manager import ResourceManager

StubEnum = enum.Enum("StubEnum", {"A": "a", "B": "b"})


class CustomString(str): ...


@dataclass
class StubFilter(ResourceManager.Filter):
    field_one: Optional[str] = None
    field_two: Optional[StubEnum] = None
    field_three_four: Optional[CustomString] = None
    maybe_list: Optional[Union[StubEnum, List[StubEnum]]] = None


@pytest.mark.parametrize(
    "filter_params, expected_query_params",
    [
        ({"field_one": "1"}, {"filter[fieldOne]": "1"}),
        ({"field_two": StubEnum.A}, {"filter[fieldTwo]": "a"}),
        ({"field_three_four": CustomString("34")}, {"filter[fieldThreeFour]": "34"}),
        ({"field_one": "1", "field_two": None}, {"filter[fieldOne]": "1"}),
        ({"field_two": StubEnum.B, "field_one": None}, {"filter[fieldTwo]": "b"}),
        (
            {"field_three_four": CustomString("34"), "field_one": None, "field_two": None},
            {"filter[fieldThreeFour]": "34"},
        ),
        ({"field_one": "1", "field_two": StubEnum.A}, {"filter[fieldOne]": "1", "filter[fieldTwo]": "a"}),
        ({"maybe_list": [StubEnum.A, StubEnum.B]}, {"filter[maybeList]": "a,b"}),
        ({"maybe_list": [StubEnum.A]}, {"filter[maybeList]": "a"}),
        ({"maybe_list": [StubEnum.A, StubEnum.B]}, {"filter[maybeList]": "a,b"}),
    ],
)
def test_resource_manager_filter_to_params_conversion(filter_params, expected_query_params):
    test_filter = StubFilter(**filter_params)
    assert test_filter.as_query_params() == expected_query_params