Fixup tests

Signed-off-by: GitHub <[email protected]>

Author: shenanigansd

tests/conftest.py

@@ -0,0 +1,18 @@
+"""Test fixtures for the test suite."""
+
+from collections.abc import Iterator
+
+import pytest
+from litestar import Litestar
+from litestar.testing import TestClient
+
+from api.server import app
+
+app.debug = True
+
+
+@pytest.fixture
+def test_client() -> Iterator[TestClient[Litestar]]:
+    """Create a test client for the Litestar app."""
+    with TestClient(app=app) as client:
+        yield client

tests/modules/test_generators.py

@@ -3,24 +3,21 @@
 import uuid
 from http import HTTPStatus
 
-from fastapi.testclient import TestClient
+from litestar import Litestar
+from litestar.testing import TestClient
 
-from api.server import app
 
-client = TestClient(app)
-
-
-def test_bulk_uuids() -> None:
+def test_bulk_uuids(test_client: TestClient[Litestar]) -> None:
     """Test bulk UUIDs."""
-    response = client.post("/generators/uuids/", json={"uuid_type": 1, "quantity": 1})
+    response = test_client.post("/generators/uuids/", json={"uuid_type": 1, "quantity": 1})
     assert response.status_code == HTTPStatus.OK
     first = response.json()["uuids"][0]
     assert isinstance(uuid.UUID(first), uuid.UUID)
 
 
-def test_random_numbers() -> None:
+def test_random_numbers(test_client: TestClient[Litestar]) -> None:
     """Test random numbers."""
-    response = client.post("/generators/random-numbers/", json={})
+    response = test_client.post("/generators/random-numbers/", json={})
     assert response.status_code == HTTPStatus.OK
     assert response.json() == {
         "numbers": [1],

tests/test_server.py

@@ -2,17 +2,15 @@
 
 from http import HTTPStatus
 
-from fastapi.testclient import TestClient
+from litestar import Litestar
+from litestar.testing import TestClient
 
 from api import __version__
-from api.server import app
 
-client = TestClient(app)
 
-
-def test_read_main() -> None:
+def test_get_metadata(test_client: TestClient[Litestar]) -> None:
     """Test getting the root route."""
-    response = client.get("/")
+    response = test_client.get("/")
     assert response.status_code == HTTPStatus.OK
     assert response.json() == {
         "version": __version__,