"""Holiday schedule targets: SQS when JOBS_QUEUE_ARN is set, else Lambda."""

import json
from unittest.mock import MagicMock

from shared.side_effects import (
    _holiday_schedule_target,
    activate_holiday_inline,
    create_holiday_schedules,
)


def test_sqs_target_when_jobs_queue_arn_set(monkeypatch):
    monkeypatch.setenv(
        "JOBS_QUEUE_ARN", "arn:aws:sqs:us-east-1:1:afterhours-shift-manager-jobs"
    )
    monkeypatch.setenv(
        "HOLIDAY_SCHEDULER_ROLE_ARN", "arn:aws:iam::1:role/tf-managed/holiday"
    )
    monkeypatch.delenv("HOLIDAY_ROUTER_ARN", raising=False)
    target = _holiday_schedule_target("activate", "2026-07-04")
    assert target["Arn"].endswith(":afterhours-shift-manager-jobs")
    payload = json.loads(target["Input"])
    assert payload == {
        "event": "holiday",
        "action": "activate",
        "date": "2026-07-04",
    }


def test_lambda_target_when_jobs_queue_unset(monkeypatch):
    monkeypatch.delenv("JOBS_QUEUE_ARN", raising=False)
    monkeypatch.setenv(
        "HOLIDAY_ROUTER_ARN",
        "arn:aws:lambda:us-east-1:1:function:afterhours-holiday-router",
    )
    monkeypatch.setenv(
        "HOLIDAY_SCHEDULER_ROLE_ARN", "arn:aws:iam::1:role/tf-managed/holiday"
    )
    target = _holiday_schedule_target("deactivate", "2026-07-04")
    assert "function:afterhours-holiday-router" in target["Arn"]
    payload = json.loads(target["Input"])
    assert payload == {"action": "deactivate", "date": "2026-07-04"}


def test_inline_activate_in_process_when_jobs_queue_url_set(monkeypatch):
    monkeypatch.setenv("JOBS_QUEUE_URL", "https://sqs.us-east-1.amazonaws.com/1/jobs")
    activate = MagicMock()
    monkeypatch.setattr("shared.holiday_flow.activate", activate)
    activate_holiday_inline(MagicMock(), "2026-07-04")
    activate.assert_called_once()


def test_inline_activate_invokes_lambda_without_jobs_queue(monkeypatch):
    monkeypatch.delenv("JOBS_QUEUE_URL", raising=False)
    monkeypatch.setenv(
        "HOLIDAY_ROUTER_ARN", "arn:aws:lambda:us-east-1:1:function:router"
    )
    client = MagicMock()
    monkeypatch.setattr("shared.side_effects.boto3.client", lambda _svc: client)
    activate_holiday_inline(MagicMock(), "2026-07-04")
    client.invoke.assert_called_once()
    assert client.invoke.call_args.kwargs["FunctionName"].endswith(":function:router")


def test_create_holiday_schedules_uses_sqs_target(monkeypatch):
    monkeypatch.setenv(
        "JOBS_QUEUE_ARN", "arn:aws:sqs:us-east-1:1:afterhours-shift-manager-jobs"
    )
    monkeypatch.setenv(
        "HOLIDAY_SCHEDULER_ROLE_ARN", "arn:aws:iam::1:role/tf-managed/holiday"
    )
    client = MagicMock()
    monkeypatch.setattr("shared.side_effects.boto3.client", lambda _svc: client)
    names = create_holiday_schedules("2026-07-04")
    assert names == ["holiday-activate-20260704", "holiday-deactivate-20260704"]
    assert client.create_schedule.call_count == 2
    first_input = json.loads(
        client.create_schedule.call_args_list[0].kwargs["Target"]["Input"]
    )
    assert first_input["event"] == "holiday"