mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-11 18:11:21 +08:00
35 lines
954 B
Python
35 lines
954 B
Python
import json
|
|
import yaml
|
|
import os.path
|
|
from ate.exception import ParamsError
|
|
|
|
def load_yaml_file(yaml_file):
|
|
with open(yaml_file, 'r+') as stream:
|
|
return yaml.load(stream)
|
|
|
|
def load_json_file(json_file):
|
|
with open(json_file) as data_file:
|
|
return json.load(data_file)
|
|
|
|
def load_testcases(testcase_file_path):
|
|
file_suffix = os.path.splitext(testcase_file_path)[1]
|
|
if file_suffix == '.json':
|
|
return load_json_file(testcase_file_path)
|
|
elif file_suffix in ['.yaml', '.yml']:
|
|
return load_yaml_file(testcase_file_path)
|
|
else:
|
|
# '' or other suffix
|
|
raise ParamsError("Bad testcase file name!")
|
|
|
|
def parse_response_object(resp_obj):
|
|
try:
|
|
resp_content = resp_obj.json()
|
|
except json.decoder.JSONDecodeError:
|
|
resp_content = resp_obj.text
|
|
|
|
return {
|
|
'status_code': resp_obj.status_code,
|
|
'headers': resp_obj.headers,
|
|
'content': resp_content
|
|
}
|