2
2
from os import environ
3
3
from sys import argv
4
4
5
- import requests
5
+ import httpx
6
6
7
7
8
8
def sign_request (req_headers : dict , secret = None , user : str = "" ):
@@ -14,43 +14,43 @@ def sign_request(req_headers: dict, secret=None, user: str = ""):
14
14
if __name__ == "__main__" :
15
15
request_url = argv [1 ] + "/sec_check?value=1"
16
16
headers = {}
17
- result = requests .put (request_url , headers = headers )
17
+ result = httpx .put (request_url , headers = headers )
18
18
assert result .status_code == 401 # Missing headers
19
19
headers .update ({
20
20
"AA-VERSION" : environ .get ("AA_VERSION" , "1.0.0" ),
21
21
"EX-APP-ID" : environ .get ("APP_ID" , "nc_py_api" ),
22
22
"EX-APP-VERSION" : environ .get ("APP_VERSION" , "1.0.0" ),
23
23
})
24
24
sign_request (headers )
25
- result = requests .put (request_url , headers = headers )
25
+ result = httpx .put (request_url , headers = headers )
26
26
assert result .status_code == 200
27
27
# Invalid AA-SIGNATURE
28
28
sign_request (headers , secret = "xxx" )
29
- result = requests .put (request_url , headers = headers )
29
+ result = httpx .put (request_url , headers = headers )
30
30
assert result .status_code == 401
31
31
sign_request (headers )
32
- result = requests .put (request_url , headers = headers )
32
+ result = httpx .put (request_url , headers = headers )
33
33
assert result .status_code == 200
34
34
# Invalid EX-APP-ID
35
35
old_app_name = headers ["EX-APP-ID" ]
36
36
headers ["EX-APP-ID" ] = "unknown_app"
37
37
sign_request (headers )
38
- result = requests .put (request_url , headers = headers )
38
+ result = httpx .put (request_url , headers = headers )
39
39
assert result .status_code == 401
40
40
headers ["EX-APP-ID" ] = old_app_name
41
41
sign_request (headers )
42
- result = requests .put (request_url , headers = headers )
42
+ result = httpx .put (request_url , headers = headers )
43
43
assert result .status_code == 200
44
44
# Invalid EX-APP-VERSION
45
45
sign_request (headers )
46
- result = requests .put (request_url , headers = headers )
46
+ result = httpx .put (request_url , headers = headers )
47
47
assert result .status_code == 200
48
48
old_version = headers ["EX-APP-VERSION" ]
49
49
headers ["EX-APP-VERSION" ] = "999.0.0"
50
50
sign_request (headers )
51
- result = requests .put (request_url , headers = headers )
51
+ result = httpx .put (request_url , headers = headers )
52
52
assert result .status_code == 401
53
53
headers ["EX-APP-VERSION" ] = old_version
54
54
sign_request (headers )
55
- result = requests .put (request_url , headers = headers )
55
+ result = httpx .put (request_url , headers = headers )
56
56
assert result .status_code == 200
0 commit comments