Increase security and reformat

core/tests/test_mcp_tools.py

@@ -7,13 +7,13 @@ from django.test import TestCase, override_settings
 from core.mcp.tools import execute_tool, tool_specs
 from core.models import (
     AIRequest,
+    DerivedTask,
+    DerivedTaskEvent,
     MCPToolAuditLog,
     MemoryItem,
     TaskProject,
     User,
     WorkspaceConversation,
-    DerivedTask,
-    DerivedTaskEvent,
 )
 
 
@@ -80,9 +80,13 @@ class MCPToolTests(TestCase):
         first_hit = (memory_payload.get("hits") or [{}])[0]
         self.assertEqual(str(self.memory.id), str(first_hit.get("memory_id")))
 
-        list_payload = execute_tool("tasks.list", {"user_id": self.user.id, "limit": 10})
+        list_payload = execute_tool(
+            "tasks.list", {"user_id": self.user.id, "limit": 10}
+        )
         self.assertEqual(1, int(list_payload.get("count") or 0))
-        self.assertEqual(str(self.task.id), str((list_payload.get("items") or [{}])[0].get("id")))
+        self.assertEqual(
+            str(self.task.id), str((list_payload.get("items") or [{}])[0].get("id"))
+        )
 
         search_payload = execute_tool(
             "tasks.search",
@@ -90,9 +94,13 @@ class MCPToolTests(TestCase):
         )
         self.assertEqual(1, int(search_payload.get("count") or 0))
 
-        events_payload = execute_tool("tasks.events", {"task_id": str(self.task.id), "limit": 5})
+        events_payload = execute_tool(
+            "tasks.events", {"task_id": str(self.task.id), "limit": 5}
+        )
         self.assertEqual(1, int(events_payload.get("count") or 0))
-        self.assertEqual("created", str((events_payload.get("items") or [{}])[0].get("event_type")))
+        self.assertEqual(
+            "created", str((events_payload.get("items") or [{}])[0].get("event_type"))
+        )
 
     def test_memory_proposal_review_flow(self):
         propose_payload = execute_tool(
@@ -182,7 +190,9 @@ class MCPToolTests(TestCase):
                 "note": "Implemented wiki tooling.",
             },
         )
-        self.assertEqual("progress", str((note_payload.get("event") or {}).get("event_type")))
+        self.assertEqual(
+            "progress", str((note_payload.get("event") or {}).get("event_type"))
+        )
 
         artifact_payload = execute_tool(
             "tasks.link_artifact",