mirror of
https://github.com/Sea-Haven-Industries/afterhours-shift-manager.git
synced 2026-09-30 06:43:12 +00:00
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
|
"""Tests for slack-bot module-level helpers."""
|
||
|
|
|
||
|
|
from freezegun import freeze_time
|
||
|
|
|
||
|
|
# Monday 2026-06-01 08:00 ET → weekday, active shift is night.
|
||
|
|
MON = "2026-06-01 12:00:00"
|
||
|
|
SAT_DATE = "2026-06-06"
|
||
|
|
|
||
|
|
|
||
|
|
class TestIsToday:
|
||
|
|
@freeze_time(MON)
|
||
|
|
def test_true_for_today(self, slackbot_app):
|
||
|
|
assert slackbot_app.is_today("2026-06-01") is True
|
||
|
|
|
||
|
|
@freeze_time(MON)
|
||
|
|
def test_false_for_other_day(self, slackbot_app):
|
||
|
|
assert slackbot_app.is_today("2026-06-02") is False
|
||
|
|
|
||
|
|
|
||
|
|
class TestIsActiveShiftType:
|
||
|
|
@freeze_time(MON)
|
||
|
|
def test_weekday_active_is_night(self, slackbot_app):
|
||
|
|
assert slackbot_app._is_active_shift_type("night") is True
|
||
|
|
assert slackbot_app._is_active_shift_type("day") is False
|
||
|
|
|
||
|
|
|
||
|
|
class TestShiftTypeLabel:
|
||
|
|
def test_weekday_has_no_label(self, slackbot_app):
|
||
|
|
assert slackbot_app._shift_type_label("Wednesday", "night") == ""
|
||
|
|
|
||
|
|
def test_weekend_day(self, slackbot_app):
|
||
|
|
assert slackbot_app._shift_type_label("Saturday", "day") == " (Day)"
|
||
|
|
|
||
|
|
def test_weekend_night(self, slackbot_app):
|
||
|
|
assert slackbot_app._shift_type_label("Saturday", "night") == " (Night)"
|
||
|
|
|
||
|
|
|
||
|
|
class TestFindEmployeeShift:
|
||
|
|
def test_weekday_night_match(self, slackbot_app, schedule, seed):
|
||
|
|
seed.weekly("Wednesday", "114", "Alice")
|
||
|
|
found = slackbot_app._find_employee_shift(
|
||
|
|
schedule, "2026-06-03", "Wednesday", "114"
|
||
|
|
)
|
||
|
|
assert found == ("114", "Alice", "weekly", "night")
|
||
|
|
|
||
|
|
def test_weekend_day_match(self, slackbot_app, schedule, seed):
|
||
|
|
seed.weekly("Saturday", "200", "Alice", shift_type="day")
|
||
|
|
found = slackbot_app._find_employee_shift(schedule, SAT_DATE, "Saturday", "200")
|
||
|
|
assert found == ("200", "Alice", "weekly", "day")
|
||
|
|
|
||
|
|
def test_weekend_night_match(self, slackbot_app, schedule, seed):
|
||
|
|
seed.weekly("Saturday", "201", "Bob", shift_type="night")
|
||
|
|
found = slackbot_app._find_employee_shift(schedule, SAT_DATE, "Saturday", "201")
|
||
|
|
assert found == ("201", "Bob", "weekly", "night")
|
||
|
|
|
||
|
|
def test_not_assigned_returns_none(self, slackbot_app, schedule, seed):
|
||
|
|
seed.weekly("Wednesday", "114", "Alice")
|
||
|
|
assert (
|
||
|
|
slackbot_app._find_employee_shift(
|
||
|
|
schedule, "2026-06-03", "Wednesday", "999"
|
||
|
|
)
|
||
|
|
is None
|
||
|
|
)
|