server_test.py: Fix crash in test_save_state (#1977)

For whatever reasons its not running on travis but crashing when you locally run the tests for luigi.
spotify · Jan 4, 2017 · 1afb37a · 1afb37a
1 parent 6cc7619
commit 1afb37a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/server_test.py b/test/server_test.py
@@ -160,12 +160,12 @@ def test_404(self):
 
     @skipOnTravis('https://travis-ci.org/spotify/luigi/jobs/72953884')
     def test_save_state(self):
-        self.sch.add_task('X', 'B', deps=('A',))
-        self.sch.add_task('X', 'A')
-        self.assertEqual(self.sch.get_work('X')['task_id'], 'A')
+        self.sch.add_task(worker='X', task_id='B', deps=('A',))
+        self.sch.add_task(worker='X', task_id='A')
+        self.assertEqual(self.sch.get_work(worker='X')['task_id'], 'A')
         self.stop_server()
         self.start_server()
-        work = self.sch.get_work('X')['running_tasks'][0]
+        work = self.sch.get_work(worker='X')['running_tasks'][0]
         self.assertEqual(work['task_id'], 'A')