diff --git a/mayan/apps/task_manager/tests/literals.py b/mayan/apps/task_manager/tests/literals.py index b1656386c0..0c52f6a5ee 100644 --- a/mayan/apps/task_manager/tests/literals.py +++ b/mayan/apps/task_manager/tests/literals.py @@ -4,3 +4,4 @@ from django.utils.translation import ugettext_lazy as _ TEST_QUEUE_LABEL = _('Test queue') TEST_QUEUE_NAME = 'test_queue' +TEST_WORKER_NAME = 'test_worker' diff --git a/mayan/apps/task_manager/tests/test_views.py b/mayan/apps/task_manager/tests/test_views.py index b470568c11..c672c48dd6 100644 --- a/mayan/apps/task_manager/tests/test_views.py +++ b/mayan/apps/task_manager/tests/test_views.py @@ -2,18 +2,25 @@ from __future__ import unicode_literals from mayan.apps.common.tests import GenericViewTestCase -from ..classes import CeleryQueue +from ..classes import Worker, CeleryQueue from ..permissions import permission_task_view -from .literals import TEST_QUEUE_LABEL, TEST_QUEUE_NAME +from .literals import TEST_QUEUE_LABEL, TEST_QUEUE_NAME, TEST_WORKER_NAME -class TaskManagerViewTestCase(GenericViewTestCase): +class TaskManagerTestMixin(object): + def _create_test_queue(self): + self.test_worker = Worker(name=TEST_WORKER_NAME) + self.test_queue = CeleryQueue( + label=TEST_QUEUE_LABEL, name=TEST_QUEUE_NAME, + worker=self.test_worker + ) + + +class TaskManagerViewTestCase(TaskManagerTestMixin, GenericViewTestCase): def setUp(self): super(TaskManagerViewTestCase, self).setUp() - self.test_queue = CeleryQueue( - label=TEST_QUEUE_LABEL, name=TEST_QUEUE_NAME - ) + self._create_test_queue() def _request_active_task_list(self): return self.get(