#!/usr/bin/env python3
"""test_onboard_pipeline.py — Unit tests for the onboarding and invite salvage pipeline."""

import json
import sys
import unittest
from pathlib import Path
from unittest.mock import MagicMock, patch

REPO_ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(REPO_ROOT / "bin"))

from onboard_pipeline import (
    OnboardState,
    finish_onboarding_redemption,
    select_urgent_beneficiary,
    start_onboarding,
    submit_onboarding_otp,
)


class TestOnboardPipeline(unittest.TestCase):
    def test_select_urgent_beneficiary_explicit(self):
        b_node, code, reason = select_urgent_beneficiary("dev", "TEST01")
        self.assertEqual(b_node, "dev")
        self.assertEqual(code, "TEST01")
        self.assertIn("Explicit", reason)

    @patch("invite.fleet_invite_status")
    @patch("invite.fleet_usage")
    def test_select_urgent_beneficiary_blocked_first(self, mock_usage, mock_invite):
        mock_usage.return_value = {
            "646": {"weekly_used_pct": 100, "additional_used_pct": 100, "additional_left": "0 tokens left"},
            "pip": {"weekly_used_pct": 100, "additional_used_pct": 10, "additional_left": "890M tokens left"},
            "muse": {"weekly_used_pct": 50, "additional_used_pct": 0, "additional_left": "950M tokens left"},
        }
        mock_invite.return_value = {
            "646": {"code": "REDCJ7"},
            "pip": {"code": "F4BGHN"},
            "muse": {"code": "81MDIR"},
        }

        b_node, code, reason = select_urgent_beneficiary()
        self.assertEqual(b_node, "646")
        self.assertEqual(code, "REDCJ7")
        self.assertIn("BLOCKED", reason)

    @patch("onboard_pipeline.provision_node_infra")
    @patch("onboard_pipeline.CredClient")
    @patch("onboard_pipeline.select_urgent_beneficiary")
    def test_start_onboarding_awaiting_otp(self, mock_select, mock_cred_cls, mock_infra):
        mock_select.return_value = ("646", "REDCJ7", "Blocked test")
        mock_infra.return_value = {"ok": True}

        mock_client = MagicMock()
        mock_client.initiate.return_value = {
            "status": "awaiting_otp",
            "message": "OTP sent",
        }
        mock_cred_cls.return_value = mock_client

        res = start_onboarding("testnode", "client@test.com")
        self.assertTrue(res["ok"])
        self.assertEqual(res["status"], "awaiting_otp")
        self.assertEqual(res["beneficiary_node"], "646")
        self.assertEqual(res["invite_code_queued"], "REDCJ7")

    @patch("invite.redeem_invite")
    @patch("subprocess.run")
    def test_finish_onboarding_redemption_success(self, mock_subproc, mock_redeem):
        mock_redeem.return_value = {"ok": True, "redemption_status": "redeemed"}
        state = OnboardState(
            node="testnode",
            email="client@test.com",
            beneficiary_node="646",
            invite_code="REDCJ7",
            stage="auth_active",
        )

        res = finish_onboarding_redemption(state)
        self.assertTrue(res["ok"])
        self.assertEqual(res["status"], "completed")
        self.assertEqual(res["beneficiary_node"], "646")
        mock_redeem.assert_called_once_with("testnode", "REDCJ7", timeout=20.0)


if __name__ == "__main__":
    unittest.main()