def test_create_subtask_if_another_worker_task(self): w = Worker.objects.create( worker_key='other-worker', name='other-worker', ) t = Task.objects.create( worker=w, arch=self._arch, channel=self._channel, owner=self._user, state=TASK_STATES['FREE'], ) req = _make_request(self._worker) with self.assertRaises(Task.DoesNotExist): worker.create_subtask(req, '', '', None, t.id)
def test_create_subtask(self): t_parent = Task.objects.create( worker=self._worker, arch=self._arch, channel=self._channel, owner=self._user, state=TASK_STATES['FREE'], ) req = _make_request(self._worker) task_id = worker.create_subtask(req, 'Label', 'Method', None, t_parent.id) self.assertTrue(task_id > 0) t_child = Task.objects.get(id=task_id) self.assertEqual(t_child.parent.id, t_parent.id) self.assertEqual(t_child.label, 'Label') self.assertEqual(t_child.method, 'Method')
def create_subtask(self, label, method, args, parent_id): return worker.create_subtask(self._request, label, method, args, parent_id)
def test_create_subtask(self): with self.assertRaises(PermissionDenied): worker.create_subtask(_make_request(None, False), '', '', None, 1)