Recovered tests

This commit is contained in:
2025-12-06 23:55:21 +01:00
parent 9ca31e45e0
commit 5b71233fb0
11 changed files with 3420 additions and 555 deletions
+8 -30
View File
@@ -1,8 +1,7 @@
"""Edge case tests for tools."""
from unittest.mock import Mock, patch
import pytest
from unittest.mock import Mock, patch, MagicMock
from pathlib import Path
from agent.tools import api as api_tools
from agent.tools import filesystem as fs_tools
@@ -16,10 +15,7 @@ class TestFindTorrentEdgeCases:
def test_empty_query(self, mock_use_case_class, memory):
"""Should handle empty query."""
mock_response = Mock()
mock_response.to_dict.return_value = {
"status": "error",
"error": "invalid_query",
}
mock_response.to_dict.return_value = {"status": "error", "error": "invalid_query"}
mock_use_case = Mock()
mock_use_case.execute.return_value = mock_response
mock_use_case_class.return_value = mock_use_case
@@ -32,11 +28,7 @@ class TestFindTorrentEdgeCases:
def test_very_long_query(self, mock_use_case_class, memory):
"""Should handle very long query."""
mock_response = Mock()
mock_response.to_dict.return_value = {
"status": "ok",
"torrents": [],
"count": 0,
}
mock_response.to_dict.return_value = {"status": "ok", "torrents": [], "count": 0}
mock_use_case = Mock()
mock_use_case.execute.return_value = mock_response
mock_use_case_class.return_value = mock_use_case
@@ -51,11 +43,7 @@ class TestFindTorrentEdgeCases:
def test_special_characters_in_query(self, mock_use_case_class, memory):
"""Should handle special characters in query."""
mock_response = Mock()
mock_response.to_dict.return_value = {
"status": "ok",
"torrents": [],
"count": 0,
}
mock_response.to_dict.return_value = {"status": "ok", "torrents": [], "count": 0}
mock_use_case = Mock()
mock_use_case.execute.return_value = mock_response
mock_use_case_class.return_value = mock_use_case
@@ -69,11 +57,7 @@ class TestFindTorrentEdgeCases:
def test_unicode_query(self, mock_use_case_class, memory):
"""Should handle unicode in query."""
mock_response = Mock()
mock_response.to_dict.return_value = {
"status": "ok",
"torrents": [],
"count": 0,
}
mock_response.to_dict.return_value = {"status": "ok", "torrents": [], "count": 0}
mock_use_case = Mock()
mock_use_case.execute.return_value = mock_response
mock_use_case_class.return_value = mock_use_case
@@ -177,10 +161,7 @@ class TestAddTorrentEdgeCases:
def test_empty_magnet_link(self, mock_use_case_class, memory):
"""Should handle empty magnet link."""
mock_response = Mock()
mock_response.to_dict.return_value = {
"status": "error",
"error": "empty_magnet",
}
mock_response.to_dict.return_value = {"status": "error", "error": "empty_magnet"}
mock_use_case = Mock()
mock_use_case.execute.return_value = mock_response
mock_use_case_class.return_value = mock_use_case
@@ -345,10 +326,7 @@ class TestFilesystemEdgeCases:
for attempt in attempts:
result = fs_tools.list_folder("download", attempt)
# Should either be forbidden or not found
assert (
result.get("error") in ["forbidden", "not_found", None]
or result.get("status") == "ok"
)
assert result.get("error") in ["forbidden", "not_found", None] or result.get("status") == "ok"
def test_path_with_null_byte(self, memory, real_folder):
"""Should block null byte injection."""