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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
from datetime import datetime
from django.utils.timezone import make_aware
from recurrence import choices, deserialize
from recurrence.models import Date, Recurrence, Rule
import pytest
import recurrence
@pytest.mark.django_db
def test_recurrence_to_recurrence_object():
limits = Recurrence.objects.create()
Rule.objects.create(
recurrence=limits,
mode=choices.INCLUSION,
freq=recurrence.WEEKLY
)
object = limits.to_recurrence_object()
assert [r.to_text() for r in object.rrules] == ['weekly']
assert object.exrules == []
assert object.rdates == []
assert object.exdates == []
@pytest.mark.django_db
def test_recurrence_to_recurrence_object_complex():
limits = Recurrence.objects.create(
dtstart=datetime(2014, 1, 1, 0, 0, 0),
dtend=datetime(2014, 12, 31, 0, 0, 0),
)
Rule.objects.create(
recurrence=limits,
mode=choices.INCLUSION,
freq=recurrence.WEEKLY,
until=make_aware(datetime(2014, 12, 31, 0, 0, 0))
)
Rule.objects.create(
recurrence=limits,
mode=choices.EXCLUSION,
freq=recurrence.MONTHLY,
until=make_aware(datetime(2013, 12, 31, 0, 0, 0))
)
Date.objects.create(
recurrence=limits,
mode=choices.INCLUSION,
dt=make_aware(datetime(2012, 12, 31, 0, 0, 0))
)
Date.objects.create(
recurrence=limits,
mode=choices.EXCLUSION,
dt=make_aware(datetime(2011, 12, 31, 0, 0, 0))
)
object = limits.to_recurrence_object()
assert object.dtstart == make_aware(datetime(2014, 1, 1, 0, 0, 0))
assert object.dtend == make_aware(datetime(2014, 12, 31, 0, 0, 0))
assert len(object.rrules) == 1
output_rule = object.rrules[0]
assert output_rule.freq == recurrence.WEEKLY
assert output_rule.until == make_aware(datetime(2014, 12, 31, 0, 0, 0))
assert len(object.exrules) == 1
output_rule = object.exrules[0]
assert output_rule.freq == recurrence.MONTHLY
assert output_rule.until == make_aware(datetime(2013, 12, 31, 0, 0, 0))
@pytest.mark.django_db
def test_recurrence_to_recurrence_object_non_naive_sd_ed():
limits = Recurrence.objects.create(
dtstart=make_aware(datetime(2014, 1, 1, 0, 0, 0)),
dtend=make_aware(datetime(2014, 12, 31, 0, 0, 0)),
)
object = limits.to_recurrence_object()
assert object.dtstart == make_aware(datetime(2014, 1, 1, 0, 0, 0))
assert object.dtend == make_aware(datetime(2014, 12, 31, 0, 0, 0))
@pytest.mark.django_db
def test_create_from_recurrence_object():
inrule = recurrence.Rule(
recurrence.WEEKLY
)
exrule = recurrence.Rule(
recurrence.MONTHLY
)
limits = recurrence.Recurrence(
dtstart=datetime(2014, 1, 1, 0, 0, 0),
dtend=datetime(2014, 2, 3, 0, 0, 0),
rrules=[inrule],
exrules=[exrule],
rdates=[datetime(2014, 2, 15, 0, 0, 0)],
exdates=[make_aware(datetime(2014, 11, 29, 0, 0, 0))]
)
object = Recurrence.objects.create_from_recurrence_object(limits)
assert object.dtstart == make_aware(datetime(2014, 1, 1, 0, 0, 0))
assert object.dtend == make_aware(datetime(2014, 2, 3, 0, 0, 0))
rules = object.rules.all()
assert len(rules) == 2
in_rules = [r for r in rules if r.mode == choices.INCLUSION]
out_rules = [r for r in rules if r.mode == choices.EXCLUSION]
assert len(in_rules) == 1
assert len(out_rules) == 1
assert in_rules[0].freq == recurrence.WEEKLY
assert out_rules[0].freq == recurrence.MONTHLY
dates = object.dates.all()
assert len(dates) == 2
in_dates = [d for d in dates if d.mode == choices.INCLUSION]
out_dates = [d for d in dates if d.mode == choices.EXCLUSION]
assert len(in_dates) == 1
assert len(out_dates) == 1
assert in_dates[0].dt == make_aware(datetime(2014, 2, 15, 0, 0, 0))
assert out_dates[0].dt == make_aware(datetime(2014, 11, 29, 0, 0, 0))
@pytest.mark.django_db
def test_deserialize_to_recurrence_regression():
"""Regression test for checking weekday serialization works correctly:
https://github.com/jazzband/django-recurrence/pull/176
"""
obj = deserialize('DTSTART:20201015T000000\nRRULE:FREQ=WEEKLY;COUNT=30;INTERVAL=1;WKST=MO')
Recurrence.objects.create_from_recurrence_object(obj)
|
