Update plugin tests

SvenskaSpel · May 30, 2024 · 8535222 · 8535222
1 parent 6cc9a66
commit 8535222
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/outputs/login_plugin.py b/tests/outputs/login_plugin.py
@@ -2,10 +2,10 @@
 
 import os
 
-from locust_plugins.listeners import StopUserOnFail
+from locust_plugins import listeners
 from svs_locust import RestUser
 
-os.environ["LOCUST_TEST_ENV"] = "itp1"
+os.environ["LOCUST_TEST_ENV"] = "test3"
 os.environ["LOCUST_TENANT"] = "lb"
 
 
@@ -14,7 +14,7 @@ def on_locust_init(environment, **_kwargs):
     import time
 
     environment.events.request.add_listener(lambda *args, **kw: time.sleep(0.1))
-    StopUserOnFail(environment)
+    listeners.StopUserOnFail(environment)
 
 
 class login(RestUser):

diff --git a/tests/outputs/reqres_plugin.py b/tests/outputs/reqres_plugin.py
@@ -3,10 +3,10 @@
 import os
 import re
 
-from locust_plugins.listeners import StopUserOnFail
+from locust_plugins import listeners
 from svs_locust import RestUser
 
-os.environ["LOCUST_TEST_ENV"] = "itp1"
+os.environ["LOCUST_TEST_ENV"] = "test3"
 os.environ["LOCUST_TENANT"] = "lb"
 
 
@@ -15,7 +15,7 @@ def on_locust_init(environment, **_kwargs):
     import time
 
     environment.events.request.add_listener(lambda *args, **kw: time.sleep(0.1))
-    StopUserOnFail(environment)
+    listeners.StopUserOnFail(environment)
 
 
 class reqres_in(RestUser):