tests/test_pipeline_factory.py

from nl2sql.pipeline_factory import (
    pipeline_from_config,
    pipeline_from_config_with_adapter,
)
from adapters.db.sqlite_adapter import SQLiteAdapter
import yaml
from pathlib import Path
import os

CONFIG_PATH = os.getenv("PIPELINE_CONFIG", "configs/sqlite_pipeline.yaml")
config_path = Path(CONFIG_PATH)


def test_pipeline_from_config_builds_and_runs(tmp_path):
    p = pipeline_from_config(CONFIG_PATH)
    result = p.run(user_query="Top 3 albums by sales")
    assert result.sql is not None
    assert isinstance(result.traces, list)


def test_pipeline_from_config_with_adapter_override(tmp_path):
    with config_path.open("r") as f:
        config = yaml.safe_load(f)
    dsn = config.get("adapter", {}).get("dsn")

    adapter = SQLiteAdapter(dsn)
    p = pipeline_from_config_with_adapter(CONFIG_PATH, adapter=adapter)
    result = p.run(user_query="Count customers")
    assert "SELECT" in result.sql.upper()
    assert isinstance(result.traces, list)


def test_full_pipeline_from_yaml(monkeypatch):
    from nl2sql.pipeline_factory import pipeline_from_config

    p = pipeline_from_config(CONFIG_PATH)
    res = p.run(user_query="List all artists")
    assert res.ok
    assert isinstance(res.sql, str)
    assert any(t["stage"] == "executor" for t in res.traces)