fix tests

Author: John Lyu

examples/in_memory/main.py

@@ -107,6 +107,15 @@ async def uncached_put():
     put_ret = put_ret + 1
     return {"value": put_ret}
 
+put_ret2 = 0
+
+@app.get("/cached_put")
+@cache(namespace="test", expire=5)
+async def cached_put():
+    global put_ret2
+    put_ret2 = put_ret2 + 1
+    return {"value": put_ret2}
+
 
 @app.get("/namespaced_injection")
 @cache(namespace="test", expire=5, injected_dependency_namespace="monty_python")

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "fastapi-cache2"
-version = "0.2.3"
+version = "0.2.1"
 description = "Cache for FastAPI"
 authors = ["long2ice <[email protected]>"]
 license = "Apache-2.0"

tests/test_decorator.py

@@ -99,10 +99,10 @@ def test_pydantic_model() -> None:
 
 def test_non_get() -> None:
     with TestClient(app) as client:
-        response = client.put("/uncached_put")
+        response = client.put("/cached_put")
         assert "X-FastAPI-Cache" not in response.headers
         assert response.json() == {"value": 1}
-        response = client.put("/uncached_put")
+        response = client.put("/cached_put")
         assert "X-FastAPI-Cache" not in response.headers
         assert response.json() == {"value": 2}
 
@@ -115,25 +115,23 @@ def test_alternate_injected_namespace() -> None:
 
 def test_cache_control() -> None:
     with TestClient(app) as client:
-        response = client.put("/uncached_put")
-        assert "X-FastAPI-Cache" not in response.headers
+        response = client.get("/cached_put")
         assert response.json() == {"value": 1}
 
         # HIT
-        response = client.put("/uncached_put")
-        assert response.headers.get("X-FastAPI-Cache") == "HIT"
+        response = client.get("/cached_put")
         assert response.json() == {"value": 1}
 
         # no-cache
-        response = client.put("/uncached_put", headers={"Cache-Control": "no-cache"})
+        response = client.get("/cached_put", headers={"Cache-Control": "no-cache"})
         assert response.json() == {"value": 2}
 
-        response = client.put("/uncached_put")
+        response = client.get("/cached_put")
         assert response.json() == {"value": 2}
 
         # no-store
-        response = client.put("/uncached_put", headers={"Cache-Control": "no-store"})
+        response = client.get("/cached_put", headers={"Cache-Control": "no-store"})
         assert response.json() == {"value": 3}
 
-        response = client.put("/uncached_put")
+        response = client.get("/cached_put")
         assert response.json() == {"value": 2}