restore test_load.py

microsoft · Sep 5, 2023 · 1ee2573 · 1ee2573
1 parent a5ec4ac
commit 1ee2573
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/test/test_load.py b/test/test_load.py
@@ -10,16 +10,12 @@
 from pyscitt import crypto
 from pyscitt.client import Client
 
-NUM_CLAIMS = 100
-LOCUST_PEAK_USERS = 500
-LOCUST_USERS_SPAWN_RATE = 10
-LOCUST_RUNTIME_SEC = 60
 
 @pytest.mark.perf
 @pytest.mark.disable_proxy
 class TestLoad:
     def test_load(self, client: Client, did_web, tmp_path: Path):
-        for i in range(NUM_CLAIMS):
+        for i in range(100):
             identity = did_web.create_identity()
             claim = crypto.sign_json_claimset(identity, {"foo": "bar"})
             (tmp_path / f"claim{i}.cose").write_bytes(claim)
@@ -36,11 +32,11 @@ def test_load(self, client: Client, did_web, tmp_path: Path):
                     "--host",
                     client.url,
                     "--users",
-                    str(LOCUST_PEAK_USERS),
+                    "100",
                     "--spawn-rate",
-                    str(LOCUST_USERS_SPAWN_RATE),
+                    "10",
                     "--run-time",
-                    str(LOCUST_RUNTIME_SEC) + "s",
+                    "1m",
                     "--scitt-claims",
                     str(tmp_path),
                 ],