Update generated test files

SvenskaSpel · Mar 15, 2024 · 2757da7 · 2757da7
1 parent 8e06a1a
commit 2757da7
Show file tree

Hide file tree

Showing 6 changed files with 725 additions and 44 deletions.
diff --git a/tests/outputs/apple-buy-a-mac.py b/tests/outputs/apple-buy-a-mac.py
diff --git a/tests/outputs/login.py b/tests/outputs/login.py
@@ -16,20 +16,14 @@ def t(self):
         with self.rest(
             "POST",
             "/player/1/authenticate/testlogin",
-            headers={
-                "accept": "application/json, text/javascript, */*; q=0.01",
-                "origin": "https://spela.test3.svenskaspel.se",
-            },
+            headers={"accept": "application/json, text/javascript, */*; q=0.01"},
             json={"personalId": "193804122491", "source": 3},
         ) as resp:
             pass
         with self.rest(
             "GET",
             "/player/1/customizedsettings?_=1636025335990",
-            headers={
-                "accept": "application/json, text/javascript, */*; q=0.01",
-                "origin": "https://spela.test3.svenskaspel.se",
-            },
+            headers={"accept": "application/json, text/javascript, */*; q=0.01"},
         ) as resp:
             pass
         with self.client.request(
@@ -51,22 +45,13 @@ def t(self):
         ) as resp:
             pass
         with self.rest(
-            "POST",
-            "/player/1/terms",
-            headers={
-                "accept": "application/json, text/javascript, */*; q=0.01",
-                "origin": "https://spela.test3.svenskaspel.se",
-            },
-            json={},
+            "POST", "/player/1/terms", headers={"accept": "application/json, text/javascript, */*; q=0.01"}, json={}
         ) as resp:
             pass
         with self.rest(
             "GET",
             "/player/1/info?include=accountBalance&_=1636025343876",
-            headers={
-                "accept": "application/json, text/javascript, */*; q=0.01",
-                "origin": "https://spela.test3.svenskaspel.se",
-            },
+            headers={"accept": "application/json, text/javascript, */*; q=0.01"},
         ) as resp:
             pass
 

diff --git a/tests/outputs/login_disable_rest.py b/tests/outputs/login_disable_rest.py
@@ -16,23 +16,15 @@ def t(self):
         with self.client.request(
             "POST",
             "/player/1/authenticate/testlogin",
-            headers={
-                "accept": "application/json, text/javascript, */*; q=0.01",
-                "content-type": "application/json",
-                "origin": "https://spela.test3.svenskaspel.se",
-            },
+            headers={"accept": "application/json, text/javascript, */*; q=0.01", "content-type": "application/json"},
             data='{"personalId":"193804122491","source":3}',
             catch_response=True,
         ) as resp:
             pass
         with self.client.request(
             "GET",
             "/player/1/customizedsettings?_=1636025335990",
-            headers={
-                "accept": "application/json, text/javascript, */*; q=0.01",
-                "content-type": "text/plain",
-                "origin": "https://spela.test3.svenskaspel.se",
-            },
+            headers={"accept": "application/json, text/javascript, */*; q=0.01", "content-type": "text/plain"},
             catch_response=True,
         ) as resp:
             pass
@@ -57,23 +49,15 @@ def t(self):
         with self.client.request(
             "POST",
             "/player/1/terms",
-            headers={
-                "accept": "application/json, text/javascript, */*; q=0.01",
-                "content-type": "application/json",
-                "origin": "https://spela.test3.svenskaspel.se",
-            },
+            headers={"accept": "application/json, text/javascript, */*; q=0.01", "content-type": "application/json"},
             data="{}",
             catch_response=True,
         ) as resp:
             pass
         with self.client.request(
             "GET",
             "/player/1/info?include=accountBalance&_=1636025343876",
-            headers={
-                "accept": "application/json, text/javascript, */*; q=0.01",
-                "content-type": "text/plain",
-                "origin": "https://spela.test3.svenskaspel.se",
-            },
+            headers={"accept": "application/json, text/javascript, */*; q=0.01", "content-type": "text/plain"},
             catch_response=True,
         ) as resp:
             pass

diff --git a/tests/outputs/login_plugin.py b/tests/outputs/login_plugin.py
@@ -1,7 +1,21 @@
-from locust import run_single_user, task
+from locust import events, run_single_user, task
 
+import os
+
+from locust_plugins.listeners import StopUserOnFail
 from svs_locust import RestUser
 
+os.environ["LOCUST_TEST_ENV"] = "itp1"
+os.environ["LOCUST_TENANT"] = "lb"
+
+
+@events.init.add_listener
+def on_locust_init(environment, **_kwargs):
+    import time
+
+    environment.events.request.add_listener(lambda *args, **kw: time.sleep(0.1))
+    StopUserOnFail(environment)
+
 
 class login(RestUser):
     default_headers = {

diff --git a/tests/outputs/reqres.in.py b/tests/outputs/reqres.in.py
@@ -9,7 +9,6 @@ class reqres_in(FastHttpUser):
         "accept-language": "sv,en;q=0.9",
         "cache-control": "no-cache",
         "content-type": "application/json",
-        "origin": "https://reqres.in",
         "pragma": "no-cache",
         "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36",
     }

diff --git a/tests/outputs/reqres_plugin.py b/tests/outputs/reqres_plugin.py
@@ -1,9 +1,22 @@
-from locust import run_single_user, task
+from locust import events, run_single_user, task
 
+import os
 import re
 
+from locust_plugins.listeners import StopUserOnFail
 from svs_locust import RestUser
 
+os.environ["LOCUST_TEST_ENV"] = "itp1"
+os.environ["LOCUST_TENANT"] = "lb"
+
+
+@events.init.add_listener
+def on_locust_init(environment, **_kwargs):
+    import time
+
+    environment.events.request.add_listener(lambda *args, **kw: time.sleep(0.1))
+    StopUserOnFail(environment)
+
 
 class reqres_in(RestUser):
     default_headers = {