minimize change

tests/unit/server/test_quota.py

@@ -52,21 +52,17 @@ def auth_app(tmp_path, request):
     db_path = tmp_path / f"quota_{request.node.name}.db"
     quota = build_quota_config(db_path)
 
-    quota_middleware = QuotaMiddleware(
-        inner_app,
-        kv_config=quota.kvstore,
-        anonymous_max_requests=quota.anonymous_max_requests,
-        authenticated_max_requests=quota.authenticated_max_requests,
-        window_seconds=86400,
+    app = InjectClientIDMiddleware(
+        QuotaMiddleware(
+            inner_app,
+            kv_config=quota.kvstore,
+            anonymous_max_requests=quota.anonymous_max_requests,
+            authenticated_max_requests=quota.authenticated_max_requests,
+            window_seconds=86400,
+        ),
+        client_id=f"client_{request.node.name}",
     )
-    app = InjectClientIDMiddleware(quota_middleware, client_id=f"client_{request.node.name}")
-
-    yield app
-
-    # Cleanup
-    import asyncio
-
-    asyncio.run(quota_middleware.close())
+    return app
 
 
 def test_authenticated_quota_allows_up_to_limit(auth_app):
@@ -85,8 +81,6 @@ def test_authenticated_quota_blocks_after_limit(auth_app):
 
 
 def test_anonymous_quota_allows_up_to_limit(tmp_path, request):
-    import asyncio
-
     inner_app = FastAPI()
 
     @inner_app.get("/test")
@@ -107,12 +101,8 @@ def test_anonymous_quota_allows_up_to_limit(tmp_path, request):
     client = TestClient(app)
     assert client.get("/test").status_code == 200
 
-    asyncio.run(app.close())
-
 
 def test_anonymous_quota_blocks_after_limit(tmp_path, request):
-    import asyncio
-
     inner_app = FastAPI()
 
     @inner_app.get("/test")
@@ -135,5 +125,3 @@ def test_anonymous_quota_blocks_after_limit(tmp_path, request):
     resp = client.get("/test")
     assert resp.status_code == 429
     assert resp.json()["error"]["message"] == "Quota exceeded"
-
-    asyncio.run(app.close())