afterhours-shift-manager/tests/ring_scheduler/test_handler.py

"""Tests for the ring-scheduler Lambda handler."""

from unittest.mock import MagicMock

import pytest
from freezegun import freeze_time

from shared.schedule import FALLBACK_EXTENSION

# Monday 2026-06-01 08:00 ET — weekday valid_hours is {8}, so the guard passes
# without force; active shift is night.
MON_0800 = "2026-06-01 12:00:00"


@pytest.fixture
def env(monkeypatch):
    monkeypatch.setenv("QUEUE_NUMBER", "800")
    monkeypatch.setenv("TCX_SECRET_PREFIX", "afterhours-shift-manager/3cx-")


@pytest.fixture
def routing(ringscheduler_app, monkeypatch):
    spy = MagicMock(name="update_queue_routing", return_value={"queue": "800"})
    monkeypatch.setattr(ringscheduler_app, "update_queue_routing", spy)
    monkeypatch.setattr(ringscheduler_app, "get_secret", lambda _id: "secret")
    return spy


@freeze_time(MON_0800)
def test_override_extension_bypasses_schedule(ringscheduler_app, env, routing):
    result = ringscheduler_app.handler({"override_extension": "199"}, None)
    assert routing.call_args.kwargs["extension"] == "199"
    assert result["shift_type"] == "night"


@freeze_time(MON_0800)
def test_resolves_assigned_extension(ringscheduler_app, schedule, seed, env, routing):
    seed.weekly("Monday", "114", "Alice")
    ringscheduler_app.handler({}, None)
    assert routing.call_args.kwargs["extension"] == "114"
    assert routing.call_args.kwargs["queue_number"] == "800"


@freeze_time(MON_0800)
def test_unassigned_uses_fallback(ringscheduler_app, schedule, seed, env, routing):
    ringscheduler_app.handler({}, None)
    assert routing.call_args.kwargs["extension"] == FALLBACK_EXTENSION


@freeze_time(MON_0800)
def test_routing_failure_returns_error(ringscheduler_app, schedule, env, routing):
    routing.side_effect = RuntimeError("3CX down")
    result = ringscheduler_app.handler({"override_extension": "199"}, None)
    assert result["error"] is True


@freeze_time("2026-06-01 13:00:00")  # 09:00 ET — not a valid scheduler hour
def test_skips_on_wrong_hour(ringscheduler_app, env, routing):
    result = ringscheduler_app.handler({}, None)
    assert result == {"skipped": True}
    routing.assert_not_called()