"""Test include functions."""

import pytest

from src.logic.utils import (
    get_project_id_from_gql_response,
    get_project_id_from_record,
)


@pytest.mark.parametrize(
    "test_input,expected",
    [
        ({}, {}),
        ({"project": None}, {}),
        ({"project": {"ids": None}}, {}),
        ({"project": {"ids": []}}, {}),
        ({"project": {"ids": [{"localId": None}]}}, {}),
        ({"project": {"ids": [{"localId": "12345"}]}}, {"projectLocalId": "12345"}),  # noqa
    ],
)
def test_include_project_id(test_input, expected):
    """Test include_project_id function."""
    result = get_project_id_from_gql_response(test_input)
    assert result == expected


@pytest.mark.parametrize(
    "test_input,expected",
    [
        ({}, None),
        ({"project_id": None}, None),
        ({"project_id": "12345"}, {"projectLocalId": "12345"}),
    ],
)
def test_release_table_update_includes_handler(test_input, expected):
    """Test get_project_id_from_record function."""
    assert get_project_id_from_record(test_input) == expected