mirror of
https://github.com/long2ice/fastapi-cache.git
synced 2026-03-25 04:57:54 +00:00
e09ede2e4c
Instead of assuming that the Request and Response injected keyword arguments can be named `request` and `response`, use namespaced keyword names starting with a double underscore. By default the parameter names now start with `__fastapi_cache_` and so are a) clearly marked as internal, and b) highly unlikely to clash with existing keyword arguments. The prefix is configurable in the unlikely event that the names would clash in specific cases.
103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
import time
|
|
from typing import Generator
|
|
|
|
import pendulum
|
|
import pytest
|
|
from starlette.testclient import TestClient
|
|
|
|
from examples.in_memory.main import app
|
|
from fastapi_cache import FastAPICache
|
|
from fastapi_cache.backends.inmemory import InMemoryBackend
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def init_cache() -> Generator:
|
|
FastAPICache.init(InMemoryBackend())
|
|
yield
|
|
FastAPICache.reset()
|
|
|
|
|
|
def test_datetime() -> None:
|
|
with TestClient(app) as client:
|
|
response = client.get("/datetime")
|
|
now = response.json().get("now")
|
|
now_ = pendulum.now().replace(microsecond=0)
|
|
assert pendulum.parse(now).replace(microsecond=0) == now_
|
|
response = client.get("/datetime")
|
|
now = response.json().get("now")
|
|
assert pendulum.parse(now).replace(microsecond=0) == now_
|
|
time.sleep(3)
|
|
response = client.get("/datetime")
|
|
now = response.json().get("now")
|
|
now = pendulum.parse(now).replace(microsecond=0)
|
|
assert now != now_
|
|
assert now == pendulum.now().replace(microsecond=0)
|
|
|
|
|
|
def test_date() -> None:
|
|
"""Test path function without request or response arguments."""
|
|
with TestClient(app) as client:
|
|
response = client.get("/date")
|
|
assert pendulum.parse(response.json()) == pendulum.today()
|
|
|
|
# do it again to test cache
|
|
response = client.get("/date")
|
|
assert pendulum.parse(response.json()) == pendulum.today()
|
|
|
|
# now test with cache disabled, as that's a separate code path
|
|
FastAPICache._enable = False
|
|
response = client.get("/date")
|
|
assert pendulum.parse(response.json()) == pendulum.today()
|
|
FastAPICache._enable = True
|
|
|
|
|
|
def test_sync() -> None:
|
|
"""Ensure that sync function support works."""
|
|
with TestClient(app) as client:
|
|
response = client.get("/sync-me")
|
|
assert response.json() == 42
|
|
|
|
|
|
def test_cache_response_obj() -> None:
|
|
with TestClient(app) as client:
|
|
cache_response = client.get("cache_response_obj")
|
|
assert cache_response.json() == {"a": 1}
|
|
get_cache_response = client.get("cache_response_obj")
|
|
assert get_cache_response.json() == {"a": 1}
|
|
assert get_cache_response.headers.get("cache-control")
|
|
assert get_cache_response.headers.get("etag")
|
|
|
|
|
|
def test_kwargs() -> None:
|
|
with TestClient(app) as client:
|
|
name = "Jon"
|
|
response = client.get("/kwargs", params={"name": name})
|
|
assert response.json() == {"name": name}
|
|
|
|
|
|
def test_method() -> None:
|
|
with TestClient(app) as client:
|
|
response = client.get("/method")
|
|
assert response.json() == 17
|
|
|
|
|
|
def test_pydantic_model() -> None:
|
|
with TestClient(app) as client:
|
|
r1 = client.get("/pydantic_instance").json()
|
|
r2 = client.get("/pydantic_instance").json()
|
|
assert r1 == r2
|
|
|
|
|
|
def test_non_get() -> None:
|
|
with TestClient(app) as client:
|
|
response = client.put("/uncached_put")
|
|
assert response.json() == {"value": 1}
|
|
response = client.put("/uncached_put")
|
|
assert response.json() == {"value": 2}
|
|
|
|
|
|
def test_alternate_injected_namespace() -> None:
|
|
with TestClient(app) as client:
|
|
response = client.get("/namespaced_injection")
|
|
assert response.json() == {"__fastapi_cache_request": 42, "__fastapi_cache_response": 17}
|