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>
124 lines
4.9 KiB
Python
124 lines
4.9 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.test import override_settings
|
|
from django.utils.encoding import force_text
|
|
|
|
from rest_framework import status
|
|
|
|
from mayan.apps.documents.tests import DocumentTestMixin
|
|
from mayan.apps.documents.permissions import permission_document_view
|
|
from mayan.apps.rest_api.tests import BaseAPITestCase
|
|
|
|
from ..models import DocumentCheckout
|
|
from ..permissions import (
|
|
permission_document_check_out, permission_document_check_out_detail_view
|
|
)
|
|
|
|
from .mixins import DocumentCheckoutTestMixin
|
|
|
|
|
|
@override_settings(OCR_AUTO_OCR=False)
|
|
class CheckoutsAPITestCase(DocumentCheckoutTestMixin, DocumentTestMixin, BaseAPITestCase):
|
|
def setUp(self):
|
|
super(CheckoutsAPITestCase, self).setUp()
|
|
self.login_user()
|
|
|
|
def _request_checkedout_document_view(self):
|
|
return self.get(
|
|
viewname='rest_api:checkedout-document-view',
|
|
args=(self.test_check_out.pk,)
|
|
)
|
|
|
|
def test_checkedout_document_view_no_access(self):
|
|
self._check_out_document()
|
|
response = self._request_checkedout_document_view()
|
|
self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
|
|
|
|
def test_checkedout_document_view_with_checkout_access(self):
|
|
self._check_out_document()
|
|
self.grant_access(
|
|
permission=permission_document_check_out_detail_view, obj=self.document
|
|
)
|
|
response = self._request_checkedout_document_view()
|
|
self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
|
|
|
|
def test_checkedout_document_view_with_document_access(self):
|
|
self._check_out_document()
|
|
self.grant_access(
|
|
permission=permission_document_view, obj=self.document
|
|
)
|
|
response = self._request_checkedout_document_view()
|
|
self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
|
|
|
|
def test_checkedout_document_view_with_access(self):
|
|
self._check_out_document()
|
|
self.grant_access(
|
|
permission=permission_document_view, obj=self.document
|
|
)
|
|
self.grant_access(
|
|
permission=permission_document_check_out_detail_view, obj=self.document
|
|
)
|
|
response = self._request_checkedout_document_view()
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertEqual(response.data['document']['uuid'], force_text(self.document.uuid))
|
|
|
|
def _request_document_checkout_view(self):
|
|
return self.post(
|
|
viewname='rest_api:checkout-document-list', data={
|
|
'document_pk': self.document.pk,
|
|
'expiration_datetime': '2099-01-01T12:00'
|
|
}
|
|
)
|
|
|
|
def test_document_checkout_no_access(self):
|
|
response = self._request_document_checkout_view()
|
|
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
|
|
self.assertEqual(DocumentCheckout.objects.count(), 0)
|
|
|
|
def test_document_checkout_with_access(self):
|
|
self.grant_access(permission=permission_document_check_out, obj=self.document)
|
|
response = self._request_document_checkout_view()
|
|
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
|
self.assertEqual(
|
|
DocumentCheckout.objects.first().document, self.document
|
|
)
|
|
|
|
def _request_checkout_list_view(self):
|
|
return self.get(viewname='rest_api:checkout-document-list')
|
|
|
|
def test_checkout_list_view_no_access(self):
|
|
self._check_out_document()
|
|
response = self._request_checkout_list_view()
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertNotContains(response=response, text=self.document.uuid)
|
|
|
|
def test_checkout_list_view_with_document_access(self):
|
|
self._check_out_document()
|
|
self.grant_access(
|
|
permission=permission_document_view, obj=self.document
|
|
)
|
|
response = self._request_checkout_list_view()
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertNotContains(response=response, text=self.document.uuid)
|
|
|
|
def test_checkout_list_view_with_checkout_access(self):
|
|
self._check_out_document()
|
|
self.grant_access(
|
|
permission=permission_document_check_out_detail_view, obj=self.document
|
|
)
|
|
response = self._request_checkout_list_view()
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertNotContains(response=response, text=self.document.uuid)
|
|
|
|
def test_checkout_list_view_with_access(self):
|
|
self._check_out_document()
|
|
self.grant_access(
|
|
permission=permission_document_view, obj=self.document
|
|
)
|
|
self.grant_access(
|
|
permission=permission_document_check_out_detail_view, obj=self.document
|
|
)
|
|
response = self._request_checkout_list_view()
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertContains(response=response, text=self.document.uuid)
|