fc29309f68
Add keyword arguments to all calls. Rename URL parameters to be
explicit ("key_id"). Add key delete view test. Update tests
to use a mixin for repeated key creation code. Grant permissions
and access the proper way using self.grant_permission and
self.grant_access.
Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
from django_downloadview.test import assert_download_response
|
|
|
|
from mayan.apps.common.tests import GenericViewTestCase
|
|
|
|
from ..models import Key
|
|
from ..permissions import (
|
|
permission_key_delete, permission_key_download, permission_key_upload
|
|
)
|
|
|
|
from .literals import TEST_KEY_DATA, TEST_KEY_FINGERPRINT
|
|
from .mixins import KeyTestMixin
|
|
|
|
|
|
class KeyViewTestCase(KeyTestMixin, GenericViewTestCase):
|
|
def _request_key_delete_view(self):
|
|
return self.post(
|
|
viewname='django_gpg:key_delete',
|
|
kwargs={'key_id': self.test_key.pk}
|
|
)
|
|
|
|
def test_key_delete_view_no_permission(self):
|
|
self._create_test_key()
|
|
|
|
response = self._request_key_delete_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
def test_key_delete_view_with_access(self):
|
|
self._create_test_key()
|
|
|
|
self.grant_access(obj=self.test_key, permission=permission_key_delete)
|
|
|
|
response = self._request_key_delete_view()
|
|
self.assertEqual(response.status_code, 302)
|
|
|
|
self.assertEqual(Key.objects.count(), 0)
|
|
|
|
def _request_key_download_view(self):
|
|
return self.get(
|
|
viewname='django_gpg:key_download',
|
|
kwargs={'key_id': self.test_key.pk}
|
|
)
|
|
|
|
def test_key_download_view_no_permission(self):
|
|
self._create_test_key()
|
|
|
|
response = self._request_key_download_view()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
def test_key_download_view_with_permission(self):
|
|
self._create_test_key()
|
|
|
|
self.grant_access(obj=self.test_key, permission=permission_key_download)
|
|
|
|
self.expected_content_type = 'application/octet-stream; charset=utf-8'
|
|
|
|
response = self._request_key_download_view()
|
|
|
|
assert_download_response(
|
|
test_case=self, response=response, content=self.test_key.key_data,
|
|
basename=self.test_key.key_id,
|
|
)
|
|
|
|
def _request_key_upload_view(self):
|
|
return self.post(
|
|
viewname='django_gpg:key_upload', data={'key_data': TEST_KEY_DATA}
|
|
)
|
|
|
|
def test_key_upload_view_no_permission(self):
|
|
response = self._request_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_key_upload_view()
|
|
self.assertEqual(response.status_code, 302)
|
|
|
|
self.assertEqual(Key.objects.count(), 1)
|
|
self.assertEqual(Key.objects.first().fingerprint, TEST_KEY_FINGERPRINT)
|