chore: make cmc / cg api tests context managed

balancednetwork · Oct 29, 2024 · dc06131 · dc06131
1 parent 8c73b13
commit dc06131
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/tests/integration/api/test_api_cmc.py b/tests/integration/api/test_api_cmc.py
@@ -4,6 +4,13 @@
 from balanced_backend.config import settings
 
 
+@pytest.fixture
+def client(event_loop) -> TestClient:
+    from balanced_backend.main_api import app
+    with TestClient(app) as client:
+        yield client
+
+
 @pytest.mark.anyio
 def test_api_get_cmc_summary(client: TestClient):
     response = client.get(f"{settings.REST_PREFIX}/coin-market-cap/summary")

diff --git a/tests/integration/api/test_api_coingecko.py b/tests/integration/api/test_api_coingecko.py
@@ -3,6 +3,12 @@
 
 from balanced_backend.config import settings
 
+@pytest.fixture
+def client(event_loop) -> TestClient:
+    from balanced_backend.main_api import app
+    with TestClient(app) as client:
+        yield client
+
 
 @pytest.mark.anyio
 def test_api_get_coingecko_summary(client: TestClient):