| Terry Chia | 36a787f | 2015-09-30 19:13:14 +0800 | [diff] [blame] | 1 | from hypothesis import given |
| Terry Chia | 3d7f8d1 | 2015-09-27 21:00:58 +0800 | [diff] [blame] | 2 | from hypothesis.strategies import binary |
| Terry Chia | 42ea0d1 | 2015-09-27 20:26:09 +0800 | [diff] [blame] | 3 | |
| Terry Chia | cc2a828 | 2015-03-22 12:06:45 +0800 | [diff] [blame] | 4 | from cryptography.fernet import Fernet |
| 5 | |
| Terry Chia | 42ea0d1 | 2015-09-27 20:26:09 +0800 | [diff] [blame] | 6 | |
| Terry Chia | 36a787f | 2015-09-30 19:13:14 +0800 | [diff] [blame] | 7 | @given(binary()) |
| Terry Chia | cc2a828 | 2015-03-22 12:06:45 +0800 | [diff] [blame] | 8 | def test_fernet(data): |
| 9 | f = Fernet(Fernet.generate_key()) |
| 10 | ct = f.encrypt(data) |
| 11 | assert f.decrypt(ct) == data |