36a51eeb73
Instead of inserting the path of the apps into the Python app, the apps are now referenced by their full import path. This solves name clashes 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>
67 lines
2.0 KiB
Python
67 lines
2.0 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_download, permission_key_upload
|
|
|
|
from .literals import TEST_KEY_DATA, TEST_KEY_FINGERPRINT
|
|
|
|
|
|
class KeyViewTestCase(GenericViewTestCase):
|
|
def test_key_download_view_no_permission(self):
|
|
key = Key.objects.create(key_data=TEST_KEY_DATA)
|
|
|
|
self.login_user()
|
|
|
|
response = self.get(
|
|
viewname='django_gpg:key_download', args=(key.pk,)
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 403)
|
|
|
|
def test_key_download_view_with_permission(self):
|
|
key = Key.objects.create(key_data=TEST_KEY_DATA)
|
|
|
|
self.login_user()
|
|
|
|
self.role.permissions.add(permission_key_download.stored_permission)
|
|
|
|
self.expected_content_type = 'application/octet-stream; charset=utf-8'
|
|
|
|
response = self.get(
|
|
viewname='django_gpg:key_download', args=(key.pk,)
|
|
)
|
|
|
|
assert_download_response(
|
|
self, response=response, content=key.key_data,
|
|
basename=key.key_id,
|
|
)
|
|
|
|
def test_key_upload_view_no_permission(self):
|
|
self.login_user()
|
|
|
|
response = self.post(
|
|
viewname='django_gpg:key_upload', data={'key_data': TEST_KEY_DATA}
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 403)
|
|
self.assertEqual(Key.objects.count(), 0)
|
|
|
|
def test_key_upload_view_with_permission(self):
|
|
self.login_user()
|
|
|
|
self.role.permissions.add(permission_key_upload.stored_permission)
|
|
|
|
response = self.post(
|
|
viewname='django_gpg:key_upload', data={'key_data': TEST_KEY_DATA},
|
|
follow=True
|
|
)
|
|
|
|
self.assertContains(response, 'created', status_code=200)
|
|
|
|
self.assertEqual(Key.objects.count(), 1)
|
|
self.assertEqual(Key.objects.first().fingerprint, TEST_KEY_FINGERPRINT)
|