mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-02-09 06:15:41 -05:00
f6a3113b64
- Made _charge_usage optionally non-blocking with async_mode parameter - Created ChargeManager for background credit processing - Added SyncManager for periodic syncing of queued updates - Integrated managers into ExecutionProcessor lifecycle - Added unit tests for caching functionality The system now validates balance quickly and charges asynchronously, while queued updates are synced every 5 seconds by the background sync manager. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
import unittest
|
|
from unittest.mock import MagicMock
|
|
|
|
from backend.executor.cached_client import wrap_client
|
|
from backend.executor.simple_cache import clear_cache, get_cache
|
|
|
|
|
|
class TestExecutionCache(unittest.TestCase):
|
|
def setUp(self):
|
|
clear_cache()
|
|
self.mock_client = MagicMock()
|
|
self.cached_client = wrap_client(self.mock_client)
|
|
|
|
def test_node_caching(self):
|
|
self.mock_client.get_node.return_value = {"id": "node_1", "data": "test"}
|
|
|
|
# First call should hit backend
|
|
result1 = self.cached_client.get_node("node_1")
|
|
self.assertEqual(self.mock_client.get_node.call_count, 1)
|
|
|
|
# Second call should use cache
|
|
result2 = self.cached_client.get_node("node_1")
|
|
self.assertEqual(self.mock_client.get_node.call_count, 1)
|
|
self.assertEqual(result1, result2)
|
|
|
|
def test_node_executions_caching(self):
|
|
self.mock_client.get_node_executions.return_value = [
|
|
{"id": "exec_1", "status": "completed"}
|
|
]
|
|
|
|
# First call should hit backend
|
|
result1 = self.cached_client.get_node_executions("graph_1")
|
|
self.assertEqual(self.mock_client.get_node_executions.call_count, 1)
|
|
|
|
# Second call should use cache
|
|
result2 = self.cached_client.get_node_executions("graph_1")
|
|
self.assertEqual(self.mock_client.get_node_executions.call_count, 1)
|
|
self.assertEqual(result1, result2)
|
|
|
|
def test_output_updates_queued(self):
|
|
# Should not call backend immediately
|
|
self.cached_client.upsert_execution_output("exec_1", {"data": "output"})
|
|
self.mock_client.upsert_execution_output.assert_not_called()
|
|
|
|
# Check that it was queued
|
|
cache = get_cache()
|
|
outputs, _ = cache.get_pending_updates()
|
|
self.assertEqual(len(outputs), 1)
|
|
self.assertEqual(outputs[0]["node_exec_id"], "exec_1")
|
|
|
|
def test_status_updates_queued(self):
|
|
# Should not call backend immediately
|
|
self.cached_client.update_node_execution_status("exec_1", "completed")
|
|
self.mock_client.update_node_execution_status.assert_not_called()
|
|
|
|
# Check that it was queued
|
|
cache = get_cache()
|
|
_, statuses = cache.get_pending_updates()
|
|
self.assertEqual(len(statuses), 1)
|
|
self.assertEqual(statuses[0]["node_exec_id"], "exec_1")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|