456c322c19
* The new module is called methods.py and found on each app. * Add keyword arguments to add_to_class instances. * Remove catch all exception handling for the check in and check out views. * Improve checkouts tests code reducing redundant code. Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from mayan.apps.documents.tests import GenericDocumentViewTestCase
|
|
|
|
from ..links import link_check_out_document, link_check_out_info
|
|
from ..permissions import (
|
|
permission_document_check_out, permission_document_check_out_detail_view
|
|
)
|
|
|
|
from .mixins import DocumentCheckoutTestMixin
|
|
|
|
|
|
class CheckoutLinksTestCase(DocumentCheckoutTestMixin, GenericDocumentViewTestCase):
|
|
def setUp(self):
|
|
super(CheckoutLinksTestCase, self).setUp()
|
|
self.login_user()
|
|
|
|
def _resolve_checkout_link(self):
|
|
self.add_test_view(test_object=self.document)
|
|
context = self.get_test_view()
|
|
context['user'] = self.user
|
|
return link_check_out_document.resolve(context=context)
|
|
|
|
def test_checkout_link_no_access(self):
|
|
resolved_link = self._resolve_checkout_link()
|
|
self.assertEqual(resolved_link, None)
|
|
|
|
def test_checkout_link_with_access(self):
|
|
self.grant_access(
|
|
obj=self.document, permission=permission_document_check_out
|
|
)
|
|
resolved_link = self._resolve_checkout_link()
|
|
self.assertNotEqual(resolved_link, None)
|
|
|
|
def _resolve_checkout_info_link(self):
|
|
self.add_test_view(test_object=self.document)
|
|
context = self.get_test_view()
|
|
context['user'] = self.user
|
|
return link_check_out_info.resolve(context=context)
|
|
|
|
def test_checkout_info_link_no_access(self):
|
|
resolved_link = self._resolve_checkout_info_link()
|
|
self.assertEqual(resolved_link, None)
|
|
|
|
def test_checkout_info_link_with_access(self):
|
|
self.grant_access(
|
|
obj=self.document, permission=permission_document_check_out_detail_view
|
|
)
|
|
resolved_link = self._resolve_checkout_info_link()
|
|
self.assertNotEqual(resolved_link, None)
|