diff options
author | Alex Gaynor <alex.gaynor@gmail.com> | 2013-12-16 15:44:06 -0800 |
---|---|---|
committer | Alex Gaynor <alex.gaynor@gmail.com> | 2013-12-16 15:44:06 -0800 |
commit | a8f0b63dddc6a22a1a982c6217d4cef2f598b781 (patch) | |
tree | cfaefed672494fc6f4076655adf67136708ecbdb /tests/test_fernet.py | |
parent | fae20715b85e84297f01b60fc153cde93a7549c7 (diff) | |
download | cryptography-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.py | 9 |
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) |