Fix task manager tests

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2019-05-13 13:20:51 -04:00
parent 98157b7083
commit d003818cae
2 changed files with 14 additions and 6 deletions

View File

@@ -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(