| Terry Chia | 4756fa9 | 2015-10-03 12:23:07 +0800 | [diff] [blame^] | 1 | # This file is dual licensed under the terms of the Apache License, Version |
| 2 | # 2.0, and the BSD License. See the LICENSE file in the root of this repository |
| 3 | # for complete details. |
| 4 | |
| Terry Chia | 36a787f | 2015-09-30 19:13:14 +0800 | [diff] [blame] | 5 | from hypothesis import given |
| Terry Chia | 3d7f8d1 | 2015-09-27 21:00:58 +0800 | [diff] [blame] | 6 | from hypothesis.strategies import binary |
| Terry Chia | 42ea0d1 | 2015-09-27 20:26:09 +0800 | [diff] [blame] | 7 | |
| Terry Chia | cc2a828 | 2015-03-22 12:06:45 +0800 | [diff] [blame] | 8 | from cryptography.fernet import Fernet |
| 9 | |
| Terry Chia | 42ea0d1 | 2015-09-27 20:26:09 +0800 | [diff] [blame] | 10 | |
| Terry Chia | 36a787f | 2015-09-30 19:13:14 +0800 | [diff] [blame] | 11 | @given(binary()) |
| Terry Chia | cc2a828 | 2015-03-22 12:06:45 +0800 | [diff] [blame] | 12 | def test_fernet(data): |
| 13 | f = Fernet(Fernet.generate_key()) |
| 14 | ct = f.encrypt(data) |
| 15 | assert f.decrypt(ct) == data |