import unittest
from unittest.mock import Mock, patch
from threading import Event
import time

from sia.auto_approver import AutoApprover
from sia.web_agent import WebAgent, WebAgentState

class AutoApproverTest(unittest.TestCase):
    def setUp(self):
        """Create mock agent and approver for each test"""
        self.mock_agent = Mock(spec=WebAgent)
        self.mock_agent.state = WebAgentState.UPDATE
        self.approver = AutoApprover(self.mock_agent)

    def test_timeout_setters(self):
        """Test timeout setters block changes when enabled"""
        self.approver.context_timeout = 10.0
        self.approver.response_timeout = 15.0
        
        self.approver.context_enabled = True
        with self.assertRaises(ValueError):
            self.approver.context_timeout = 20.0
            
        self.approver.response_enabled = True
        with self.assertRaises(ValueError):
            self.approver.response_timeout = 25.0

    def test_enable_starts_thread_in_correct_state(self):
        """Test enabling only starts thread in matching state"""
        self.mock_agent.state = WebAgentState.CONTEXT_APPROVAL
        self.approver.context_enabled = True
        time.sleep(0.1)  # Allow thread to start
        self.assertTrue(self.approver._context_thread.is_alive())
        self.assertIsNone(self.approver._response_thread)

    def test_state_change_stops_threads(self):
        """Test state changes stop irrelevant threads"""
        # Start in context approval with thread
        self.mock_agent.state = WebAgentState.CONTEXT_APPROVAL
        self.approver.context_enabled = True
        time.sleep(0.1)
        self.assertTrue(self.approver._context_thread.is_alive())
        
        # Change state and verify thread stops
        self.mock_agent.state = WebAgentState.INFERENCE
        self.approver._handle_state_change(WebAgentState.INFERENCE)
        time.sleep(0.1)
        self.assertIsNone(self.approver._context_thread)

    def test_quick_state_cycle(self):
        """Test thread stops when state changes before timeout"""
        self.approver.context_timeout = 5.0
        self.mock_agent.state = WebAgentState.CONTEXT_APPROVAL
        self.approver.context_enabled = True
        
        # Change state quickly
        time.sleep(0.1)
        self.mock_agent.state = WebAgentState.INFERENCE
        self.approver._handle_state_change(WebAgentState.INFERENCE)
        
        # Verify no approval happened
        self.mock_agent.approve_context.assert_not_called()

    def test_disable_stops_thread(self):
        """Test disabling stops active thread"""
        self.mock_agent.state = WebAgentState.CONTEXT_APPROVAL
        self.approver.context_enabled = True
        time.sleep(0.1)
        self.assertTrue(self.approver._context_thread.is_alive())
        
        self.approver.context_enabled = False
        self.assertIsNone(self.approver._context_thread)

    def test_successful_auto_approval(self):
        """Test thread approves after timeout when conditions met"""
        self.approver.context_timeout = 0.1
        self.mock_agent.state = WebAgentState.CONTEXT_APPROVAL
        self.approver.context_enabled = True
        
        time.sleep(0.2)
        self.mock_agent.approve_context.assert_called_once()

    def tearDown(self):
        """Clean up any running threads"""
        self.approver.context_enabled = False
        self.approver.response_enabled = False