mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 02:21:29 +08:00
43 lines
969 B
Python
43 lines
969 B
Python
import hashlib
|
|
import hmac
|
|
import json
|
|
import random
|
|
import string
|
|
import time
|
|
|
|
try:
|
|
string_type = basestring
|
|
PYTHON_VERSION = 2
|
|
import urllib
|
|
except NameError:
|
|
string_type = str
|
|
PYTHON_VERSION = 3
|
|
import urllib.parse as urllib
|
|
|
|
SECRET_KEY = "DebugTalk"
|
|
BASE_URL = "http://127.0.0.1:5000"
|
|
|
|
def get_sign(*args):
|
|
content = ''.join(args).encode('ascii')
|
|
sign_key = SECRET_KEY.encode('ascii')
|
|
sign = hmac.new(sign_key, content, hashlib.sha1).hexdigest()
|
|
return sign
|
|
|
|
get_sign_lambda = lambda *args: hmac.new(
|
|
'DebugTalk'.encode('ascii'),
|
|
''.join(args).encode('ascii'),
|
|
hashlib.sha1).hexdigest()
|
|
|
|
def gen_md5(*args):
|
|
return hashlib.md5("".join(args).encode('utf-8')).hexdigest()
|
|
|
|
def sum_status_code(status_code, expect_sum):
|
|
""" sum status code digits
|
|
e.g. 400 => 4, 201 => 3
|
|
"""
|
|
sum_value = 0
|
|
for digit in str(status_code):
|
|
sum_value += int(digit)
|
|
|
|
assert sum_value == expect_sum
|