#!/usr/bin/env python3
"""test_box_onboard_tui.py — Unit tests for dedicated Onboard Connects & Tmux Auto-Approver TUI."""

import curses
import os
import sys
import unittest
from pathlib import Path
from unittest.mock import MagicMock, patch

REPO_ROOT = Path(__file__).resolve().parent.parent
import importlib.util
tui_path = REPO_ROOT / "bin" / "box-onboard-tui.py"
spec = importlib.util.spec_from_file_location("box_onboard_tui", tui_path)
box_onboard_tui = importlib.util.module_from_spec(spec)
sys.modules["box_onboard_tui"] = box_onboard_tui
spec.loader.exec_module(box_onboard_tui)
BoxOnboardTUI = box_onboard_tui.BoxOnboardTUI


class TestBoxOnboardTUI(unittest.TestCase):
    @patch("curses.init_pair")
    @patch("curses.color_pair")
    @patch("curses.has_colors", return_value=True)
    @patch("curses.start_color")
    @patch("box_onboard_tui.gather_tmux_tally")
    @patch("box_onboard_tui.get_all_connects")
    def test_tui_initialization_and_tabs(
        self, mock_connects, mock_tally, mock_start_color, mock_has_colors, mock_color_pair, mock_init_pair
    ):
        mock_stdscr = MagicMock()
        mock_stdscr.getmaxyx.return_value = (40, 120)

        mock_connects.return_value = [
            {"node": "muse", "type": "fleet_agent", "email": "muse@test.com", "stage": "active_fleet"},
            {"node": "test-client", "type": "onboard_pipeline", "email": "c@test.com", "stage": "awaiting_otp"},
        ]

        mock_tally.return_value = MagicMock(
            total_sockets=2,
            total_sessions=3,
            total_panes=5,
            active_workers=2,
            panes=[],
            by_agent={"muse": {"sessions": 1, "panes": 2, "active_commands": ["muse-bin"], "auto_approve": True}},
        )

        app = BoxOnboardTUI(mock_stdscr)
        self.assertEqual(app.current_tab, 0)
        self.assertEqual(len(app.tabs), 4)
        self.assertEqual(len(app.connects), 2)

        # Tab navigation via key
        app._handle_key(ord("\t"))
        self.assertEqual(app.current_tab, 1)

        app._handle_key(ord("\t"))
        self.assertEqual(app.current_tab, 2)

        app._handle_key(ord("\t"))
        self.assertEqual(app.current_tab, 3)

        app._handle_key(ord("\t"))
        self.assertEqual(app.current_tab, 0)

    @patch("curses.init_pair")
    @patch("curses.color_pair")
    @patch("curses.has_colors", return_value=True)
    @patch("box_onboard_tui.gather_tmux_tally")
    @patch("box_onboard_tui.get_all_connects")
    def test_navigation_keys(
        self, mock_connects, mock_tally, mock_has_colors, mock_color_pair, mock_init_pair
    ):
        mock_stdscr = MagicMock()
        mock_stdscr.getmaxyx.return_value = (40, 120)
        mock_connects.return_value = [
            {"node": f"node-{i}", "type": "fleet_agent"} for i in range(5)
        ]

        app = BoxOnboardTUI(mock_stdscr)
        self.assertEqual(app.sel_connect_idx, 0)

        # Move down
        app._handle_key(curses.KEY_DOWN)
        self.assertEqual(app.sel_connect_idx, 1)

        # Move up
        app._handle_key(curses.KEY_UP)
        self.assertEqual(app.sel_connect_idx, 0)

        # Switch tab via number key '2'
        app._handle_key(ord("2"))
        self.assertEqual(app.current_tab, 1)

        # Switch tab via number key '3'
        app._handle_key(ord("3"))
        self.assertEqual(app.current_tab, 2)

        # Switch tab via number key '4'
        app._handle_key(ord("4"))
        self.assertEqual(app.current_tab, 3)


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