Use selected deployment for Locust test

Author: nadove-ucsc

README.md

@@ -1100,13 +1100,14 @@ running it is straight-forward with your development environment set up.
 
    If it is not installed, perform step 2.3 in this README.
 
-2. To scale test the Azul web service on integration run
+2. To scale test the Azul web service, run
 
    ```
    locust -f scripts/locust/service.py
    ```
 
-   If you want to test against a different stage use the `--host` option:
+   By default, the tests will target the currently selected deployment.
+   If you want to test against a different deployment, use the `--host` option:
 
    ```
    locust -f scripts/locust/service.py --host https://service.dev.explore.data.humancellatlas.org

scripts/locust/service.py

@@ -36,7 +36,8 @@
     JSON,
 )
 
-# This script simluates a user triggering Azul endpoints via the data browser GUI.
+# This script simluates a user triggering Azul endpoints via the data browser
+# GUI.
 # To run:
 #  - `locust -f scripts/locust/service.py`
 #  - in browser go to localhost:8089
@@ -65,7 +66,10 @@ def parallel_requests():
     group.join()
 
 
-def browse_page(client: HttpSession, index_name: str, filters: JSON, **extra_index_params):
+def browse_page(client: HttpSession,
+                index_name: str,
+                filters: JSON,
+                **extra_index_params):
     filters = json.dumps(filters)
     with parallel_requests() as group:
         group.spawn(lambda: client.get(endpoint('/repository/summary',
@@ -94,7 +98,10 @@ def projects_start_page(self):
 
     @task
     def filter_mtx_files(self):
-        browse_page(self.client, 'projects', {"fileFormat": {"is": ["mtx"]}}, **browser_search_params)
+        browse_page(self.client,
+                    'projects',
+                    {"fileFormat": {"is": ["mtx"]}},
+                    **browser_search_params)
 
 
 class ManifestTaskSet(SequentialTaskSet):
@@ -104,7 +111,11 @@ class ManifestTaskSet(SequentialTaskSet):
 
     # Select islet of Langerhans since it's present in the develop deployment.
     organ_part_filter = {"organPart": {"is": ["islet of Langerhans"]}}
-    manifest_file_format_filter = {"fileFormat": {"is": ["fastq.gz", "bai", "bam", "csv", "results", "txt"]}}
+    manifest_file_format_filter = {
+        "fileFormat": {
+            "is": ["fastq.gz", "bai", "bam", "csv", "results", "txt"]
+        }
+    }
 
     @task
     def start_page(self):
@@ -119,7 +130,8 @@ def filter_organ_part(self):
 
     @task
     def download_manifest(self):
-        self.client.get(endpoint('/repository/summary', filters=json.dumps(self.organ_part_filter)),
+        self.client.get(endpoint('/repository/summary',
+                                 filters=json.dumps(self.organ_part_filter)),
                         name='/repository/summary')
         export_url = endpoint('/manifest/files',
                               filters=json.dumps({**self.organ_part_filter,
@@ -129,7 +141,8 @@ def download_manifest(self):
                              name='/manifest/files',
                              catch_response=True,
                              allow_redirects=False) as response:
-            # This is necessary because non 2xx response are counted as failures unless specified like this
+            # This is necessary because non 2xx response are counted as failures
+            # unless specified like this
             if response.status_code == 301 or 200 <= response.status_code < 300:
                 response.success()
         while response.status_code == 301:
@@ -201,7 +214,7 @@ def dos(self):
 
 
 class ServiceLocust(HttpUser):
-    host = 'https://service.integration.explore.data.humancellatlas.org'
+    host = config.service_endpoint()
     tasks = {
         MatrixTaskSet: 1,
         GA4GHTaskSet: 1,