X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/blobdiff_plain/e9ee75194dcb6c60ff6b204f6b6ad135d4859628..b9539265b8094c49d02317f97fd2cabb585e69a8:/tasklib/tests.py

diff --git a/tasklib/tests.py b/tasklib/tests.py
index 5ebea13..ea74613 100644
--- a/tasklib/tests.py
+++ b/tasklib/tests.py
@@ -535,12 +535,13 @@ class TaskFromHookTest(TasklibTest):
          '"description":"test task"}')
 
     def test_setting_up_from_add_hook_input(self):
-        t = Task.from_input(input_file=self.input_add_data)
+        t = Task.from_input(input_file=self.input_add_data, warrior=self.tw)
         self.assertEqual(t['description'], "Buy some milk")
         self.assertEqual(t.pending, True)
 
     def test_setting_up_from_modified_hook_input(self):
-        t = Task.from_input(input_file=self.input_modify_data, modify=True)
+        t = Task.from_input(input_file=self.input_modify_data, modify=True,
+                            warrior=self.tw)
         self.assertEqual(t['description'], "Buy some milk finally")
         self.assertEqual(t.pending, False)
         self.assertEqual(t.completed, True)