import os
from fastapi.testclient import TestClient
from slowquery_proxy.main import app
from slowquery_proxy.config import DEV_PARQUET_URI


client = TestClient(app)


def test_dev_parquet_count():
    os.environ["APP_ENV"] = "dev"
    assert os.path.exists(DEV_PARQUET_URI)
    payload = {
        "sql": "SELECT count(*) AS c FROM slowlogs",
        "vendor": "local",
        "sources": [
            {"name": "slowlogs", "type": "parquet", "deltalake_uri": DEV_PARQUET_URI}
        ],
    }
    r = client.post("/slowquery/o11y-query", json=payload)
    assert r.status_code == 200
    data = r.json()
    assert "rows" in data and "meta" in data
    assert "columns" in data["meta"]