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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
import pytest
from oauth2_provider import utils
def test_jwk_from_pem_caches_jwk():
a_tiny_rsa_key = """-----BEGIN RSA PRIVATE KEY-----
MGQCAQACEQCxqYaL6GtPooVMhVwcZrCfAgMBAAECECyNmdsuHvMqIEl9/Fex27kC
CQDlc0deuSVrtQIJAMY4MTw2eCeDAgkA5VzfMykQ5yECCQCgkF4Zl0nHPwIJALPv
+IAFUPv3
-----END RSA PRIVATE KEY-----"""
# For the same private key we expect the same object to be returned
jwk1 = utils.jwk_from_pem(a_tiny_rsa_key)
jwk2 = utils.jwk_from_pem(a_tiny_rsa_key)
assert jwk1 is jwk2
a_different_tiny_rsa_key = """-----BEGIN RSA PRIVATE KEY-----
MGMCAQACEQCvyNNNw4J201yzFVogcfgnAgMBAAECEE3oXe5bNlle+xU4EVHTUIEC
CQDpSvwIvDMSIQIJAMDk47DzG9FHAghtvg1TWpy3oQIJAL6NHlS+RBufAgkA6QLA
2GK4aDc=
-----END RSA PRIVATE KEY-----"""
# But for a different key, a different object
jwk3 = utils.jwk_from_pem(a_different_tiny_rsa_key)
assert jwk3 is not jwk1
def test_user_code_generator():
# Default argument, 8 characters
user_code = utils.user_code_generator()
assert isinstance(user_code, str)
assert len(user_code) == 8
for character in user_code:
assert character >= "0"
assert character <= "V"
another_user_code = utils.user_code_generator()
assert another_user_code != user_code
shorter_user_code = utils.user_code_generator(user_code_length=1)
assert len(shorter_user_code) == 1
with pytest.raises(ValueError):
utils.user_code_generator(user_code_length=0)
utils.user_code_generator(user_code_length=-1)
|
