gen_md5: pass in *args

ate/utils.py

@@ -19,9 +19,8 @@ def gen_random_string(str_len):
     return ''.join(
         random.choice(string.ascii_letters + string.digits) for _ in range(str_len))
 
-def gen_md5(str_list):
+def gen_md5(*str_args):
-    authorization_str = "".join(str_list)
+    return hashlib.md5("".join(str_args).encode('utf-8')).hexdigest()
-    return hashlib.md5(authorization_str.encode('utf-8')).hexdigest()
 
 def handle_req_data(data):
 

test/api_server.py

@@ -37,7 +37,7 @@ def validate_request(func):
             req_authorization = req_headers['Authorization']
             random_str = req_headers['Random']
             data = utils.handle_req_data(request.data)
-            authorization = utils.gen_md5([TOKEN, data, random_str])
+            authorization = utils.gen_md5(TOKEN, data, random_str)
             assert authorization == req_authorization
             return func(*args, **kwds)
         except (KeyError, AssertionError):

test/base.py

@@ -29,7 +29,7 @@ class ApiServerUnittest(unittest.TestCase):
         token = api_server.TOKEN
         data = utils.handle_req_data(data)
         random_str = utils.gen_random_string(5)
-        authorization = utils.gen_md5([token, data, random_str])
+        authorization = utils.gen_md5(token, data, random_str)
 
         headers = {
             'authorization': authorization,