diff options
Diffstat (limited to 'servers/taskwarrior/tests')
| -rw-r--r-- | servers/taskwarrior/tests/test_server.py | 29 |
1 files changed, 26 insertions, 3 deletions
diff --git a/servers/taskwarrior/tests/test_server.py b/servers/taskwarrior/tests/test_server.py index f96f817..97f7268 100644 --- a/servers/taskwarrior/tests/test_server.py +++ b/servers/taskwarrior/tests/test_server.py @@ -120,7 +120,7 @@ class TestTaskWarriorServer: @pytest.mark.asyncio async def test_done_task(self, taskwarrior_server): - """Test marking a task as done.""" + """Test marking a single task as done.""" mock_output = "Completed task test-uuid-123." with patch.object( @@ -128,9 +128,32 @@ class TestTaskWarriorServer: new_callable=AsyncMock, return_value=(mock_output, '', 0) ): - result = await taskwarrior_server.done_task("test-uuid-123") - assert result["uuid"] == "test-uuid-123" + result = await taskwarrior_server.done_task(["test-uuid-123"]) + assert result["uuids"] == ["test-uuid-123"] + assert result["status"] == "completed" + + @pytest.mark.asyncio + async def test_done_task_multiple(self, taskwarrior_server): + """Test marking multiple tasks as done.""" + mock_output = "Completed task test-uuid-123.\nCompleted task test-uuid-456." + + with patch.object( + taskwarrior_server, '_run_task_command', + new_callable=AsyncMock, + return_value=(mock_output, '', 0) + ): + result = await taskwarrior_server.done_task(["test-uuid-123", "test-uuid-456"]) + assert result["uuids"] == ["test-uuid-123", "test-uuid-456"] assert result["status"] == "completed" + assert len(result["uuids"]) == 2 + + @pytest.mark.asyncio + async def test_done_task_empty_list(self, taskwarrior_server): + """Test that marking tasks with empty list raises ValueError.""" + with pytest.raises(ValueError) as exc_info: + await taskwarrior_server.done_task([]) + + assert "At least one UUID is required" in str(exc_info.value) @pytest.mark.asyncio async def test_delete_task(self, taskwarrior_server): |