locust-cloud 1.21.10.dev1__tar.gz → 1.21.10.dev3__tar.gz

{locust_cloud-1.21.10.dev1 → locust_cloud-1.21.10.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust-cloud
-Version: 1.21.10.dev1
+Version: 1.21.10.dev3
 Summary: Locust Cloud
 Project-URL: homepage, https://locust.cloud
 Project-URL: repository, https://github.com/locustcloud/locust-cloud

{locust_cloud-1.21.10.dev1 → locust_cloud-1.21.10.dev3}/locust_cloud/args.py

@@ -230,6 +230,14 @@ cloud_parser.add_argument(
     default=None,
     help=configargparse.SUPPRESS,  # overrides the locust-cloud docker image tag. for internal use
 )
+cloud_parser.add_argument(
+    "--local-instance",
+    action="store_true",
+    default=False,
+    help=configargparse.SUPPRESS,
+    # for internal use. Assumes you've started locust/exporter with something like:
+    # LOCUSTCLOUD_SESSION_ID=valid-session-id LOCUST_WEB_LOGIN=1 LOCUST_LOGLEVEL=DEBUG python -m bootstrap
+)
 cloud_parser.add_argument(
     "--mock-server",
     action="store_true",

{locust_cloud-1.21.10.dev1 → locust_cloud-1.21.10.dev3}/locust_cloud/cloud.py

@@ -106,6 +106,11 @@ def main():
             payload["extra_packages"] = options.extra_packages
 
         for attempt in range(1, 16):
+            if options.local_instance:
+                response = requests.Response()
+                response.status_code = 200
+                js = {"log_ws_url": "ws://localhost:1095/socket-logs", "session_id": "valid-session-id"}
+                break
             try:
                 response = session.post("/deploy", json=payload)
                 js = response.json()
@@ -154,7 +159,10 @@ def main():
 
     except KeyboardInterrupt:
         logger.debug("Interrupted by user")
-        delete(session)
+        if options.local_instance:
+            os.system("pkill -TERM -f bootstrap")
+        else:
+            delete(session)
         try:
             websocket.wait(timeout=True)
         except (WebsocketTimeout, SessionMismatchError) as e: