tests: Test execute command with return_all flag

diff --git a/tasklib/tests.py b/tasklib/tests.py
index 19950a6c83e7fb8e6d1bb191e04ae0d613c7c728..29b88a04f5429abe56be4d444908ddcae10c238b 100644 (file)
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -676,6 +676,16 @@ class TaskTest(TasklibTest):
 
         self.tw.execute_command(['+test', 'mod', 'unified', 'description'])
 
+    def test_return_all_from_executed_command(self):
+        Task(self.tw, description="test task", tags=['test']).save()
+        out, err, rc = self.tw.execute_command(['count'], return_all=True)
+        self.assertEqual(rc, 0)
+
+    def test_return_all_from_failed_executed_command(self):
+        Task(self.tw, description="test task", tags=['test']).save()
+        out, err, rc = self.tw.execute_command(['countinvalid'], return_all=True)
+        self.assertNotEqual(rc, 0)
+
 
 class TaskFromHookTest(TasklibTest):