Hopiu/django-downloadview
1
0
Fork 0
mirror of https://github.com/jazzband/django-downloadview.git synced 2026-05-28 16:58:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test for signature_required wrapper

Browse source
This commit is contained in:
Aleksi Häkli 2020-01-13 17:35:03 +02:00 committed by GitHub
parent 965218aafb
commit df64eaef80
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 26 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
tests/signature.py
View file

@ -20,6 +20,10 @@ class SignedTestStorage(SignedURLMixin, TestStorage):
pass
def signable_test_view(request):
return True
class SignatureGeneratorTestCase(unittest.TestCase):
def test_signed_storage(self):
"""
@ -32,10 +36,26 @@ class SignatureGeneratorTestCase(unittest.TestCase):
class SignatureValidatorTestCase(unittest.TestCase):
def test_verify_signature(self):
def test_signature_required(self):
"""
django_downloadview.decorators._signature_is_valid returns True on
valid signatures.
django_downloadview.decorators.signature_required
wraps view-like callables and returns their values.
"""
signer = TimestampSigner()
request = unittest.mock.MagicMock()
request.path = "test"
request.GET = {"X-Signature": signer.sign("test")}
response = signature_required(signable_test_view)(request)
self.assertTrue(response)
def test_signature_is_valid(self):
"""
django_downloadview.decorators._signature_is_valid
returns None on valid signatures.
"""
signer = TimestampSigner()
@ -46,10 +66,10 @@ class SignatureValidatorTestCase(unittest.TestCase):
self.assertIsNone(_signature_is_valid(request))
def test_verify_signature_invalid(self):
def test_signature_is_valid_error(self):
"""
django_downloadview.decorators._signature_is_valid raises PermissionDenied
on invalid signatures.
django_downloadview.decorators._signature_is_valid
raises PermissionDenied on invalid signatures.
"""
request = unittest.mock.MagicMock()