a7b31fc171
- Use modified port of Django 2.2 FileResponse. - Remove Django DownloadView library. Signed-off-by: Roberto Rosario <roberto.rosario@mayan-edms.com>
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
from mayan.apps.common.tests.base import GenericViewTestCase
|
|
|
|
from ..models import Key
|
|
from ..permissions import permission_key_download, permission_key_upload
|
|
|
|
from .literals import TEST_KEY_PRIVATE_FINGERPRINT
|
|
from .mixins import KeyTestMixin, KeyViewTestMixin
|
|
|
|
|
|
class KeyViewTestCase(KeyTestMixin, KeyViewTestMixin, GenericViewTestCase):
|
|
def test_key_download_view_no_permission(self):
|
|
self._create_test_key_private()
|
|
|
|
response = self._request_test_key_download_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
def test_key_download_view_with_permission(self):
|
|
self.expected_content_types = ('text/html; charset=utf-8',)
|
|
|
|
self._create_test_key_private()
|
|
|
|
self.grant_access(
|
|
obj=self.test_key_private, permission=permission_key_download
|
|
)
|
|
|
|
response = self._request_test_key_download_view()
|
|
self.assert_download_response(
|
|
response=response, content=self.test_key_private.key_data,
|
|
filename=self.test_key_private.key_id,
|
|
)
|
|
|
|
def test_key_upload_view_no_permission(self):
|
|
response = self._request_test_key_upload_view()
|
|
self.assertEqual(response.status_code, 403)
|
|
|
|
self.assertEqual(Key.objects.count(), 0)
|
|
|
|
def test_key_upload_view_with_permission(self):
|
|
self.grant_permission(permission=permission_key_upload)
|
|
|
|
response = self._request_test_key_upload_view()
|
|
self.assertEqual(response.status_code, 302)
|
|
|
|
self.assertEqual(Key.objects.count(), 1)
|
|
self.assertEqual(
|
|
Key.objects.first().fingerprint, TEST_KEY_PRIVATE_FINGERPRINT
|
|
)
|