mirror of https://github.com/aliasrobotics/cai.git
51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
import os
|
|
import pytest
|
|
import json
|
|
from unittest.mock import MagicMock
|
|
|
|
# Set test environment variables to avoid OpenAI client initialization errors
|
|
os.environ["OPENAI_API_KEY"] = "test_key_for_ci_environment"
|
|
|
|
from cai.tools.reconnaissance.generic_linux_command import generic_linux_command
|
|
|
|
async def test_generic_linux_command_regular_commands():
|
|
"""Test the execution of a regular command using the generic Linux command tool."""
|
|
mock_ctx = MagicMock() # Create a mock context for the command execution
|
|
params = {
|
|
"command": "echo", # Command to be executed
|
|
"args": "'hello'" # Arguments for the command
|
|
}
|
|
|
|
# Invoke the tool with the specified parameters and await the result
|
|
result = await generic_linux_command.on_invoke_tool(mock_ctx, json.dumps(params))
|
|
|
|
# Assert that the result matches the expected output
|
|
assert result.replace("\n", "") == 'hello'
|
|
|
|
async def test_generic_linux_command_ls():
|
|
"""Test the execution of the 'ls' command using the generic Linux command tool."""
|
|
mock_ctx = MagicMock() # Create a mock context for the command execution
|
|
params = {
|
|
"command": "ls", # Command to be executed
|
|
"args": "-l" # Arguments for the command
|
|
}
|
|
|
|
# Invoke the tool with the specified parameters and await the result
|
|
result = await generic_linux_command.on_invoke_tool(mock_ctx, json.dumps(params))
|
|
|
|
# Assert that the output contains 'total', which is typical for 'ls -l'
|
|
assert "total" in result
|
|
|
|
async def test_generic_linux_command_invalid_command():
|
|
"""Test the handling of an invalid command using the generic Linux command tool."""
|
|
mock_ctx = MagicMock() # Create a mock context for the command execution
|
|
params = {
|
|
"command": "invalid_command", # Invalid command to be executed
|
|
"args": "" # No arguments for the command
|
|
}
|
|
|
|
# Invoke the tool with the specified parameters and await the result
|
|
result = await generic_linux_command.on_invoke_tool(mock_ctx, json.dumps(params))
|
|
|
|
# Assert that the result indicates the command was not found
|
|
assert "not found" in result |