Cleanup documents test mixins

Signed-off-by: Roberto Rosario <roberto.rosario@mayan-edms.com>
This commit is contained in:
Roberto Rosario
2019-10-24 15:45:09 -04:00
parent 83f876cde9
commit 2a501c537a
3 changed files with 17 additions and 17 deletions

View File

@@ -152,6 +152,20 @@ class DocumentVersionTestMixin(object):
)
class DocumentVersionViewTestMixin(object):
def _request_document_version_list_view(self):
return self.get(
viewname='documents:document_version_list',
kwargs={'pk': self.test_document.pk}
)
def _request_document_version_revert_view(self, document_version):
return self.post(
viewname='documents:document_version_revert',
kwargs={'pk': document_version.pk}
)
class DocumentViewTestMixin(object):
def _create_document_transformation(self):
layer_saved_transformations.add_transformation_to(

View File

@@ -6,21 +6,7 @@ from ..permissions import (
from .base import GenericDocumentViewTestCase
from .literals import TEST_VERSION_COMMENT
from .mixins import DocumentVersionTestMixin
class DocumentVersionViewTestMixin(object):
def _request_document_version_list_view(self):
return self.get(
viewname='documents:document_version_list',
kwargs={'pk': self.test_document.pk}
)
def _request_document_version_revert_view(self, document_version):
return self.post(
viewname='documents:document_version_revert',
kwargs={'pk': document_version.pk}
)
from .mixins import DocumentVersionTestMixin, DocumentVersionViewTestMixin
class DocumentVersionViewTestCase(

View File

@@ -1,7 +1,7 @@
from __future__ import unicode_literals
from mayan.apps.documents.permissions import permission_document_view
from mayan.apps.documents.search import document_search, document_page_search
from ..permissions import permission_document_view
from ..search import document_search, document_page_search
from .base import GenericDocumentViewTestCase