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** **Bugfixes**
- fix extend with testcase reference in format version 2 - fix extend with testcase reference in format version 2
- fix ImportError when locustio is not installed
## 2.2.0 (2019-06-24) ## 2.2.0 (2019-06-24)

View File

@@ -91,14 +91,12 @@ def main_hrun():
def main_locust(): def main_locust():
""" Performance test with locust: parse command line options and run commands. """ 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: 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 from httprunner import locusts
except ImportError: except ImportError:
msg = "Locust is not installed, install first and try again.\n" msg = "Locust is not installed, install first and try again.\n"