from typing import Any

import pytest

SQL_QUERY_LIMIT = 5


# noinspection SqlNoDataSourceInspection
@pytest.fixture(scope="module")
def sample_payload() -> dict[str, Any]:
    """Return a sample payload for the lambda function.
    It includes only the required fields.
    """
    query = f"""
        SELECT TABLE_NAME
        FROM FACTS.INFORMATION_SCHEMA.TABLES
        LIMIT {SQL_QUERY_LIMIT};
    """
    return {"sql": query}