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
|
from datetime import UTC, datetime, time
from unittest.mock import patch
from zoneinfo import ZoneInfo
import pytest
from generalresearch.models.definitions import Source
from generalresearch.models.thl.survey.model import Survey, TimeInterval
@pytest.mark.parametrize(
("hour", "minute", "expected"),
[
(6, 59, False),
(7, 0, True),
(12, 0, True),
(22, 0, True),
(22, 1, False),
],
)
def test_is_currently_active_in_worker_local_time(hour, minute, expected):
interval = TimeInterval(start=time(7), end=time(22), timezone=None)
worker_time = datetime(
2026, 1, 15, hour, minute, tzinfo=ZoneInfo("America/Chicago")
)
assert interval.is_currently_active(worker_time) is expected
@pytest.mark.parametrize(
("hour", "expected"),
[
(21, False),
(22, True),
(23, True),
(0, True),
(6, True),
(7, False),
],
)
def test_is_currently_active_for_interval_crossing_midnight(hour, expected):
interval = TimeInterval(start=time(22), end=time(6), timezone=None)
worker_time = datetime(2026, 1, 15, hour, tzinfo=UTC)
assert interval.is_currently_active(worker_time) is expected
def test_is_currently_active_converts_to_interval_timezone():
interval = TimeInterval(
start=time(7),
end=time(9),
timezone="America/New_York",
)
# 08:00 in New York on this date.
worker_time = datetime(
2026, 1, 15, 7, tzinfo=ZoneInfo("America/Chicago")
)
assert interval.is_currently_active(worker_time) is True
def test_is_currently_active_requires_aware_worker_time():
interval = TimeInterval(start=time(7), end=time(22), timezone=None)
with pytest.raises(AssertionError):
interval.is_currently_active(datetime(2026, 1, 15, 12))
def test_is_currently_active_requires_worker_timezone_for_local_interval():
interval = TimeInterval(start=time(7), end=time(22), timezone=None)
assert interval.is_currently_active_by_worker_tz(worker_tz=None) is False
@pytest.mark.parametrize(
("interval_timezone", "worker_timezone"),
[
(None, ZoneInfo("America/Chicago")),
("America/New_York", None),
],
)
def test_is_currently_active_supports_interval_crossing_midnight(
interval_timezone, worker_timezone
):
interval = TimeInterval(
start=time(22),
end=time(6),
timezone=interval_timezone,
)
current_timezone = (
ZoneInfo(interval_timezone) if interval_timezone else worker_timezone
)
current_time = datetime(2026, 1, 15, 23, tzinfo=current_timezone)
with patch("generalresearch.models.thl.survey.model.datetime") as mock_datetime:
mock_datetime.now.return_value = current_time
assert interval.is_currently_active_by_worker_tz(worker_timezone) is True
mock_datetime.now.assert_called_once_with(tz=current_timezone)
@pytest.mark.parametrize(
("off_start", "off_end", "active_start", "active_end"),
[
(time(1), time(5), time(5), time(1)),
(time(22), time(6), time(6), time(22)),
],
)
def test_from_off_hours_returns_complementary_active_hours(
off_start, off_end, active_start, active_end
):
off_hours = TimeInterval(
start=off_start,
end=off_end,
timezone="America/New_York",
)
active_hours = Survey.active_hours_from_off_hours(off_hours)
assert active_hours == [
TimeInterval(
start=active_start,
end=active_end,
timezone="America/New_York",
)
]
def test_from_off_hours_result_can_be_assigned_to_survey():
off_hours = TimeInterval(start=time(1), end=time(5), timezone=None)
survey = Survey(
source=Source.TESTING,
survey_id="off-hours-test",
active_hours=Survey.active_hours_from_off_hours(off_hours),
)
assert survey.active_hours == [
TimeInterval(start=time(5), end=time(1), timezone=None)
]
|
