Files
mayan-edms/mayan/apps/acls/tests/test_links.py
Roberto Rosario 0267c79b07 Backport test improvements
Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
2019-04-20 02:39:57 -04:00

106 lines
3.1 KiB
Python

from __future__ import unicode_literals
from django.contrib.contenttypes.models import ContentType
from django.urls import reverse
from mayan.apps.documents.tests import GenericDocumentViewTestCase
from ..links import (
link_acl_delete, link_acl_list, link_acl_create, link_acl_permissions
)
from ..permissions import permission_acl_edit, permission_acl_view
from .mixins import ACLTestMixin
class ACLsLinksTestCase(ACLTestMixin, GenericDocumentViewTestCase):
def setUp(self):
super(ACLsLinksTestCase, self).setUp()
self.test_object = self.test_document
def test_document_acl_create_link(self):
self.grant_access(
obj=self.test_document, permission=permission_acl_edit
)
self.add_test_view(test_object=self.document)
context = self.get_test_view()
resolved_link = link_acl_create.resolve(context=context)
self.assertNotEqual(resolved_link, None)
content_type = ContentType.objects.get_for_model(self.test_document)
kwargs = {
'app_label': content_type.app_label,
'model': content_type.model,
'object_id': self.document.pk
}
self.assertEqual(
resolved_link.url, reverse(
viewname=link_acl_create.view, kwargs=kwargs
)
)
def test_document_acl_delete_link(self):
self._create_test_acl()
self.grant_access(
obj=self.test_document, permission=permission_acl_edit
)
self.add_test_view(test_object=self.test_acl)
context = self.get_test_view()
resolved_link = link_acl_delete.resolve(context=context)
self.assertNotEqual(resolved_link, None)
self.assertEqual(
resolved_link.url, reverse(
viewname=link_acl_delete.view, args=(self.test_acl.pk,)
)
)
def test_document_acl_edit_link(self):
self._create_test_acl()
self.grant_access(
obj=self.test_document, permission=permission_acl_edit
)
self.add_test_view(test_object=self.test_acl)
context = self.get_test_view()
resolved_link = link_acl_permissions.resolve(context=context)
self.assertNotEqual(resolved_link, None)
self.assertEqual(
resolved_link.url, reverse(
viewname=link_acl_permissions.view,
args=(self.test_acl.pk,)
)
)
def test_document_acl_list_link(self):
self.grant_access(
obj=self.test_document, permission=permission_acl_view
)
self.add_test_view(test_object=self.test_document)
context = self.get_test_view()
resolved_link = link_acl_list.resolve(context=context)
self.assertNotEqual(resolved_link, None)
content_type = ContentType.objects.get_for_model(self.test_document)
kwargs = {
'app_label': content_type.app_label,
'model': content_type.model,
'object_id': self.test_document.pk
}
self.assertEqual(
resolved_link.url, reverse(
viewname=link_acl_list.view, kwargs=kwargs
)
)