mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
import os
|
|
import tempfile
|
|
from unittest.mock import patch, MagicMock
|
|
|
|
import pytest
|
|
from click.testing import CliRunner
|
|
|
|
from crewai.cli.cli import reset_memories
|
|
from crewai.cli.reset_memories_command import reset_memories_command
|
|
|
|
|
|
def test_reset_memories_command_parameters():
|
|
"""Test that the CLI parameters match the function parameters."""
|
|
# Create a mock for reset_memories_command
|
|
with patch('crewai.cli.cli.reset_memories_command') as mock_reset:
|
|
runner = CliRunner()
|
|
|
|
# Test with entities flag
|
|
result = runner.invoke(reset_memories, ['--entities'])
|
|
assert result.exit_code == 0
|
|
|
|
# Check that the function was called with the correct parameters
|
|
# The third parameter should be True for entities
|
|
mock_reset.assert_called_once_with(False, False, True, False, False, False)
|
|
|
|
|
|
def test_reset_memories_all_flag():
|
|
"""Test that the --all flag resets all memories."""
|
|
with patch('crewai.cli.cli.reset_memories_command') as mock_reset:
|
|
runner = CliRunner()
|
|
|
|
# Test with all flag
|
|
result = runner.invoke(reset_memories, ['--all'])
|
|
assert result.exit_code == 0
|
|
|
|
# Check that the function was called with the correct parameters
|
|
# The last parameter should be True for all
|
|
mock_reset.assert_called_once_with(False, False, False, False, False, True)
|
|
|
|
|
|
def test_reset_memories_knowledge_flag():
|
|
"""Test that the --knowledge flag resets knowledge storage."""
|
|
with patch('crewai.cli.cli.reset_memories_command') as mock_reset:
|
|
runner = CliRunner()
|
|
|
|
# Test with knowledge flag
|
|
result = runner.invoke(reset_memories, ['--knowledge'])
|
|
assert result.exit_code == 0
|
|
|
|
# Check that the function was called with the correct parameters
|
|
# The fourth parameter should be True for knowledge
|
|
mock_reset.assert_called_once_with(False, False, False, True, False, False)
|
|
|
|
|
|
def test_reset_memories_no_flags():
|
|
"""Test that an error message is shown when no flags are provided."""
|
|
runner = CliRunner()
|
|
|
|
# Test with no flags
|
|
result = runner.invoke(reset_memories, [])
|
|
assert result.exit_code == 0
|
|
assert "Please specify at least one memory type" in result.output
|