"""Test file for common tracing helpers"""

from typing import Any, Dict, List

import pytest

from concrete.common.tracing.tracing_helpers import prepare_function_parameters


@pytest.mark.parametrize(
    "function,function_parameters,ref_dict",
    [
        pytest.param(lambda x: None, {}, {}, id="Missing x", marks=pytest.mark.xfail(strict=True)),
        pytest.param(lambda x: None, {"x": None}, {"x": None}, id="Only x"),
        pytest.param(
            lambda x: None, {"x": None, "y": None}, {"x": None}, id="Additional y filtered"
        ),
    ],
)
def test_prepare_function_parameters(
    function, function_parameters: Dict[str, Any], ref_dict: Dict[str, Any]
):
    """Test prepare_function_parameters"""
    prepared_dict = prepare_function_parameters(function, function_parameters)

    assert prepared_dict == ref_dict


@pytest.mark.parametrize(
    "function,function_parameters,expected_ordered_keys",
    [
        (lambda x: None, {"x": None}, ["x"]),
        (lambda x, y: None, {"x": None, "y": None}, ["x", "y"]),
        (lambda x, y: None, {"y": None, "x": None}, ["x", "y"]),
        (lambda z, x, y: None, {"y": None, "z": None, "x": None}, ["z", "x", "y"]),
    ],
)
def test_prepare_function_parameters_order(
    function, function_parameters: Dict[str, Any], expected_ordered_keys: List[str]
):
    """Test prepare_function_parameters output order"""
    prepared_dict = prepare_function_parameters(function, function_parameters)

    assert list(prepared_dict.keys()) == expected_ordered_keys