22from os import environ
33from sys import argv
44
5- import requests
5+ import httpx
66
77
88def sign_request (req_headers : dict , secret = None , user : str = "" ):
@@ -14,43 +14,43 @@ def sign_request(req_headers: dict, secret=None, user: str = ""):
1414if __name__ == "__main__" :
1515 request_url = argv [1 ] + "/sec_check?value=1"
1616 headers = {}
17- result = requests .put (request_url , headers = headers )
17+ result = httpx .put (request_url , headers = headers )
1818 assert result .status_code == 401 # Missing headers
1919 headers .update ({
2020 "AA-VERSION" : environ .get ("AA_VERSION" , "1.0.0" ),
2121 "EX-APP-ID" : environ .get ("APP_ID" , "nc_py_api" ),
2222 "EX-APP-VERSION" : environ .get ("APP_VERSION" , "1.0.0" ),
2323 })
2424 sign_request (headers )
25- result = requests .put (request_url , headers = headers )
25+ result = httpx .put (request_url , headers = headers )
2626 assert result .status_code == 200
2727 # Invalid AA-SIGNATURE
2828 sign_request (headers , secret = "xxx" )
29- result = requests .put (request_url , headers = headers )
29+ result = httpx .put (request_url , headers = headers )
3030 assert result .status_code == 401
3131 sign_request (headers )
32- result = requests .put (request_url , headers = headers )
32+ result = httpx .put (request_url , headers = headers )
3333 assert result .status_code == 200
3434 # Invalid EX-APP-ID
3535 old_app_name = headers ["EX-APP-ID" ]
3636 headers ["EX-APP-ID" ] = "unknown_app"
3737 sign_request (headers )
38- result = requests .put (request_url , headers = headers )
38+ result = httpx .put (request_url , headers = headers )
3939 assert result .status_code == 401
4040 headers ["EX-APP-ID" ] = old_app_name
4141 sign_request (headers )
42- result = requests .put (request_url , headers = headers )
42+ result = httpx .put (request_url , headers = headers )
4343 assert result .status_code == 200
4444 # Invalid EX-APP-VERSION
4545 sign_request (headers )
46- result = requests .put (request_url , headers = headers )
46+ result = httpx .put (request_url , headers = headers )
4747 assert result .status_code == 200
4848 old_version = headers ["EX-APP-VERSION" ]
4949 headers ["EX-APP-VERSION" ] = "999.0.0"
5050 sign_request (headers )
51- result = requests .put (request_url , headers = headers )
51+ result = httpx .put (request_url , headers = headers )
5252 assert result .status_code == 401
5353 headers ["EX-APP-VERSION" ] = old_version
5454 sign_request (headers )
55- result = requests .put (request_url , headers = headers )
55+ result = httpx .put (request_url , headers = headers )
5656 assert result .status_code == 200
0 commit comments