fix: ImportError when locustio is not installed

This commit is contained in:
debugtalk
2019-06-25 20:53:25 +08:00
parent 2222b27634
commit 38fe359151
2 changed files with 6 additions and 7 deletions

View File

@@ -11,6 +11,7 @@
**Bugfixes**
- fix extend with testcase reference in format version 2
- fix ImportError when locustio is not installed
## 2.2.0 (2019-06-24)

View File

@@ -91,14 +91,12 @@ def main_hrun():
def main_locust():
""" Performance test with locust: parse command line options and run commands.
"""
# monkey patch ssl at beginning to avoid RecursionError when running locust.
from gevent import monkey; monkey.patch_ssl()
import multiprocessing
import sys
from httprunner import logger
try:
# monkey patch ssl at beginning to avoid RecursionError when running locust.
from gevent import monkey; monkey.patch_ssl()
import multiprocessing
import sys
from httprunner import logger
from httprunner import locusts
except ImportError:
msg = "Locust is not installed, install first and try again.\n"