from pathlib import Path

import pytest

import ehrql

from ehrql.query_engines.local_file import LocalFileQueryEngine

from ehrql.query_language import BaseFrame

from ehrql.tables import core

# Example CSV files are given for all core tables

EXAMPLE_TABLES = [getattr(core, table) for table in core.__all__]

EXAMPLE_DATA_DIR = Path(ehrql.__file__).parent / "example-data"

@pytest.mark.parametrize(

    "ql_table",

    EXAMPLE_TABLES,

    ids=lambda t: f"{t.__module__}.{t.__class__.__qualname__}",

)

def test_populate_database_using_example_data(ql_table: BaseFrame):

    # The engine populates the database with the example data and validates the column

    # specs in the process

    engine = LocalFileQueryEngine(EXAMPLE_DATA_DIR)

    engine.populate_database([ql_table._qm_node], allow_missing_columns=False)