aboutsummaryrefslogtreecommitdiffstats
path: root/tests/test_fernet.py
diff options
context:
space:
mode:
authorAlex Gaynor <alex.gaynor@gmail.com>2013-12-16 15:44:06 -0800
committerAlex Gaynor <alex.gaynor@gmail.com>2013-12-16 15:44:06 -0800
commita8f0b63dddc6a22a1a982c6217d4cef2f598b781 (patch)
treecfaefed672494fc6f4076655adf67136708ecbdb /tests/test_fernet.py
parentfae20715b85e84297f01b60fc153cde93a7549c7 (diff)
downloadcryptography-a8f0b63dddc6a22a1a982c6217d4cef2f598b781.tar.gz
cryptography-a8f0b63dddc6a22a1a982c6217d4cef2f598b781.tar.bz2
cryptography-a8f0b63dddc6a22a1a982c6217d4cef2f598b781.zip
Replace assertions with real error checks
Diffstat (limited to 'tests/test_fernet.py')
-rw-r--r--tests/test_fernet.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/test_fernet.py b/tests/test_fernet.py
index 48df867c..77661180 100644
--- a/tests/test_fernet.py
+++ b/tests/test_fernet.py
@@ -69,6 +69,11 @@ class TestFernet(object):
with pytest.raises(InvalidToken):
f.decrypt(token.encode("ascii"), ttl=ttl_sec)
+ def test_invalid_start_byte(self, backend):
+ f = Fernet(Fernet.generate_key(), backend=backend)
+ with pytest.raises(InvalidToken):
+ f.decrypt(base64.urlsafe_b64encode(b"\x81"))
+
def test_unicode(self, backend):
f = Fernet(base64.urlsafe_b64encode(b"\x00" * 32), backend=backend)
with pytest.raises(TypeError):
@@ -84,3 +89,7 @@ class TestFernet(object):
def test_default_backend(self):
f = Fernet(Fernet.generate_key())
assert f._backend is default_backend()
+
+ def test_bad_key(self, backend):
+ with pytest.raises(ValueError):
+ Fernet(base64.urlsafe_b64encode(b"abc"), backend=backend)