88 lines
3.4 KiB
Python
88 lines
3.4 KiB
Python
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
|