mirror of
https://github.com/Sea-Haven-Industries/afterhours-shift-manager.git
synced 2026-09-30 13:43:12 +00:00
117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
"""Tests for the weekly-post Lambda handler orchestration."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
from freezegun import freeze_time
|
|
|
|
# 2026-06-08 is a Monday. Frozen to ET 08:00; handler is invoked with force=True
|
|
# to bypass the 7am DST guard except where the guard itself is under test.
|
|
MON_0800 = "2026-06-08 12:00:00"
|
|
|
|
|
|
@pytest.fixture
|
|
def slack(weeklypost_app, monkeypatch):
|
|
"""Fake Slack WebClient; chat_postMessage returns a message ts."""
|
|
fake = MagicMock(name="slack")
|
|
fake.chat_postMessage.return_value = {"ts": "999.000"}
|
|
monkeypatch.setattr(weeklypost_app, "WebClient", MagicMock(return_value=fake))
|
|
monkeypatch.setattr(weeklypost_app, "get_secret", lambda _id: "xoxb-test")
|
|
return fake
|
|
|
|
|
|
@pytest.fixture
|
|
def env(monkeypatch):
|
|
monkeypatch.setenv(
|
|
"SLACK_BOT_TOKEN_SECRET", "afterhours-shift-manager/slack-bot-token"
|
|
)
|
|
monkeypatch.setenv("PAY_REPORT_USER", "U_BOSS")
|
|
monkeypatch.delenv("PAYROLL_RECIPIENTS", raising=False) # skip SES email
|
|
|
|
|
|
@freeze_time(MON_0800)
|
|
def test_posts_schedule_and_saves_post(weeklypost_app, schedule, seed, slack, env):
|
|
result = weeklypost_app.handler({"force": True}, None)
|
|
|
|
assert result["posted"] is True
|
|
assert result["message_ts"] == "999.000"
|
|
# The new schedule post was persisted for next week's cleanup.
|
|
assert schedule.get_schedule_post("C_TEST")["message_ts"] == "999.000"
|
|
slack.chat_postMessage.assert_called()
|
|
|
|
|
|
@freeze_time(MON_0800)
|
|
def test_calculates_and_dms_pay(weeklypost_app, schedule, seed, slack, env):
|
|
# Previous week (Mon 2026-06-01) had Alice on the Monday night shift.
|
|
seed.config(shift_rate="50")
|
|
seed.weekly("Monday", "114", "Alice")
|
|
|
|
result = weeklypost_app.handler({"force": True}, None)
|
|
|
|
assert result["pay_calculated"] is True
|
|
# Pay record saved under previous Monday's key.
|
|
assert schedule.get_pay_record("2026-06-01") is not None
|
|
# A DM went to the configured pay-report user.
|
|
dm_calls = [
|
|
c
|
|
for c in slack.chat_postMessage.call_args_list
|
|
if c.kwargs.get("channel") == "U_BOSS"
|
|
]
|
|
assert dm_calls
|
|
|
|
|
|
@freeze_time(MON_0800)
|
|
def test_pay_email_failure_does_not_block_schedule_post(
|
|
weeklypost_app, schedule, seed, slack, env, monkeypatch
|
|
):
|
|
# Previous week has an assigned shift, so the pay/email path runs.
|
|
seed.config(shift_rate="50")
|
|
seed.weekly("Monday", "114", "Alice")
|
|
# SES delivery blows up (e.g. a permission/identity issue).
|
|
monkeypatch.setattr(
|
|
weeklypost_app,
|
|
"_send_pay_email",
|
|
MagicMock(side_effect=Exception("SES AccessDenied")),
|
|
)
|
|
|
|
result = weeklypost_app.handler({"force": True}, None)
|
|
|
|
# The email failure is swallowed; the schedule post still goes out.
|
|
assert result["posted"] is True
|
|
assert schedule.get_schedule_post("C_TEST")["message_ts"] == "999.000"
|
|
|
|
|
|
@freeze_time(MON_0800)
|
|
def test_rolls_existing_post_forward(weeklypost_app, schedule, seed, slack, env):
|
|
# An existing post is edited in place (stable permalink), never deleted.
|
|
seed.schedule_post("C_TEST", "111.111")
|
|
result = weeklypost_app.handler({"force": True}, None)
|
|
|
|
slack.chat_delete.assert_not_called()
|
|
slack.chat_update.assert_called_once()
|
|
assert slack.chat_update.call_args.kwargs["ts"] == "111.111"
|
|
# The stored ts is unchanged, so the permalink stays stable.
|
|
assert result["message_ts"] == "111.111"
|
|
assert schedule.get_schedule_post("C_TEST")["message_ts"] == "111.111"
|
|
|
|
|
|
@freeze_time(MON_0800)
|
|
def test_reposts_when_update_fails(weeklypost_app, schedule, seed, slack, env):
|
|
# A stored post that can no longer be edited (e.g. deleted manually) falls
|
|
# back to a fresh post and re-saves the new ts.
|
|
seed.schedule_post("C_TEST", "111.111")
|
|
slack.chat_update.side_effect = Exception("message_not_found")
|
|
|
|
result = weeklypost_app.handler({"force": True}, None)
|
|
|
|
slack.chat_postMessage.assert_called()
|
|
assert result["message_ts"] == "999.000"
|
|
assert schedule.get_schedule_post("C_TEST")["message_ts"] == "999.000"
|
|
|
|
|
|
@freeze_time(MON_0800)
|
|
def test_skips_when_not_7am_and_not_forced(weeklypost_app, schedule, slack, env):
|
|
# Frozen hour is 08:00 ET, not 07:00 → skip unless forced.
|
|
result = weeklypost_app.handler({}, None)
|
|
assert result == {"skipped": True}
|
|
slack.chat_postMessage.assert_not_called()
|