gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-10-02 17:35:57 +01:00
parent a607d8f213
commit 96ab005d4e
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/tests/cli_tests/cli_unit_tests/test_cli_edge_cases.py
View file

@ -379,10 +379,15 @@ class TestCognifyCommandEdgeCases:
class TestDeleteCommandEdgeCases:
"""Test edge cases for DeleteCommand"""
@patch("cognee.cli.commands.delete_command.get_deletion_counts")
@patch("cognee.cli.commands.delete_command.fmt.confirm")
@patch("cognee.cli.commands.delete_command.asyncio.run", side_effect=_mock_run)
def test_delete_all_with_user_id(self, mock_asyncio_run, mock_confirm):
def test_delete_all_with_user_id(
self, mock_asyncio_run, mock_confirm, mock_get_deletion_counts
):
"""Test delete command with both --all and --user-id"""
mock_get_deletion_counts.return_value = DeletionCountsPreview()
# Mock the cognee module
mock_cognee = MagicMock()
mock_cognee.delete = AsyncMock()