From 3572ecf1c70c94eb944a6ea28de1c933ab0cb063 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 23 Apr 2025 22:06:10 +0000 Subject: [PATCH] Fix duplicate test_reset_knowledge_memories function Co-Authored-By: Joe Moura --- tests/cli/cli_test.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/cli/cli_test.py b/tests/cli/cli_test.py index 9ea3cb955..c0367f421 100644 --- a/tests/cli/cli_test.py +++ b/tests/cli/cli_test.py @@ -132,10 +132,12 @@ def test_reset_knowledge(mock_get_crew, runner): assert result.output == "Knowledge has been reset.\n" -@mock.patch("crewai.knowledge.storage.knowledge_storage.KnowledgeStorage") -def test_reset_knowledge_memories(MockKnowledgeStorage, runner): +@mock.patch("crewai.cli.reset_memories_command.get_crew") +def test_reset_knowledge_with_kn_flag(mock_get_crew, runner): + mock_crew = mock.Mock() + mock_get_crew.return_value = mock_crew result = runner.invoke(reset_memories, ["-kn"]) - MockKnowledgeStorage().reset.assert_called_once() + mock_crew.reset_memories.assert_called_once_with(command_type="knowledge") assert result.output == "Knowledge has been reset.\n"