mirror of
https://github.com/long2ice/fastapi-cache.git
synced 2026-03-25 13:07:53 +00:00
73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
from dataclasses import dataclass
|
|
from typing import Any, Dict, List, Optional, Type
|
|
|
|
import pytest
|
|
from pydantic import BaseModel
|
|
|
|
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
|
|
|
|
|
|
# vicchi: 2024/11/09 - some values commented out until #460 is resolved
|
|
@pytest.mark.parametrize(
|
|
("value", "return_type"),
|
|
[
|
|
(1, int),
|
|
(1, None),
|
|
("some_string", str),
|
|
("some_string", None),
|
|
# ((1, 2), Tuple[int, int]),
|
|
([1, 2, 3], List[int]),
|
|
([1, 2, 3], None),
|
|
({"some_key": 1, "other_key": 2}, Dict[str, int]),
|
|
({"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: Type[Any]) -> 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
|
|
|
|
|
|
# vicchi: 2024/11/09 - test commented out until #460 is resolved
|
|
# def test_json_coder_validation_error() -> None:
|
|
# invalid = b'{"name": "incomplete"}'
|
|
# with pytest.raises(ValidationError):
|
|
# JsonCoder.decode_as_type(invalid, type_=PDItem)
|