fix: NameError, name 'locust_main' is not defined

docs/CHANGELOG.md

@@ -1,12 +1,16 @@
 # Release History
 
-## 3.1.2 (2020-06-28)
+## 3.1.2 (2020-06-29)
 
 **Fixed**
 
 - fix: missing setup/teardown hooks for referenced testcase
 - fix: compatibility for `black` on Android termux that does not support multiprocessing well
 
+**Changed**
+
+- change: import locust at beginning to monkey patch all modules
+
 ## 3.1.1 (2020-06-23)
 
 **Added**

httprunner/ext/locust/__init__.py

@@ -1,16 +1,12 @@
-import importlib.util
-import inspect
-import os
 import sys
-from typing import List
 
-if sys.argv[0].endswith("locusts"):
+if "locust" in sys.argv[0]:
     try:
-        # monkey patch ssl at beginning to avoid RecursionError when running locust.
-        from gevent import monkey
-
-        monkey.patch_ssl()
+        # monkey patch all at beginning to avoid RecursionError when running locust.
+        # `from gevent import monkey; monkey.patch_all()` will be triggered when importing locust
         from locust import main as locust_main
+
+        print("NOTICE: gevent monkey patches have been applied !!!")
     except ImportError:
         msg = """
 Locust is not installed, install first and try again.
@@ -20,9 +16,14 @@ $ pip install locust
         print(msg)
         sys.exit(1)
 
+import importlib.util
+import inspect
+import os
+from typing import List
 
 from loguru import logger
 
+
 """ converted pytest files from YAML/JSON testcases
 """
 pytest_files: List = []