Close GitLab issue #302 'New Document' button available to users who do not have permission.

Thanks to kg @kgraves.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2018-04-03 01:13:59 -04:00
parent 92bd82320b
commit ed5d7cd812
4 changed files with 53 additions and 3 deletions

View File

@@ -0,0 +1,31 @@
from __future__ import unicode_literals
from django.contrib.contenttypes.models import ContentType
from django.urls import reverse
from acls.models import AccessControlList
from documents.tests import GenericDocumentViewTestCase
from documents.permissions import permission_document_create
from ..links import link_document_create_multiple
class SourcesLinksTestCase(GenericDocumentViewTestCase):
def setUp(self):
super(SourcesLinksTestCase, self).setUp()
self.login_user()
def _get_document_create_link(self):
self.add_test_view(test_object=self.document)
context = self.get_test_view()
context['user'] = self.user
return link_document_create_multiple.resolve(context=context)
def test_document_create_link_no_access(self):
resolved_link = self._get_document_create_link()
self.assertEqual(resolved_link, None)
def test_document_create_link_with_access(self):
self.grant_access(permission=permission_document_create, obj=self.document_type)
resolved_link = self._get_document_create_link()
self.assertNotEqual(resolved_link, None)