GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-09-05 23:51:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

bugfix #35: load YAML/JSON config base_url as locust host

Browse Source
This commit is contained in:
httprunner
2017-08-29 12:37:30 +08:00
parent 4ca827ada6
commit 969c3cce78
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ate/locusts.py
+5 -1
View File
@@ -3,6 +3,7 @@ import multiprocessing
import os import os
import sys import sys
from ate.utils import load_testcases_by_path
from locust.main import main from locust.main import main
@@ -35,10 +36,13 @@ def gen_locustfile(testcase_file_path):
os.path.dirname(os.path.realpath(__file__)), os.path.dirname(os.path.realpath(__file__)),
'locustfile_template' 'locustfile_template'
) )
testsets = load_testcases_by_path(testcase_file_path)
host = testsets[0].get("config", {}).get("request", {}).get("base_url", "")
with codecs.open(template_path, encoding='utf-8') as template: with codecs.open(template_path, encoding='utf-8') as template:
with codecs.open(locustfile_path, 'w', encoding='utf-8') as locustfile: with codecs.open(locustfile_path, 'w', encoding='utf-8') as locustfile:
template_content = template.read() template_content = template.read()
template_content = template_content.replace("$HOST", "https://skypixel.com") template_content = template_content.replace("$HOST", host)
template_content = template_content.replace("$TESTCASE_FILE", testcase_file_path) template_content = template_content.replace("$TESTCASE_FILE", testcase_file_path)
locustfile.write(template_content) locustfile.write(template_content)