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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
"""Tests for DURATION data types."""
import datetime
from pydantic import field_serializer
import pytest
from ical.exceptions import CalendarParseError
from ical.component import ComponentModel
from ical.parsing.component import ParsedComponent
from ical.parsing.property import ParsedProperty
from ical.types import Period
from ical.types.data_types import serialize_field
class FakeModel(ComponentModel):
"""Model under test."""
example: Period
serialize_fields = field_serializer("*")(serialize_field) # type: ignore[pydantic-field]
def test_period() -> None:
"""Test for period fields."""
# Time period with end date
model = FakeModel.model_validate(
{
"example": [
ParsedProperty(
name="example", value="19970101T180000Z/19970102T070000Z"
)
]
},
)
assert model.example.start == datetime.datetime(
1997, 1, 1, 18, 0, 0, tzinfo=datetime.timezone.utc
)
assert model.example.end
assert not model.example.duration
assert model.example.end_value == datetime.datetime(
1997, 1, 2, 7, 0, 0, tzinfo=datetime.timezone.utc
)
# Time period with duration
model = FakeModel.model_validate(
{"example": [ParsedProperty(name="example", value="19970101T180000Z/PT5H30M")]},
)
assert model.example.start == datetime.datetime(
1997, 1, 1, 18, 0, 0, tzinfo=datetime.timezone.utc
)
assert not model.example.end
assert model.example.duration
assert model.example.end_value == datetime.datetime(
1997, 1, 1, 23, 30, 0, tzinfo=datetime.timezone.utc
)
with pytest.raises(CalendarParseError):
FakeModel.model_validate(
{"example": [ParsedProperty(name="example", value="a")]}
)
with pytest.raises(CalendarParseError):
FakeModel.model_validate(
{"example": [ParsedProperty(name="example", value="19970101T180000Z/a")]}
)
with pytest.raises(CalendarParseError):
FakeModel.model_validate(
{"example": [ParsedProperty(name="example", value="a/19970102T070000Z")]}
)
with pytest.raises(CalendarParseError):
FakeModel.model_validate(
{"example": [ParsedProperty(name="example", value="a/PT5H30M")]}
)
def test_encode_period() -> None:
"""Test encoded period."""
model = FakeModel(
example=Period(
start=datetime.datetime(2022, 8, 7, 6, 0, 0),
end=datetime.datetime(2022, 8, 7, 6, 30, 0),
)
)
assert model.__encode_component_root__() == ParsedComponent(
name="FakeModel",
properties=[
ParsedProperty(name="example", value="20220807T060000/20220807T063000")
],
)
model = FakeModel(
example=Period(
start=datetime.datetime(2022, 8, 7, 6, 0, 0),
duration=datetime.timedelta(hours=5, minutes=30),
)
)
assert model.__encode_component_root__() == ParsedComponent(
name="FakeModel",
properties=[ParsedProperty(name="example", value="20220807T060000/PT5H30M")],
)
|
