blob: f7c06b95bf818bd078b632f18720eb5a6b0d810f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
import base64
from cryptography.fernet import Fernet
class TestFernet(object):
def test_generate(self):
f = Fernet(base64.urlsafe_b64decode(
b"cw_0x689RpI-jtRR7oE8h_eQsKImvJapLeSbXpwF4e4="
))
token = f._encrypt_from_parts(
b"hello",
499162800,
b"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F",
)
assert token == (b"gAAAAAAdwJ6wAAECAwQFBgcICQoLDA0ODy021cpGVWKZ_eEwCGM"
"4BLLF_5CV9dOPmrhuVUPgJobwOz7JcbmrR64jVmpU4IwqDA==")
def test_verify(self):
f = Fernet(base64.urlsafe_b64decode(
b"cw_0x689RpI-jtRR7oE8h_eQsKImvJapLeSbXpwF4e4="
))
payload = f.decrypt(
(b"gAAAAAAdwJ6wAAECAwQFBgcICQoLDA0ODy021cpGVWKZ_eEwCGM4BLLF_5CV9dO"
"PmrhuVUPgJobwOz7JcbmrR64jVmpU4IwqDA=="),
ttl=60,
current_time=499162801
)
assert payload == b"hello"
|
