set execution_node right when jobs are accepted

This commit is contained in:
AlanCoding
2017-09-20 09:41:19 -04:00
parent c868f7e91f
commit 892ca98709
3 changed files with 8 additions and 8 deletions

View File

@@ -259,7 +259,7 @@ class TestGenericRun(TestJobExecution):
with pytest.raises(Exception):
self.task.run(self.pk)
for c in [
mock.call(self.pk, status='running'),
mock.call(self.pk, execution_node=settings.CLUSTER_HOST_ID, status='running'),
mock.call(self.pk, output_replacements=[], result_traceback=mock.ANY, status='canceled')
]:
assert c in self.task.update_model.call_args_list