mirror of
https://github.com/long2ice/fastapi-cache.git
synced 2026-03-25 04:57:54 +00:00
013be85f97
- Compatibility with older Python versions - use `Optional` and `Union` instead of `... | None` and `a | b` - use `typing_extensions.Protocol` instead of `typing.Protocol` - use `typing.Dict`, `typing.List`, etc. instead of the concrete types. - Fix backend `.get()` annotations; not all were marked as `Optional[str]` - Don't return anything from `Backend.set()` methods. - The `Coder.decode_as_type()` type parameter must be a type to be compatible with `ModelField(..., type_=...)`. - Clean up `Optional[]` use, remove where it is not needed. - Clean up variable use in decorator, keeping the raw cached value separate from the return value from the wrapped endpoint. - Annotate the wrapper as returning either the original type _or_ a Response (returning a 304 Not Modified response). - Clean up small edge-case where `response` could be `None`. - Correct type annotation on `JsonCoder.decode()` to match `Coder.decode()`.
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
from dataclasses import dataclass
|
|
from typing import Any, Optional, Tuple
|
|
|
|
import pytest
|
|
from pydantic import BaseModel, ValidationError
|
|
|
|
from fastapi_cache.coder import JsonCoder, PickleCoder
|
|
|
|
|
|
@dataclass
|
|
class DCItem:
|
|
name: str
|
|
price: float
|
|
description: Optional[str] = None
|
|
tax: Optional[float] = None
|
|
|
|
|
|
class PDItem(BaseModel):
|
|
name: str
|
|
price: float
|
|
description: Optional[str] = None
|
|
tax: Optional[float] = None
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"value",
|
|
[
|
|
1,
|
|
"some_string",
|
|
(1, 2),
|
|
[1, 2, 3],
|
|
{"some_key": 1, "other_key": 2},
|
|
DCItem(name="foo", price=42.0, description="some dataclass item", tax=0.2),
|
|
PDItem(name="foo", price=42.0, description="some pydantic item", tax=0.2),
|
|
],
|
|
)
|
|
def test_pickle_coder(value: Any) -> None:
|
|
encoded_value = PickleCoder.encode(value)
|
|
assert isinstance(encoded_value, bytes)
|
|
decoded_value = PickleCoder.decode(encoded_value)
|
|
assert decoded_value == value
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("value", "return_type"),
|
|
[
|
|
(1, None),
|
|
("some_string", None),
|
|
((1, 2), Tuple[int, int]),
|
|
([1, 2, 3], None),
|
|
({"some_key": 1, "other_key": 2}, None),
|
|
(DCItem(name="foo", price=42.0, description="some dataclass item", tax=0.2), DCItem),
|
|
(PDItem(name="foo", price=42.0, description="some pydantic item", tax=0.2), PDItem),
|
|
],
|
|
)
|
|
def test_json_coder(value: Any, return_type) -> None:
|
|
encoded_value = JsonCoder.encode(value)
|
|
assert isinstance(encoded_value, bytes)
|
|
decoded_value = JsonCoder.decode_as_type(encoded_value, type_=return_type)
|
|
assert decoded_value == value
|
|
|
|
|
|
def test_json_coder_validation_error() -> None:
|
|
invalid = b'{"name": "incomplete"}'
|
|
with pytest.raises(ValidationError):
|
|
JsonCoder.decode_as_type(invalid, type_=PDItem)
|