mirror of
https://github.com/long2ice/fastapi-cache.git
synced 2026-03-25 04:57:54 +00:00
23d439f83a
This is, for the majority of backends, the native format anyway, and so we save encoding and decoding when using the PickleCodec or if (in future) a orjson Coder was to be added. For the JsonCodec, the only thing that changed is the location where the JSON data is encoded to bytes and decoded back again to a string.
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
from dataclasses import dataclass
|
|
from typing import Any, Optional
|
|
|
|
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)
|