todo-agent 0.2.9__tar.gz → 0.3.1__tar.gz

{todo_agent-0.2.9 → todo_agent-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: todo-agent
-Version: 0.2.9
+Version: 0.3.1
 Summary: A natural language interface for todo.sh task management
 Author: codeprimate
 Maintainer: codeprimate

{todo_agent-0.2.9 → todo_agent-0.3.1}/tests/test_core/test_todo_manager.py

@@ -3,10 +3,9 @@ Tests for TodoManager.
 """
 
 import unittest
-from unittest.mock import Mock
+from unittest.mock import Mock, patch
 
 import pytest
-from unittest.mock import patch
 
 try:
     from todo_agent.core.todo_manager import TodoManager
@@ -187,6 +186,36 @@ class TestTodoManager(unittest.TestCase):
             self.todo_manager.add_task("Test task", recurring="rec:weekly:0")
         self.assertIn("Must be a positive integer", str(context.exception))
 
+    def test_add_task_with_invalid_duration_format(self):
+        """Test adding a task with invalid duration format raises ValueError."""
+        with self.assertRaises(ValueError) as context:
+            self.todo_manager.add_task("Test task", duration="30s")
+        self.assertIn("Invalid duration format", str(context.exception))
+
+    def test_add_task_with_invalid_duration_unit(self):
+        """Test adding a task with invalid duration unit raises ValueError."""
+        with self.assertRaises(ValueError) as context:
+            self.todo_manager.add_task("Test task", duration="30s")
+        self.assertIn("Invalid duration format", str(context.exception))
+
+    def test_add_task_with_invalid_duration_value(self):
+        """Test adding a task with invalid duration value raises ValueError."""
+        with self.assertRaises(ValueError) as context:
+            self.todo_manager.add_task("Test task", duration="0m")
+        self.assertIn("Must be a positive number", str(context.exception))
+
+    def test_add_task_with_negative_duration(self):
+        """Test adding a task with negative duration raises ValueError."""
+        with self.assertRaises(ValueError) as context:
+            self.todo_manager.add_task("Test task", duration="-30m")
+        self.assertIn("Must be a positive number", str(context.exception))
+
+    def test_add_task_with_empty_duration(self):
+        """Test adding a task with empty duration raises ValueError."""
+        with self.assertRaises(ValueError) as context:
+            self.todo_manager.add_task("Test task", duration="")
+        self.assertIn("Duration must be a non-empty string", str(context.exception))
+
     def test_add_task_with_all_parameters_including_recurring(self):
         """Test adding a task with all parameters including recurring."""
         self.todo_shell.add.return_value = "Test task"
@@ -205,6 +234,36 @@ class TestTodoManager(unittest.TestCase):
             "(A) Test task +work @office due:2024-01-15 rec:daily"
         )
 
+    def test_add_task_with_duration(self):
+        """Test adding a task with duration parameter."""
+        self.todo_shell.add.return_value = "Test task"
+        result = self.todo_manager.add_task(
+            "Test task",
+            duration="30m",
+        )
+        self.assertEqual(result, "Added task: Test task duration:30m")
+        self.todo_shell.add.assert_called_once_with("Test task duration:30m")
+
+    def test_add_task_with_all_parameters_including_duration(self):
+        """Test adding a task with all parameters including duration."""
+        self.todo_shell.add.return_value = "Test task"
+        result = self.todo_manager.add_task(
+            "Test task",
+            priority="A",
+            project="work",
+            context="office",
+            due="2024-01-15",
+            recurring="rec:daily",
+            duration="2h",
+        )
+        self.assertEqual(
+            result,
+            "Added task: (A) Test task +work @office due:2024-01-15 rec:daily duration:2h",
+        )
+        self.todo_shell.add.assert_called_once_with(
+            "(A) Test task +work @office due:2024-01-15 rec:daily duration:2h"
+        )
+
     def test_list_tasks(self):
         """Test listing tasks."""
         self.todo_shell.list_tasks.return_value = "1. Task 1\n2. Task 2"
@@ -299,60 +358,62 @@ class TestTodoManager(unittest.TestCase):
         self.assertEqual(result, "No completed tasks found matching the criteria.")
         self.todo_shell.list_completed.assert_called_once_with("+nonexistent")
 
-
-
     def test_add_task_sanitizes_inputs(self):
         """Test that add_task sanitizes inputs to prevent duplicates."""
         # Mock the todo_shell
         mock_shell = Mock()
         mock_shell.add.return_value = "1 Test task +project1 @context1"
-        
+
         manager = TodoManager(mock_shell)
-        
+
         # Test with inputs that already have + and @ symbols
         result = manager.add_task(
             description="Test task",
             project="+project1",  # Already has + symbol
-            context="@context1"   # Already has @ symbol
+            context="@context1",  # Already has @ symbol
         )
-        
+
         # Verify the task was added with properly formatted projects and contexts
         mock_shell.add.assert_called_once()
         call_args = mock_shell.add.call_args[0][0]
-        
+
         # Should have exactly one instance of each (sanitization prevents duplicates)
         assert call_args.count("+project1") == 1
         assert call_args.count("@context1") == 1
-        
+
         # Verify the result message
         assert "Added task:" in result
 
-
-
     def test_parse_task_components_deduplicates(self):
         """Test that _parse_task_components deduplicates projects and contexts."""
         from todo_agent.infrastructure.todo_shell import TodoShell
-        
+
         # Create a TodoShell instance
         shell = TodoShell("/tmp/todo.txt")
-        
+
         # Test with duplicate projects and contexts in the task line
         task_line = "1 (A) Task description +project1 +project1 +project2 @context1 @context1 @context2"
         components = shell._parse_task_components(task_line)
-        
+
         # Verify deduplication
-        assert components["projects"] == ["+project1", "+project2"]  # Sorted and deduplicated
-        assert components["contexts"] == ["@context1", "@context2"]  # Sorted and deduplicated
+        assert components["projects"] == [
+            "+project1",
+            "+project2",
+        ]  # Sorted and deduplicated
+        assert components["contexts"] == [
+            "@context1",
+            "@context2",
+        ]  # Sorted and deduplicated
         assert components["priority"] == "A"
         assert components["description"] == "Task description"
 
     def test_reconstruct_task_maintains_deduplication(self):
         """Test that _reconstruct_task maintains deduplication."""
         from todo_agent.infrastructure.todo_shell import TodoShell
-        
+
         # Create a TodoShell instance
         shell = TodoShell("/tmp/todo.txt")
-        
+
         # Create components with potential duplicates
         components = {
             "priority": "A",
@@ -361,12 +422,12 @@ class TestTodoManager(unittest.TestCase):
             "contexts": ["@context1", "@context1", "@context2"],  # Duplicate context1
             "due": "2024-01-01",
             "recurring": None,
-            "other_tags": []
+            "other_tags": [],
         }
-        
+
         # Reconstruct the task
         reconstructed = shell._reconstruct_task(components)
-        
+
         # Verify that duplicates are preserved in reconstruction (since we handle deduplication in parsing)
         # This is expected behavior - reconstruction should preserve the exact components given
         assert reconstructed.count("+project1") == 2
@@ -376,83 +437,96 @@ class TestTodoManager(unittest.TestCase):
 
     def test_todo_shell_set_project_actually_prevents_duplicates(self):
         """Test that the todo_shell set_project method actually prevents duplicates."""
-        from todo_agent.infrastructure.todo_shell import TodoShell
         from unittest.mock import patch
-        
+
+        from todo_agent.infrastructure.todo_shell import TodoShell
+
         # Create a TodoShell instance
         shell = TodoShell("/tmp/todo.txt")
-        
+
         # Mock the list_tasks method to return a task with existing projects
-        with patch.object(shell, 'list_tasks') as mock_list_tasks:
-            mock_list_tasks.return_value = "1 (A) Existing task +existing_project @existing_context"
-            
+        with patch.object(shell, "list_tasks") as mock_list_tasks:
+            mock_list_tasks.return_value = (
+                "1 (A) Existing task +existing_project @existing_context"
+            )
+
             # Mock the replace method to capture what gets called
-            with patch.object(shell, 'replace') as mock_replace:
-                mock_replace.return_value = "1 (A) Existing task +existing_project @existing_context"
-                
+            with patch.object(shell, "replace") as mock_replace:
+                mock_replace.return_value = (
+                    "1 (A) Existing task +existing_project @existing_context"
+                )
+
                 # Try to add a project that already exists
                 result = shell.set_project(1, ["existing_project"])
-                
+
                 # Since the project already exists, replace should NOT be called
                 # (our deduplication logic prevents unnecessary updates)
                 mock_replace.assert_not_called()
-                
+
                 # The result should be the reconstructed task without duplicates
                 assert "+existing_project" in result
                 assert result.count("+existing_project") == 1  # Only one instance
 
     def test_todo_shell_set_context_actually_prevents_duplicates(self):
         """Test that the todo_shell set_context method actually prevents duplicates."""
-        from todo_agent.infrastructure.todo_shell import TodoShell
         from unittest.mock import patch
-        
+
+        from todo_agent.infrastructure.todo_shell import TodoShell
+
         # Create a TodoShell instance
         shell = TodoShell("/tmp/todo.txt")
-        
+
         # Mock the list_tasks method to return a task with existing context
-        with patch.object(shell, 'list_tasks') as mock_list_tasks:
-            mock_list_tasks.return_value = "1 (A) Existing task +existing_project @existing_context"
-            
+        with patch.object(shell, "list_tasks") as mock_list_tasks:
+            mock_list_tasks.return_value = (
+                "1 (A) Existing task +existing_project @existing_context"
+            )
+
             # Mock the replace method to capture what gets called
-            with patch.object(shell, 'replace') as mock_replace:
-                mock_replace.return_value = "1 (A) Existing task +existing_project @existing_context"
-                
+            with patch.object(shell, "replace") as mock_replace:
+                mock_replace.return_value = (
+                    "1 (A) Existing task +existing_project @existing_context"
+                )
+
                 # Try to set a context that already exists
                 result = shell.set_context(1, "existing_context")
-                
+
                 # Since the context already exists, replace should NOT be called
                 # (our deduplication logic prevents unnecessary updates)
                 mock_replace.assert_not_called()
-                
+
                 # The result should be the reconstructed task without duplicates
                 assert "@existing_context" in result
                 assert result.count("@existing_context") == 1  # Only one instance
 
     def test_todo_shell_set_project_adds_new_projects(self):
         """Test that the todo_shell set_project method adds new projects and prevents duplicates."""
-        from todo_agent.infrastructure.todo_shell import TodoShell
         from unittest.mock import patch
-        
+
+        from todo_agent.infrastructure.todo_shell import TodoShell
+
         # Create a TodoShell instance
         shell = TodoShell("/tmp/todo.txt")
-        
+
         # Mock the list_tasks method to return a task with existing projects
-        with patch.object(shell, 'list_tasks') as mock_list_tasks:
-            mock_list_tasks.return_value = "1 (A) Existing task +existing_project @existing_context"
-            
+        with patch.object(shell, "list_tasks") as mock_list_tasks:
+            mock_list_tasks.return_value = (
+                "1 (A) Existing task +existing_project @existing_context"
+            )
+
             # Mock the replace method to capture what gets called
-            with patch.object(shell, 'replace') as mock_replace:
+            with patch.object(shell, "replace") as mock_replace:
                 mock_replace.return_value = "1 (A) Existing task +existing_project +new_project @existing_context"
-                
+
                 # Try to add a new project
-                result = shell.set_project(1, ["new_project"])
-                
+                shell.set_project(1, ["new_project"])
+
                 # Since we're adding a new project, replace should be called
                 mock_replace.assert_called_once()
-                
+
                 # Get the actual task description that was passed to replace
                 actual_task_description = mock_replace.call_args[0][1]
-                
+
                 # Verify that the new project was added and no duplicates exist
                 assert "+new_project" in actual_task_description
                 assert actual_task_description.count("+existing_project") == 1
@@ -460,56 +534,65 @@ class TestTodoManager(unittest.TestCase):
 
     def test_todo_shell_set_context_adds_new_contexts(self):
         """Test that the todo_shell set_context method adds new contexts and prevents duplicates."""
-        from todo_agent.infrastructure.todo_shell import TodoShell
         from unittest.mock import patch
-        
+
+        from todo_agent.infrastructure.todo_shell import TodoShell
+
         # Create a TodoShell instance
         shell = TodoShell("/tmp/todo.txt")
-        
+
         # Mock the list_tasks method to return a task with existing context
-        with patch.object(shell, 'list_tasks') as mock_list_tasks:
-            mock_list_tasks.return_value = "1 (A) Existing task +existing_project @existing_context"
-            
+        with patch.object(shell, "list_tasks") as mock_list_tasks:
+            mock_list_tasks.return_value = (
+                "1 (A) Existing task +existing_project @existing_context"
+            )
+
             # Mock the replace method to capture what gets called
-            with patch.object(shell, 'replace') as mock_replace:
-                mock_replace.return_value = "1 (A) Existing task +existing_project @new_context"
-                
+            with patch.object(shell, "replace") as mock_replace:
+                mock_replace.return_value = (
+                    "1 (A) Existing task +existing_project @new_context"
+                )
+
                 # Try to set a new context
-                result = shell.set_context(1, "new_context")
-                
+                shell.set_context(1, "new_context")
+
                 # Since we're adding a new context, replace should be called
                 mock_replace.assert_called_once()
-                
+
                 # Get the actual task description that was passed to replace
                 actual_task_description = mock_replace.call_args[0][1]
-                
+
                 # Verify that the new context was added and the old context was replaced
                 # (set_context replaces all contexts with the new one)
                 assert "@new_context" in actual_task_description
-                assert "@existing_context" not in actual_task_description  # Old context replaced
-                assert actual_task_description.count("@new_context") == 1  # Only one instance
+                assert (
+                    "@existing_context" not in actual_task_description
+                )  # Old context replaced
+                assert (
+                    actual_task_description.count("@new_context") == 1
+                )  # Only one instance
 
     def test_parse_task_components_actually_deduplicates(self):
         """Test that _parse_task_components actually removes duplicates from input."""
         from todo_agent.infrastructure.todo_shell import TodoShell
-        
+
         # Create a TodoShell instance
         shell = TodoShell("/tmp/todo.txt")
-        
+
         # Test with a task line that has multiple duplicates
         task_line = "1 (A) Task description +project1 +project1 +project1 +project2 @context1 @context1 @context2"
         components = shell._parse_task_components(task_line)
-        
+
         # Verify that duplicates were actually removed
         assert len(components["projects"]) == 2  # Should only have 2 unique projects
         assert len(components["contexts"]) == 2  # Should only have 2 unique contexts
-        
+
         # Verify the specific projects and contexts
         assert "+project1" in components["projects"]
         assert "+project2" in components["projects"]
         assert "@context1" in components["contexts"]
         assert "@context2" in components["contexts"]
-        
+
         # Verify no duplicates exist in the lists
         assert components["projects"].count("+project1") == 1
         assert components["projects"].count("+project2") == 1

{todo_agent-0.2.9 → todo_agent-0.3.1}/tests/test_infrastructure/test_todo_shell.py

@@ -219,7 +219,7 @@ class TestTodoShell:
         sample_task = "1 (B) Call dentist +health @phone"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_due_date(1, "2025-01-20")
@@ -236,7 +236,7 @@ class TestTodoShell:
         sample_task = "1 (C) Review quarterly report +work @office rec:weekly due:2025-01-10 custom:tag"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_due_date(1, "2025-01-25")
@@ -253,7 +253,7 @@ class TestTodoShell:
         sample_task = "1 Buy milk +shopping @grocery"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_due_date(1, "2025-01-30")
@@ -268,9 +268,7 @@ class TestTodoShell:
         """Test that set_due_date raises error for invalid task number."""
         # Mock the list_tasks to return only one task
         sample_task = "1 Test task"
-        with patch.object(
-            self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list:
+        with patch.object(self.todo_shell, "list_tasks", return_value=sample_task):
             with pytest.raises(TodoShellError, match="Task number 5 not found"):
                 self.todo_shell.set_due_date(5, "2025-01-15")
 
@@ -280,7 +278,7 @@ class TestTodoShell:
         sample_task = "1 (A) Buy groceries +shopping @home due:2025-01-10"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_due_date(1, "")
@@ -295,7 +293,7 @@ class TestTodoShell:
         sample_task = "1 (B) Call dentist +health @phone due:2025-01-15"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_due_date(1, "   ")
@@ -310,7 +308,7 @@ class TestTodoShell:
         sample_task = "1 Buy milk +shopping @grocery due:2025-01-20"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_due_date(1, "")
@@ -344,7 +342,7 @@ class TestTodoShell:
         sample_task = "1 (B) Call dentist +health"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_context(1, "phone")
@@ -359,7 +357,7 @@ class TestTodoShell:
         sample_task = "1 (C) Review quarterly report +work @office rec:weekly due:2025-01-10 custom:tag"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_context(1, "home")
@@ -375,7 +373,7 @@ class TestTodoShell:
         sample_task = "1 Buy milk +shopping @grocery"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_context(1, "store")
@@ -390,7 +388,7 @@ class TestTodoShell:
         sample_task = "1 (A) Buy groceries +shopping @home due:2025-01-10"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_context(1, "")
@@ -407,7 +405,7 @@ class TestTodoShell:
         sample_task = "1 (B) Call dentist +health @phone"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_context(1, "   ")
@@ -422,7 +420,7 @@ class TestTodoShell:
         sample_task = "1 Test task +project"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_context(1, "@office")
@@ -435,9 +433,7 @@ class TestTodoShell:
         """Test that set_context raises error for context that becomes empty after cleaning."""
         # Mock the list_tasks to return a task
         sample_task = "1 Test task"
-        with patch.object(
-            self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list:
+        with patch.object(self.todo_shell, "list_tasks", return_value=sample_task):
             with pytest.raises(TodoShellError, match="Context name cannot be empty"):
                 self.todo_shell.set_context(1, "@")
 
@@ -445,9 +441,7 @@ class TestTodoShell:
         """Test that set_context raises error for invalid task number."""
         # Mock the list_tasks to return only one task
         sample_task = "1 Test task"
-        with patch.object(
-            self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list:
+        with patch.object(self.todo_shell, "list_tasks", return_value=sample_task):
             with pytest.raises(TodoShellError, match="Task number 5 not found"):
                 self.todo_shell.set_context(5, "office")
 
@@ -590,7 +584,7 @@ class TestTodoShell:
         sample_task = "1 (A) Buy groceries @home due:2025-01-10"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, ["shopping", "errands"])
@@ -607,7 +601,7 @@ class TestTodoShell:
         sample_task = "1 (B) Call dentist +health @phone"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, ["appointment", "personal"])
@@ -624,7 +618,7 @@ class TestTodoShell:
         sample_task = "1 (C) Review report +work +urgent +review @office"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, ["-urgent", "-review"])
@@ -639,7 +633,7 @@ class TestTodoShell:
         sample_task = "1 (A) Task with projects +old +keep @context"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, ["-old", "new", "another"])
@@ -656,7 +650,7 @@ class TestTodoShell:
         sample_task = "1 (B) Test task +existing @context"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, [])
@@ -672,7 +666,7 @@ class TestTodoShell:
         sample_task = "1 (C) Test task +existing @context"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, ["", "valid", "  ", "another"])
@@ -689,7 +683,7 @@ class TestTodoShell:
         sample_task = "1 Test task @context"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, ["+work", "+home"])
@@ -704,7 +698,7 @@ class TestTodoShell:
         sample_task = "1 Test task +work +home +shopping @context"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, ["-+work", "-+shopping"])
@@ -716,10 +710,12 @@ class TestTodoShell:
     def test_set_project_preserves_all_other_components(self):
         """Test that set_project preserves all task components when modifying projects."""
         # Mock the list_tasks to return a complex task
-        sample_task = "1 (A) Complex task +old +keep @office rec:weekly due:2025-01-15 custom:tag"
+        sample_task = (
+            "1 (A) Complex task +old +keep @office rec:weekly due:2025-01-15 custom:tag"
+        )
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, ["-old", "new"])
@@ -733,9 +729,7 @@ class TestTodoShell:
         """Test that set_project raises error for project that becomes empty after cleaning."""
         # Mock the list_tasks to return a task
         sample_task = "1 Test task"
-        with patch.object(
-            self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list:
+        with patch.object(self.todo_shell, "list_tasks", return_value=sample_task):
             with pytest.raises(TodoShellError, match="Project name cannot be empty"):
                 self.todo_shell.set_project(1, ["+"])
 
@@ -743,9 +737,7 @@ class TestTodoShell:
         """Test that set_project raises error for remove project that becomes empty after cleaning."""
         # Mock the list_tasks to return a task
         sample_task = "1 Test task"
-        with patch.object(
-            self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list:
+        with patch.object(self.todo_shell, "list_tasks", return_value=sample_task):
             with pytest.raises(TodoShellError, match="Project name cannot be empty"):
                 self.todo_shell.set_project(1, ["-+"])
 
@@ -753,9 +745,7 @@ class TestTodoShell:
         """Test that set_project raises error for invalid task number."""
         # Mock the list_tasks to return only one task
         sample_task = "1 Test task"
-        with patch.object(
-            self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list:
+        with patch.object(self.todo_shell, "list_tasks", return_value=sample_task):
             with pytest.raises(TodoShellError, match="Task number 5 not found"):
                 self.todo_shell.set_project(5, ["work"])
 
@@ -765,7 +755,7 @@ class TestTodoShell:
         sample_task = "1 Test task +work @context"
         with patch.object(
             self.todo_shell, "list_tasks", return_value=sample_task
-        ) as mock_list, patch.object(
+        ), patch.object(
             self.todo_shell, "replace", return_value="Task updated"
         ) as mock_replace:
             result = self.todo_shell.set_project(1, ["work", "new"])