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
|
from __future__ import annotations
import pytest
from _pytest.monkeypatch import MonkeyPatch
from litestar.testing import TestClient
pytestmark = pytest.mark.xdist_group("sqlalchemy_examples")
def test_sync_app(monkeypatch: MonkeyPatch) -> None:
from docs.examples.plugins.sqlalchemy_init_plugin import sqlalchemy_sync
monkeypatch.setattr(sqlalchemy_sync.sqlalchemy_config, "connection_string", "sqlite://")
with TestClient(app=sqlalchemy_sync.app) as client:
res = client.get("/sqlalchemy-app")
assert res.status_code == 200
assert res.text == "1 2"
def test_async_app(monkeypatch: MonkeyPatch) -> None:
from docs.examples.plugins.sqlalchemy_init_plugin import sqlalchemy_async
monkeypatch.setattr(sqlalchemy_async.sqlalchemy_config, "connection_string", "sqlite+aiosqlite://")
with TestClient(app=sqlalchemy_async.app) as client:
res = client.get("/sqlalchemy-app")
assert res.status_code == 200
assert res.text == "1 2"
|