Init commit

tests/test_tool_executor.py

@@ -0,0 +1,57 @@
+import pytest
+
+from core.tool_executor import execute_server_tool
+
+
+@pytest.mark.asyncio
+async def test_code_interpreter_simple_expression():
+    result = await execute_server_tool(
+        {
+            "id": "call_ci_ok",
+            "function": {
+                "name": "code_interpreter",
+                "arguments": '{"code":"sum([1, 2, 3]) + 4"}',
+            },
+        }
+    )
+    assert result["status"]["code"] == 200
+    assert result["output"]["result"] == 10
+
+
+@pytest.mark.asyncio
+async def test_code_interpreter_blocks_import_and_io():
+    result = await execute_server_tool(
+        {
+            "id": "call_ci_bad",
+            "function": {
+                "name": "code_interpreter",
+                "arguments": '{"code":"__import__(\\"os\\").system(\\"ls\\")"}',
+            },
+        }
+    )
+    assert result["status"]["code"] == 422
+    assert result["status"]["message"] == "invalid_code"
+
+
+@pytest.mark.asyncio
+async def test_current_time_uses_local_system_clock(monkeypatch):
+    async def _should_not_be_called(_tool_id):
+        raise AssertionError("fetch_tool_resource should not be called for current_time")
+
+    monkeypatch.setattr("core.tool_executor.fetch_tool_resource", _should_not_be_called)
+
+    result = await execute_server_tool(
+        {
+            "id": "call_time_ok",
+            "function": {
+                "name": "current_time",
+                "arguments": "{}",
+            },
+        }
+    )
+
+    assert result["status"]["code"] == 200
+    assert result["status"]["message"] == "ok"
+    assert "local_time" in result["output"]
+    assert "iso" in result["output"]
+    assert "timestamp" in result["output"]