4937d8b776
Add keyword arguments. Remove source column functions and move their code to the model. Use the FilteredSelectionForm for the key selection in the document version signing view. Update the field definition of the DetailForm subclasses to use the new internface. Update URL parameters to use the "_id" form. Update views filtering to comply with MERC 6. Move repeated test code to its own test mixin. Update links to work with the new Link class interface. Modernize tests. Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
263 lines
9.5 KiB
Python
263 lines
9.5 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
import logging
|
|
|
|
from django_downloadview.test import assert_download_response
|
|
|
|
from mayan.apps.documents.models import DocumentVersion
|
|
from mayan.apps.documents.tests import (
|
|
TEST_DOCUMENT_PATH, GenericDocumentViewTestCase
|
|
)
|
|
|
|
from ..models import DetachedSignature, EmbeddedSignature
|
|
from ..permissions import (
|
|
permission_document_version_signature_delete,
|
|
permission_document_version_signature_download,
|
|
permission_document_version_signature_upload,
|
|
permission_document_version_signature_verify,
|
|
permission_document_version_signature_view
|
|
)
|
|
|
|
from .literals import TEST_SIGNATURE_FILE_PATH, TEST_SIGNED_DOCUMENT_PATH
|
|
from .mixins import SignaturesTestMixin
|
|
|
|
TEST_UNSIGNED_DOCUMENT_COUNT = 4
|
|
TEST_SIGNED_DOCUMENT_COUNT = 2
|
|
|
|
|
|
class SignaturesViewTestCase(SignaturesTestMixin, GenericDocumentViewTestCase):
|
|
def _request_document_version_signature_delete_view(self):
|
|
return self.post(
|
|
viewname='signatures:document_version_signature_delete',
|
|
kwargs={'signature_id': self.test_signature.pk}
|
|
)
|
|
|
|
def test_signature_delete_view_no_permission(self):
|
|
self._create_test_key()
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
self._upload_test_signature()
|
|
|
|
response = self._request_document_version_signature_delete_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
self.assertEqual(DetachedSignature.objects.count(), 1)
|
|
|
|
def test_signature_delete_view_with_access(self):
|
|
self._create_test_key()
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
self._upload_test_signature()
|
|
|
|
self.grant_access(
|
|
obj=self.test_document,
|
|
permission=permission_document_version_signature_delete
|
|
)
|
|
response = self._request_document_version_signature_delete_view()
|
|
self.assertEqual(response.status_code, 302)
|
|
self.assertEqual(DetachedSignature.objects.count(), 0)
|
|
|
|
def _request_document_version_signature_details_view(self):
|
|
return self.get(
|
|
viewname='signatures:document_version_signature_details',
|
|
kwargs={'signature_id': self.test_signature.pk}
|
|
)
|
|
|
|
def test_signature_detail_view_no_permission(self):
|
|
self._create_test_key()
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
self._upload_test_signature()
|
|
|
|
response = self._request_document_version_signature_details_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
def test_signature_detail_view_with_access(self):
|
|
self._create_test_key()
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
self._upload_test_signature()
|
|
self.grant_access(
|
|
obj=self.test_document,
|
|
permission=permission_document_version_signature_view
|
|
)
|
|
|
|
response = self._request_document_version_signature_details_view()
|
|
self.assertContains(
|
|
response=response, text=self.test_signature.signature_id,
|
|
status_code=200
|
|
)
|
|
|
|
def _request_document_version_signature_download_view(self):
|
|
return self.get(
|
|
viewname='signatures:document_version_signature_download',
|
|
kwargs={'signature_id': self.test_signature.pk}
|
|
)
|
|
|
|
def test_signature_download_view_no_permission(self):
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
self._upload_test_signature()
|
|
|
|
response = self._request_document_version_signature_download_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
def test_signature_download_view_with_access(self):
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
|
|
self._upload_test_signature()
|
|
|
|
self.grant_access(
|
|
obj=self.test_document,
|
|
permission=permission_document_version_signature_download
|
|
)
|
|
|
|
self.expected_content_type = 'application/octet-stream; charset=utf-8'
|
|
|
|
response = self._request_document_version_signature_download_view()
|
|
|
|
with self.test_signature.signature_file as file_object:
|
|
assert_download_response(
|
|
self, response=response, content=file_object.read(),
|
|
)
|
|
|
|
def _request_document_version_signature_list_view(self):
|
|
return self.get(
|
|
viewname='signatures:document_version_signature_list',
|
|
kwargs={'document_version_id': self.test_document.latest_version.pk}
|
|
)
|
|
|
|
def test_signature_list_view_no_permission(self):
|
|
self._create_test_key()
|
|
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
|
|
self._upload_test_signature()
|
|
|
|
response = self._request_document_version_signature_list_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
def test_signature_list_view_with_access(self):
|
|
self._create_test_key()
|
|
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
|
|
self._upload_test_signature()
|
|
|
|
self.grant_access(
|
|
obj=self.test_document,
|
|
permission=permission_document_version_signature_view
|
|
)
|
|
|
|
response = self._request_document_version_signature_list_view()
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.context['object_list'].count(), 1)
|
|
|
|
def _request_document_version_signature_upload_view(self):
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
return self.post(
|
|
viewname='signatures:document_version_signature_upload',
|
|
kwargs={'document_version_id': self.test_document.latest_version.pk},
|
|
data={'signature_file': file_object}
|
|
)
|
|
|
|
def test_signature_upload_view_no_permission(self):
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
|
|
response = self._request_document_version_signature_upload_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
self.assertEqual(DetachedSignature.objects.count(), 0)
|
|
|
|
def test_signature_upload_view_with_access(self):
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.test_document = self.upload_document()
|
|
|
|
self.grant_access(
|
|
obj=self.test_document,
|
|
permission=permission_document_version_signature_upload
|
|
)
|
|
|
|
response = self._request_document_version_signature_upload_view()
|
|
self.assertEqual(response.status_code, 302)
|
|
self.assertEqual(DetachedSignature.objects.count(), 1)
|
|
|
|
def _request_all_document_version_signature_verify_view(self):
|
|
return self.post(
|
|
viewname='signatures:all_document_version_signature_verify'
|
|
)
|
|
|
|
def test_missing_signature_verify_view_no_permission(self):
|
|
# Silence converter logging
|
|
logging.getLogger('mayan.apps.converter.backends').setLevel(
|
|
level=logging.CRITICAL
|
|
)
|
|
|
|
for document in self.document_type.documents.all():
|
|
document.delete(to_trash=False)
|
|
|
|
old_hooks = DocumentVersion._post_save_hooks
|
|
DocumentVersion._post_save_hooks = {}
|
|
for count in range(TEST_UNSIGNED_DOCUMENT_COUNT):
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.upload_document()
|
|
|
|
for count in range(TEST_SIGNED_DOCUMENT_COUNT):
|
|
self.test_document_path = TEST_SIGNED_DOCUMENT_PATH
|
|
self.upload_document()
|
|
|
|
self.assertEqual(
|
|
EmbeddedSignature.objects.unsigned_document_versions().count(),
|
|
TEST_UNSIGNED_DOCUMENT_COUNT + TEST_SIGNED_DOCUMENT_COUNT
|
|
)
|
|
|
|
DocumentVersion._post_save_hooks = old_hooks
|
|
|
|
response = self._request_all_document_version_signature_verify_view()
|
|
self.assertEqual(response.status_code, 403)
|
|
|
|
self.assertEqual(
|
|
EmbeddedSignature.objects.unsigned_document_versions().count(),
|
|
TEST_UNSIGNED_DOCUMENT_COUNT + TEST_SIGNED_DOCUMENT_COUNT
|
|
)
|
|
|
|
def test_missing_signature_verify_view_with_permission(self):
|
|
# Silence converter logging
|
|
logging.getLogger('mayan.apps.converter.backends').setLevel(
|
|
level=logging.CRITICAL
|
|
)
|
|
|
|
for document in self.document_type.documents.all():
|
|
document.delete(to_trash=False)
|
|
|
|
old_hooks = DocumentVersion._post_save_hooks
|
|
DocumentVersion._post_save_hooks = {}
|
|
for count in range(TEST_UNSIGNED_DOCUMENT_COUNT):
|
|
self.test_document_path = TEST_DOCUMENT_PATH
|
|
self.upload_document()
|
|
|
|
for count in range(TEST_SIGNED_DOCUMENT_COUNT):
|
|
self.test_document_path = TEST_SIGNED_DOCUMENT_PATH
|
|
self.upload_document()
|
|
|
|
self.assertEqual(
|
|
EmbeddedSignature.objects.unsigned_document_versions().count(),
|
|
TEST_UNSIGNED_DOCUMENT_COUNT + TEST_SIGNED_DOCUMENT_COUNT
|
|
)
|
|
|
|
DocumentVersion._post_save_hooks = old_hooks
|
|
|
|
self.grant_permission(
|
|
permission=permission_document_version_signature_verify
|
|
)
|
|
|
|
response = self._request_all_document_version_signature_verify_view()
|
|
self.assertEqual(response.status_code, 302)
|
|
|
|
self.assertEqual(
|
|
EmbeddedSignature.objects.unsigned_document_versions().count(),
|
|
TEST_UNSIGNED_DOCUMENT_COUNT
|
|
)
|