def test_get_tasks_to_assign_limit_tasks(self): for _ in range(10): Task.objects.create( worker=self._worker, arch=self._arch, channel=self._channel, owner=self._user, state=TASK_STATES['FREE'], ) Task.objects.create( worker=self._worker, arch=self._arch, channel=self._channel, owner=self._user, state=TASK_STATES['ASSIGNED'], exclusive=True, ) Task.objects.create( worker=self._worker, arch=self._arch, channel=self._channel, owner=self._user, state=TASK_STATES['FREE'], awaited=True, ) req = _make_request(self._worker) tasks = worker.get_tasks_to_assign(req) self.assertEqual(len(tasks), 10) self.assertEqual(len([t for t in tasks if t['state'] == TASK_STATES['ASSIGNED'] and t['exclusive']]), 10)
def get_tasks_to_assign(self): return worker.get_tasks_to_assign(self._request, )
def test_get_tasks_to_assign(self): with self.assertRaises(PermissionDenied): worker.get_tasks_to_assign(_make_request(None, False))