64 lines
2 KiB
Python
64 lines
2 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import shutil
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from alembic import command
|
||
|
|
from alembic.config import Config
|
||
|
|
from fastapi.testclient import TestClient
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
|
||
|
|
from canned_prompts_service.api import create_app
|
||
|
|
from canned_prompts_service.db import make_engine, make_session_factory
|
||
|
|
from canned_prompts_service.settings import Settings
|
||
|
|
from canned_prompts_service import store
|
||
|
|
|
||
|
|
ROOT = Path(__file__).resolve().parents[1]
|
||
|
|
REPO = ROOT.parent
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def db_url(tmp_path: Path) -> str:
|
||
|
|
url = f"sqlite:///{tmp_path / 'svc.db'}"
|
||
|
|
config = Config(str(ROOT / "alembic.ini"))
|
||
|
|
config.set_main_option("script_location", str(ROOT / "migrations"))
|
||
|
|
config.set_main_option("sqlalchemy.url", url)
|
||
|
|
command.upgrade(config, "head")
|
||
|
|
return url
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def session(db_url: str) -> Session:
|
||
|
|
with make_session_factory(make_engine(db_url))() as s:
|
||
|
|
yield s
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def client(db_url: str) -> TestClient:
|
||
|
|
return TestClient(create_app(Settings(database_url=db_url), make_engine(db_url)))
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def seeded(db_url: str) -> TestClient:
|
||
|
|
"""A store holding the repo's own example packages."""
|
||
|
|
engine = make_engine(db_url)
|
||
|
|
with make_session_factory(engine)() as s:
|
||
|
|
for name in ("house-style", "pqrst-estimate"):
|
||
|
|
store.ingest(s, "default", "local", REPO / "examples" / name,
|
||
|
|
source=f"examples/{name}", method="publish")
|
||
|
|
s.commit()
|
||
|
|
return TestClient(create_app(Settings(database_url=db_url), engine))
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def two_registries(db_url: str, tmp_path: Path) -> TestClient:
|
||
|
|
"""The same id in two registries — § 3.2 ambiguity."""
|
||
|
|
engine = make_engine(db_url)
|
||
|
|
with make_session_factory(engine)() as s:
|
||
|
|
for registry in ("local", "house"):
|
||
|
|
store.ingest(s, "default", registry, REPO / "examples" / "house-style",
|
||
|
|
source=registry, method="publish")
|
||
|
|
s.commit()
|
||
|
|
return TestClient(create_app(Settings(database_url=db_url), engine))
|