8e69178e07
Instead of inserting the path of the apps into the Python app, the apps are now referenced by their full import path. This app name claves with external or native Python libraries. Example: Mayan statistics app vs. Python new statistics library. Every app reference is now prepended with 'mayan.apps'. Existing config.yml files need to be updated manually. Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
392 lines
14 KiB
Python
392 lines
14 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
import logging
|
|
|
|
from django.core.files import File
|
|
|
|
from django_downloadview.test import assert_download_response
|
|
|
|
from mayan.apps.django_gpg.models import Key
|
|
from mayan.apps.documents.models import DocumentVersion
|
|
from mayan.apps.documents.tests import (
|
|
GenericDocumentViewTestCase, TEST_DOCUMENT_PATH
|
|
)
|
|
|
|
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, TEST_KEY_FILE
|
|
)
|
|
|
|
TEST_UNSIGNED_DOCUMENT_COUNT = 4
|
|
TEST_SIGNED_DOCUMENT_COUNT = 2
|
|
|
|
|
|
class SignaturesViewTestCase(GenericDocumentViewTestCase):
|
|
def _request_document_version_signature_list_view(self, document):
|
|
return self.get(
|
|
viewname='signatures:document_version_signature_list',
|
|
args=(document.latest_version.pk,)
|
|
)
|
|
|
|
def test_signature_list_view_no_permission(self):
|
|
with open(TEST_KEY_FILE, mode='rb') as file_object:
|
|
Key.objects.create(key_data=file_object.read())
|
|
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
DetachedSignature.objects.create(
|
|
document_version=document.latest_version,
|
|
signature_file=File(file_object)
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
response = self._request_document_version_signature_list_view(
|
|
document=document
|
|
)
|
|
self.assertEqual(response.status_code, 403)
|
|
|
|
def test_signature_list_view_with_access(self):
|
|
with open(TEST_KEY_FILE, mode='rb') as file_object:
|
|
Key.objects.create(key_data=file_object.read())
|
|
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
DetachedSignature.objects.create(
|
|
document_version=document.latest_version,
|
|
signature_file=File(file_object)
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
self.grant_access(
|
|
obj=document,
|
|
permission=permission_document_version_signature_view
|
|
)
|
|
|
|
response = self._request_document_version_signature_list_view(
|
|
document=document
|
|
)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.context['object_list'].count(), 1)
|
|
|
|
def _request_document_version_signature_details_view(self, signature):
|
|
return self.get(
|
|
viewname='signatures:document_version_signature_details',
|
|
args=(signature.pk,)
|
|
)
|
|
|
|
def test_signature_detail_view_no_permission(self):
|
|
with open(TEST_KEY_FILE, mode='rb') as file_object:
|
|
Key.objects.create(key_data=file_object.read())
|
|
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
signature = DetachedSignature.objects.create(
|
|
document_version=document.latest_version,
|
|
signature_file=File(file_object)
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
response = self._request_document_version_signature_details_view(
|
|
signature=signature
|
|
)
|
|
self.assertEqual(response.status_code, 403)
|
|
|
|
def test_signature_detail_view_with_access(self):
|
|
with open(TEST_KEY_FILE, mode='rb') as file_object:
|
|
Key.objects.create(key_data=file_object.read())
|
|
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
signature = DetachedSignature.objects.create(
|
|
document_version=document.latest_version,
|
|
signature_file=File(file_object)
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
self.grant_access(
|
|
obj=document,
|
|
permission=permission_document_version_signature_view
|
|
)
|
|
|
|
response = self._request_document_version_signature_details_view(
|
|
signature=signature
|
|
)
|
|
self.assertContains(
|
|
response=response, text=signature.signature_id, status_code=200
|
|
)
|
|
|
|
def _request_document_version_signature_upload_view(self, document_version):
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
return self.post(
|
|
viewname='signatures:document_version_signature_upload',
|
|
args=(document_version.pk,),
|
|
data={'signature_file': file_object}
|
|
)
|
|
|
|
def test_signature_upload_view_no_permission(self):
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
response = self._request_document_version_signature_upload_view(
|
|
document_version=document.latest_version
|
|
)
|
|
self.assertEqual(response.status_code, 403)
|
|
self.assertEqual(DetachedSignature.objects.count(), 0)
|
|
|
|
def test_signature_upload_view_with_access(self):
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
self.grant_access(
|
|
obj=document,
|
|
permission=permission_document_version_signature_upload
|
|
)
|
|
|
|
response = self._request_document_version_signature_upload_view(
|
|
document_version=document.latest_version
|
|
)
|
|
self.assertEqual(response.status_code, 302)
|
|
self.assertEqual(DetachedSignature.objects.count(), 1)
|
|
|
|
def _request_document_version_signature_download_view(self, signature):
|
|
return self.get(
|
|
viewname='signatures:document_version_signature_download',
|
|
args=(signature.pk,),
|
|
)
|
|
|
|
def test_signature_download_view_no_permission(self):
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
signature = DetachedSignature.objects.create(
|
|
document_version=document.latest_version,
|
|
signature_file=File(file_object)
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
response = self._request_document_version_signature_download_view(
|
|
signature=signature
|
|
)
|
|
self.assertEqual(response.status_code, 403)
|
|
|
|
def test_signature_download_view_with_access(self):
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
signature = DetachedSignature.objects.create(
|
|
document_version=document.latest_version,
|
|
signature_file=File(file_object)
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
self.grant_access(
|
|
obj=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(
|
|
signature=signature
|
|
)
|
|
|
|
with signature.signature_file as file_object:
|
|
assert_download_response(
|
|
self, response=response, content=file_object.read(),
|
|
)
|
|
|
|
def _request_document_version_signature_delete_view(self, signature):
|
|
return self.post(
|
|
viewname='signatures:document_version_signature_delete',
|
|
args=(signature.pk,)
|
|
)
|
|
|
|
def test_signature_delete_view_no_permission(self):
|
|
with open(TEST_KEY_FILE, mode='rb') as file_object:
|
|
Key.objects.create(key_data=file_object.read())
|
|
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
signature = DetachedSignature.objects.create(
|
|
document_version=document.latest_version,
|
|
signature_file=File(file_object)
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
self.grant_access(
|
|
obj=document,
|
|
permission=permission_document_version_signature_view
|
|
)
|
|
|
|
response = self._request_document_version_signature_delete_view(
|
|
signature=signature
|
|
)
|
|
self.assertEqual(response.status_code, 403)
|
|
self.assertEqual(DetachedSignature.objects.count(), 1)
|
|
|
|
def test_signature_delete_view_with_access(self):
|
|
with open(TEST_KEY_FILE, mode='rb') as file_object:
|
|
Key.objects.create(key_data=file_object.read())
|
|
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
with open(TEST_SIGNATURE_FILE_PATH, mode='rb') as file_object:
|
|
signature = DetachedSignature.objects.create(
|
|
document_version=document.latest_version,
|
|
signature_file=File(file_object)
|
|
)
|
|
|
|
self.login_user()
|
|
|
|
self.grant_access(
|
|
obj=document,
|
|
permission=permission_document_version_signature_delete
|
|
)
|
|
self.grant_access(
|
|
obj=document,
|
|
permission=permission_document_version_signature_view
|
|
)
|
|
|
|
response = self._request_document_version_signature_delete_view(
|
|
signature=signature
|
|
)
|
|
self.assertEqual(response.status_code, 302)
|
|
self.assertEqual(DetachedSignature.objects.count(), 0)
|
|
|
|
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('converter.backends').setLevel(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):
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
for count in range(TEST_SIGNED_DOCUMENT_COUNT):
|
|
with open(TEST_SIGNED_DOCUMENT_PATH, mode='rb') as file_object:
|
|
self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
self.assertEqual(
|
|
EmbeddedSignature.objects.unsigned_document_versions().count(),
|
|
TEST_UNSIGNED_DOCUMENT_COUNT + TEST_SIGNED_DOCUMENT_COUNT
|
|
)
|
|
|
|
DocumentVersion._post_save_hooks = old_hooks
|
|
|
|
self.login_user()
|
|
|
|
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('converter.backends').setLevel(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):
|
|
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
|
|
self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
for count in range(TEST_SIGNED_DOCUMENT_COUNT):
|
|
with open(TEST_SIGNED_DOCUMENT_PATH, mode='rb') as file_object:
|
|
self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
self.assertEqual(
|
|
EmbeddedSignature.objects.unsigned_document_versions().count(),
|
|
TEST_UNSIGNED_DOCUMENT_COUNT + TEST_SIGNED_DOCUMENT_COUNT
|
|
)
|
|
|
|
DocumentVersion._post_save_hooks = old_hooks
|
|
|
|
self.login_user()
|
|
|
|
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
|
|
)
|