diff --git a/tests/outputs/login_plugin.py b/tests/outputs/login_plugin.py index 697e37d..a3ee61a 100644 --- 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 index e817bf9..5930658 100644 --- 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):