mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-15 04:19:28 +08:00
145 lines
4.0 KiB
Python
145 lines
4.0 KiB
Python
""" upload test extension.
|
|
|
|
If you want to use this extension, you should install the following dependencies first.
|
|
|
|
- requests_toolbelt
|
|
- filetype
|
|
|
|
Then you can write upload test script as below:
|
|
|
|
- test:
|
|
name: upload file
|
|
request:
|
|
url: http://httpbin.org/upload
|
|
method: POST
|
|
headers:
|
|
Cookie: session=AAA-BBB-CCC
|
|
upload:
|
|
file: "data/file_to_upload"
|
|
field1: "value1"
|
|
field2: "value2"
|
|
validate:
|
|
- eq: ["status_code", 200]
|
|
|
|
For compatibility, you can also write upload test script in old way:
|
|
|
|
- test:
|
|
name: upload file
|
|
variables:
|
|
file: "data/file_to_upload"
|
|
field1: "value1"
|
|
field2: "value2"
|
|
m_encoder: ${multipart_encoder(file=$file, field1=$field1, field2=$field2)}
|
|
request:
|
|
url: http://httpbin.org/upload
|
|
method: POST
|
|
headers:
|
|
Content-Type: ${multipart_content_type($m_encoder)}
|
|
Cookie: session=AAA-BBB-CCC
|
|
data: $m_encoder
|
|
validate:
|
|
- eq: ["status_code", 200]
|
|
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
|
|
try:
|
|
import filetype
|
|
from requests_toolbelt import MultipartEncoder
|
|
except ImportError:
|
|
msg = """
|
|
uploader extension dependencies uninstalled, install first and try again.
|
|
install with pip:
|
|
$ pip install requests_toolbelt filetype
|
|
"""
|
|
print(msg)
|
|
sys.exit(0)
|
|
|
|
from httprunner.exceptions import ParamsError
|
|
|
|
|
|
def prepare_upload_test(test_dict):
|
|
""" preprocess for upload test
|
|
replace `upload` info with MultipartEncoder
|
|
|
|
Args:
|
|
test_dict (dict):
|
|
|
|
{
|
|
"variables": {},
|
|
"request": {
|
|
"url": "http://httpbin.org/upload",
|
|
"method": "POST",
|
|
"headers": {
|
|
"Cookie": "session=AAA-BBB-CCC"
|
|
},
|
|
"upload": {
|
|
"file": "data/file_to_upload"
|
|
"md5": "123"
|
|
}
|
|
}
|
|
}
|
|
|
|
"""
|
|
upload_json = test_dict["request"].pop("upload", {})
|
|
if not upload_json:
|
|
raise ParamsError(f"invalid upload info: {upload_json}")
|
|
|
|
params_list = []
|
|
for key, value in upload_json.items():
|
|
test_dict["variables"][key] = value
|
|
params_list.append(f"{key}=${key}")
|
|
|
|
params_str = ", ".join(params_list)
|
|
test_dict["variables"]["m_encoder"] = "${multipart_encoder(" + params_str + ")}"
|
|
|
|
test_dict["request"].setdefault("headers", {})
|
|
test_dict["request"]["headers"]["Content-Type"] = "${multipart_content_type($m_encoder)}"
|
|
|
|
test_dict["request"]["data"] = "$m_encoder"
|
|
|
|
|
|
def multipart_encoder(**kwargs):
|
|
""" initialize MultipartEncoder with uploading fields.
|
|
"""
|
|
|
|
def get_filetype(file_path):
|
|
file_type = filetype.guess(file_path)
|
|
if file_type:
|
|
return file_type.mime
|
|
else:
|
|
return "text/html"
|
|
|
|
fields_dict = {}
|
|
for key, value in kwargs.items():
|
|
|
|
if os.path.isabs(value):
|
|
# value is absolute file path
|
|
_file_path = value
|
|
is_exists_file = os.path.isfile(value)
|
|
else:
|
|
# value is not absolute file path, check if it is relative file path
|
|
from httprunner.loader import get_pwd
|
|
_file_path = os.path.join(get_pwd(), value)
|
|
is_exists_file = os.path.isfile(_file_path)
|
|
|
|
if is_exists_file:
|
|
# value is file path to upload
|
|
filename = os.path.basename(_file_path)
|
|
mime_type = get_filetype(_file_path)
|
|
# TODO: fix ResourceWarning for unclosed file
|
|
file_handler = open(_file_path, 'rb')
|
|
fields_dict[key] = (filename, file_handler, mime_type)
|
|
else:
|
|
fields_dict[key] = value
|
|
|
|
return MultipartEncoder(fields=fields_dict)
|
|
|
|
|
|
def multipart_content_type(m_encoder):
|
|
""" prepare Content-Type for request headers
|
|
"""
|
|
return m_encoder.content_type
|