minion-code 0.1.0__py3-none-any.whl

tests/test_basic.py

@@ -0,0 +1,20 @@
+"""Basic tests for the minion-code package."""
+
+import sys
+import os
+import pytest
+
+# 添加 minion 框架路径
+sys.path.insert(0, "/Users/femtozheng/python-project/minion1")
+
+from minion_code import __init__
+
+
+def test_package_imports():
+    """Test that the package can be imported."""
+    assert __init__ is not None
+
+
+def test_basic_functionality():
+    """Basic test to ensure CI pipeline works."""
+    assert 1 + 1 == 2

tests/test_readonly_tools.py

@@ -0,0 +1,102 @@
+"""Tests for readonly tools functionality."""
+
+import sys
+
+# 添加 minion 框架路径
+sys.path.insert(0, "/Users/femtozheng/python-project/minion1")
+
+import pytest
+from minion_code.tools import (
+    FileReadTool,
+    FileWriteTool,
+    BashTool,
+    GrepTool,
+    GlobTool,
+    LsTool,
+    PythonInterpreterTool,
+    WebSearchTool,
+    WikipediaSearchTool,
+    VisitWebpageTool,
+    UserInputTool,
+
+)
+
+
+def test_readonly_tools():
+    """测试只读工具的 readonly 属性"""
+    readonly_tools = [
+        FileReadTool(),
+        GrepTool(),
+        GlobTool(),
+        LsTool(),
+        WebSearchTool(),
+        WikipediaSearchTool(),
+        VisitWebpageTool(),
+        UserInputTool(),
+    ]
+
+    for tool in readonly_tools:
+        assert tool.readonly is True, f"{tool.__class__.__name__} 应该是只读工具"
+
+
+def test_readwrite_tools():
+    """测试读写工具的 readonly 属性"""
+    readwrite_tools = [
+        FileWriteTool(),
+        BashTool(),
+        PythonInterpreterTool(),
+    ]
+
+    for tool in readwrite_tools:
+        assert tool.readonly is False, f"{tool.__class__.__name__} 应该是读写工具"
+
+
+def test_web_search_tool():
+    """测试网络搜索工具"""
+    tool = WebSearchTool()
+    result = tool.forward("Python")
+    assert "搜索查询: Python" in result
+    assert "找到" in result
+
+
+def test_wikipedia_search_tool():
+    """测试 Wikipedia 搜索工具"""
+    tool = WikipediaSearchTool()
+    result = tool.forward("Python", "zh")
+    assert "Wikipedia 搜索结果" in result
+    assert "Python" in result
+
+
+def test_visit_webpage_tool():
+    """测试网页访问工具"""
+    tool = VisitWebpageTool()
+    result = tool.forward("https://www.example.com")
+    assert "网页访问结果" in result
+    assert "https://www.example.com" in result
+
+
+
+def test_tool_inheritance():
+    """测试所有工具都正确继承了 BaseTool"""
+    from minion.tools import BaseTool
+
+    tools = [
+        FileReadTool(),
+        FileWriteTool(),
+        BashTool(),
+        GrepTool(),
+        GlobTool(),
+        LsTool(),
+        PythonInterpreterTool(),
+        WebSearchTool(),
+        WikipediaSearchTool(),
+        VisitWebpageTool(),
+        UserInputTool(),
+    ]
+
+    for tool in tools:
+        assert isinstance(tool, BaseTool)
+        assert hasattr(tool, "readonly")
+        assert hasattr(tool, "name")
+        assert hasattr(tool, "description")
+        assert hasattr(tool, "forward")

tests/test_tools.py

@@ -0,0 +1,83 @@
+"""Tests for minion-code tools."""
+
+import sys
+import tempfile
+import os
+from pathlib import Path
+
+# 添加 minion 框架路径
+sys.path.insert(0, "/Users/femtozheng/python-project/minion1")
+
+import pytest
+from minion_code.tools import FileReadTool, FileWriteTool, BashTool, TOOL_MAPPING
+
+
+def test_tool_mapping():
+    """测试工具映射是否正确"""
+    assert "file_read" in TOOL_MAPPING
+    assert "file_write" in TOOL_MAPPING
+    assert "bash" in TOOL_MAPPING
+    assert len(TOOL_MAPPING) == 7  # 应该有7个工具
+
+
+def test_file_read_tool():
+    """测试文件读取工具"""
+    tool = FileReadTool()
+
+    # 创建临时文件
+    with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
+        f.write("Hello\nWorld\nTest")
+        temp_file = f.name
+
+    try:
+        # 测试读取文件
+        result = tool.forward(temp_file)
+        assert "Hello" in result
+        assert "World" in result
+        assert "Test" in result
+        assert "总行数：3" in result
+    finally:
+        os.unlink(temp_file)
+
+
+def test_file_write_tool():
+    """测试文件写入工具"""
+    tool = FileWriteTool()
+
+    with tempfile.TemporaryDirectory() as temp_dir:
+        test_file = Path(temp_dir) / "test.txt"
+        content = "Hello, World!"
+
+        # 测试写入文件
+        result = tool.forward(str(test_file), content)
+        assert "成功写入文件" in result
+        assert str(test_file) in result
+
+        # 验证文件内容
+        assert test_file.exists()
+        assert test_file.read_text() == content
+
+
+def test_bash_tool():
+    """测试 Bash 工具"""
+    tool = BashTool()
+
+    # 测试简单命令
+    result = tool.forward("echo 'Hello World'")
+    assert "Hello World" in result
+    assert "退出码：0" in result
+
+    # 测试危险命令被阻止
+    result = tool.forward("rm -rf /")
+    assert "错误：禁止执行危险命令" in result
+
+
+def test_tool_inheritance():
+    """测试工具是否正确继承了 BaseTool"""
+    from minion.tools import BaseTool
+
+    tool = FileReadTool()
+    assert isinstance(tool, BaseTool)
+    assert hasattr(tool, "forward")
+    assert hasattr(tool, "name")
+    assert hasattr(tool, "description")