1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
from __future__ import annotations
from collections.abc import Generator
import pytest
from fastapi.testclient import TestClient
from sqlalchemy.orm import Session, sessionmaker
from srht_contrib.config import Settings
from srht_contrib.db import Base, make_engine
from srht_contrib.main import create_app
@pytest.fixture()
def settings() -> Settings:
return Settings(
API_KEY="test-api-key",
ENABLE_SCHEDULER=False,
SRHT_TOKEN="test-token",
DATABASE_URL="sqlite://",
TODO_SRHT_ENDPOINT="https://todo.sr.ht/query",
GIT_SRHT_ENDPOINT="https://git.sr.ht/query",
DEFAULT_ACTOR="~ccleberg",
POLL_INTERVAL_SECONDS=3600,
GIT_TRACKED_REPOSITORIES=[],
)
@pytest.fixture()
def db_engine(settings: Settings):
engine = make_engine(settings)
Base.metadata.create_all(bind=engine)
try:
yield engine
finally:
Base.metadata.drop_all(bind=engine)
@pytest.fixture()
def session_factory(db_engine) -> sessionmaker[Session]:
return sessionmaker(bind=db_engine, autoflush=False, autocommit=False, expire_on_commit=False)
@pytest.fixture()
def db_session(session_factory: sessionmaker[Session]) -> Generator[Session, None, None]:
session = session_factory()
try:
yield session
finally:
session.close()
@pytest.fixture()
def client(settings: Settings, db_engine, session_factory: sessionmaker[Session]) -> Generator[TestClient, None, None]:
app = create_app(settings, engine=db_engine, session_factory=session_factory)
with TestClient(app) as test_client:
test_client.headers.update({"X-API-Key": settings.api_key})
yield test_client
|
