Formatting

tests/test_tools_edge_cases.py

@@ -1,7 +1,8 @@
 """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
@@ -15,7 +16,10 @@ 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
@@ -28,7 +32,11 @@ 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
@@ -43,7 +51,11 @@ 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
@@ -57,7 +69,11 @@ 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
@@ -161,7 +177,10 @@ 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
@@ -326,7 +345,10 @@ 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."""