1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
"""Test fixtures."""
from collections.abc import Generator
import dataclasses
import json
from typing import Any
from unittest.mock import patch
from pydantic_core import to_jsonable_python
import pytest
PRODID = "-//example//1.2.3"
class DataclassEncoder(json.JSONEncoder):
"""Class that can dump data classes as dict for comparison to golden."""
def default(self, o: Any) -> Any:
if dataclasses.is_dataclass(o):
# Omit empty
return {k: v for (k, v) in dataclasses.asdict(o).items() if v}
if isinstance(o, dict):
return {k: v for (k, v) in o.items() if v}
return to_jsonable_python(o)
@pytest.fixture
def json_encoder() -> json.JSONEncoder:
"""Fixture that creates a json encoder."""
return DataclassEncoder()
@pytest.fixture(autouse=True)
def mock_prodid() -> Generator[None, None, None]:
"""Mock out the prodid used in tests."""
with patch("ical.calendar.prodid_factory", return_value=PRODID):
yield
|
