import pytest
from RunPollingCommand import prepare_arg_dict

IDS_ARGS = [
    # sanity
    (
        ("ids", ["a", "b", "c"], None, None, "SomeIntegration"),
        {"ids": "a,b,c", "using": "SomeIntegration"},
    ),
    # single ID
    (
        ("ids", "a", None, None),
        {"ids": "a"},
    ),
    # numeric IDs
    (
        ("ids", [1, 2, 3], None, None),
        {"ids": "1,2,3"},
    ),
    # extra args
    (
        ("ids", ["a", "b", "c"], "arg1", "value1"),
        {"ids": "a,b,c", "arg1": "value1"},
    ),
    # extra args as unicode lists
    (
        ("ids", ["a", "b", "c"], ["arg1", "arg2"], ["value1", "value2"]),
        {"ids": "a,b,c", "arg1": "value1", "arg2": "value2"},
    ),
    # extra args as chane of unicode
    (
        ("ids", ["a", "b", "c"], "arg1, arg2,arg3", ["value1", "value2", "value3"]),
        {"ids": "a,b,c", "arg1": "value1", "arg2": "value2", "arg3": "value3"},
    ),
    # extra args as string
    (
        ("ids", ["שלום"], "היי, arg2,arg3", ["ביי", "value2", "value3"]),
        {"ids": "שלום", "היי": "ביי", "arg2": "value2", "arg3": "value3"},
    ),
    # extra args as string with int values
    (
        ("ids", ["a", "b", "c"], "arg1,arg2", [1, 2]),
        {"ids": "a,b,c", "arg1": "1", "arg2": "2"},
    ),
]


@pytest.mark.parametrize("inputs, expected_args", IDS_ARGS)
def test_prepare_arg_dict(inputs, expected_args):
    args = prepare_arg_dict(*inputs)
    assert args == expected_args


def test_prepare_arg_dict__error():
    with pytest.raises(ValueError):
        prepare_arg_dict("ids", "a", "arg1", None)