Import trio before gevent patching if LOCUST_PLAYWRIGHT is set. For s…

…upporting locust-plugins's Playwright User.
locustio · Feb 6, 2022 · aedc3b9 · aedc3b9
1 parent b14464c
commit aedc3b9
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/locust/__init__.py b/locust/__init__.py
@@ -1,3 +1,11 @@
+import os
+
+if os.getenv("LOCUST_PLAYWRIGHT", False):
+    # This is a hack to make Playwright testing possible. If trio is not imported before gevent's monkey patching,
+    # it raises "NotImplementedError: unsupported platform"
+    # Playwright is used by PlaywrightUser, see https://github.com/SvenskaSpel/locust-plugins/blob/master/examples/playwright_ex.py
+    import trio
+
 from gevent import monkey
 
 monkey.patch_all()