Skip to content

Commit 47f65ab

Browse files
ELI-674 - fixed unit tests
1 parent de258e9 commit 47f65ab

2 files changed

Lines changed: 9 additions & 5 deletions

File tree

tests/unit/validation/test_app.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,11 @@ def test_next_iteration_exists():
136136
next_mock = Mock()
137137
next_mock.iteration_number = 8
138138
next_mock.iteration_date = today + timedelta(days=5)
139-
next_mock.iteration_datetime = today + timedelta(days=5)
139+
next_mock.iteration_datetime = datetime.combine(
140+
next_mock.iteration_date,
141+
datetime.min.time(),
142+
tzinfo=UTC,
143+
)
140144

141145
result = Mock()
142146
result.campaign_config.end_date = today + timedelta(days=10)

tests/unit/validation/test_iteration_validator.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -519,13 +519,13 @@ def test_invalid_iteration_collects_errors_if_iteration_rules_have_invalid_data(
519519
("iteration_time_input", "default_time_iteration_input", "expected_date_time"),
520520
[
521521
# Case 1: Iteration time overrides default
522-
("14:30:00", "09:00:00", datetime(2025, 1, 2, 14, 30, 0)), # noqa: DTZ001
522+
("14:30:00", "09:00:00", datetime(2025, 1, 2, 14, 30, 0, tzinfo=UTC)),
523523
# Case 2: Iteration time is missing, so it uses campaign config iteration_time
524-
(None, "09:00:00", datetime(2025, 1, 2, 9, 0, 0)), # noqa: DTZ001
524+
(None, "09:00:00", datetime(2025, 1, 2, 9, 0, 0, tzinfo=UTC)),
525525
# Case 3: Both are the same
526-
("10:00:00", "10:00:00", datetime(2025, 1, 2, 10, 0, 0)), # noqa: DTZ001
526+
("10:00:00", "10:00:00", datetime(2025, 1, 2, 10, 0, 0, tzinfo=UTC)),
527527
# Case 4: Both are None, falls back to default value (12 AM) in campaign config iteration_time
528-
(None, None, datetime(2025, 1, 2, 0, 0, 0)), # noqa: DTZ001
528+
(None, None, datetime(2025, 1, 2, 0, 0, 0, tzinfo=UTC)),
529529
],
530530
)
531531
def test_iteration_full_datetime_validation(

0 commit comments

Comments
 (0)