matthias/mayan-edms
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Consolidate repeated test user logins

Browse Source 
Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2019-04-11 22:57:39 -04:00
parent ae814a18c5
commit 0188afcd54
14 changed files with 69 additions and 212 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
mayan/apps/documents/tests/test_links.py
View File

@@ -12,6 +12,7 @@ from ..links import (
link_document_restore, link_document_version_download,
link_document_version_revert
)
from ..models import DeletedDocument
from ..permissions import (
permission_document_download, permission_document_restore,
permission_document_version_revert
@@ -22,14 +23,16 @@ from .literals import TEST_SMALL_DOCUMENT_PATH
class DocumentsLinksTestCase(GenericDocumentViewTestCase):
def setUp(self):
super(DocumentsLinksTestCase, self).setUp()
self.login_user()
def test_document_version_revert_link_no_permission(self):
with open(TEST_SMALL_DOCUMENT_PATH, mode='rb') as file_object:
self.document.new_version(file_object=file_object)
self.assertTrue(self.document.versions.count(), 2)
self.login_user()
self.add_test_view(test_object=self.document.versions.first())
context = self.get_test_view()
resolved_link = link_document_version_revert.resolve(context=context)
@@ -46,8 +49,6 @@ class DocumentsLinksTestCase(GenericDocumentViewTestCase):
self.assertTrue(self.document.versions.count(), 2)
self.login_user()
acl = AccessControlList.objects.create(
content_object=self.document, role=self.role
)
@@ -69,8 +70,6 @@ class DocumentsLinksTestCase(GenericDocumentViewTestCase):
)
def test_document_version_download_link_no_permission(self):
self.login_user()
self.add_test_view(test_object=self.document.latest_version)
context = self.get_test_view()
resolved_link = link_document_version_download.resolve(context=context)
@@ -78,8 +77,6 @@ class DocumentsLinksTestCase(GenericDocumentViewTestCase):
self.assertEqual(resolved_link, None)
def test_document_version_download_link_with_permission(self):
self.login_user()
acl = AccessControlList.objects.create(
content_object=self.document, role=self.role
)
@@ -100,38 +97,30 @@ class DocumentsLinksTestCase(GenericDocumentViewTestCase):
class DeletedDocumentsLinksTestCase(GenericDocumentViewTestCase):
def test_deleted_document_restore_link_no_permission(self):
self.document.delete()
def setUp(self):
super(DeletedDocumentsLinksTestCase, self).setUp()
self.login_user()
self.document.delete()
self.test_deleted_document = DeletedDocument.objects.get(
pk=self.document.pk
)
self.add_test_view(test_object=self.test_deleted_document)
self.context = self.get_test_view()
self.add_test_view(test_object=self.document)
context = self.get_test_view()
resolved_link = link_document_restore.resolve(context=context)
def test_deleted_document_restore_link_no_permission(self):
resolved_link = link_document_restore.resolve(context=self.context)
self.assertEqual(resolved_link, None)
def test_deleted_document_restore_link_with_permission(self):
self.document.delete()
self.login_user()
acl = AccessControlList.objects.create(
content_object=self.document, role=self.role
self.grant_access(
obj=self.document, permission=permission_document_restore
)
acl.permissions.add(
permission_document_restore.stored_permission
)
self.add_test_view(test_object=self.document)
context = self.get_test_view()
resolved_link = link_document_restore.resolve(context=context)
resolved_link = link_document_restore.resolve(context=self.context)
self.assertNotEqual(resolved_link, None)
self.assertEqual(
resolved_link.url,
reverse(
'documents:document_restore',
args=(self.document.pk,)
viewname=link_document_restore.view,
args=(self.test_deleted_document.pk,)
)
)