Fix task manager tests
Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user