Merge pull request #1040 from MobilityData/1024-load-tests-are-failing-with-gcp-token-message

fix: 1024 load tests are failing with gcp token message

Author: qcdyx

load-test/gtfs_user_test.py

@@ -7,7 +7,7 @@
 
 class gtfs_user(HttpUser):
 
-    wait_time = between(.1, 1)
+    wait_time = between(0.1, 1)
 
     def print_response(self, response, indent):
         print(indent, "Contents of response:")
@@ -26,19 +26,18 @@ def get_valid(self, endpoint, allow404=False):
                 print("Error in response.")
                 self.print_response(response, "")
                 sys.exit(1)
-            # json_response = response.json()  # Try to parse response content as JSON
         except json.JSONDecodeError:
             print("Error: Response not json.")
             self.print_response(response, "")
             sys.exit(1)
 
     def on_start(self):
-        # access_token = os.environ.get('FEEDS_AUTH_TOKEN')
-        # if access_token is None or access_token == "":
-        #     print("Error: FEEDS_AUTH_TOKEN is not defined or empty")
-        #     sys.exit(1)
-        # self.client.headers = {'Authorization': "Bearer " + access_token}
-        pass
+        access_token = os.environ.get("FEEDS_AUTH_TOKEN")
+        if access_token is None or access_token == "":
+            print("Warning: FEEDS_AUTH_TOKEN is not defined or empty. Skipping tests.")
+            self.client.headers = {}  # Clear headers to indicate no token
+        else:
+            self.client.headers = {"Authorization": "Bearer " + access_token}
 
     @task
     def feeds(self):
@@ -47,15 +46,15 @@ def feeds(self):
     @task
     def feed_byId(self):
         # Allow error 404 since we are not sure the feed ID exists
-        self.get_valid("/v1/feeds/mdb-10", allow404=True)
+        self.get_valid("/v1/feeds/mdb-5", allow404=True)
 
     @task
     def gtfs_feeds(self):
         self.get_valid("/v1/gtfs_feeds?limit=1000")
 
     @task
     def gtfs_feed_byId(self):
-        self.get_valid("/v1/gtfs_feeds/mdb-10", allow404=True)
+        self.get_valid("/v1/gtfs_feeds/mdb-5", allow404=True)
 
     @task
     def gtfs_realtime_feeds(self):
@@ -67,8 +66,4 @@ def gtfs_realtime_feed_byId(self):
 
     @task
     def gtfs_feeds_datasets(self):
-        self.get_valid("/v1/gtfs_feeds/mdb-10/datasets", allow404=True)
-
-    # @task
-    # def gtfs_dataset(self):
-    #     self.get_valid("/v1/datasets/gtfs/mdb-10-202402071805", allow404=True)
+        self.get_valid("/v1/gtfs_feeds/mdb-5/datasets", allow404=True)