minor fix in tests

Author: whummer

tests/test_infra.py

@@ -5,12 +5,15 @@
 import json
 import time
 
+API_NAME = 'QuizAPI'
+STAGE_NAME = "prod"
+
+
 @pytest.fixture(scope='module')
 def api_endpoint():
     apigateway_client = boto3.client('apigateway', endpoint_url='http://localhost:4566')
     lambda_client = boto3.client('lambda', endpoint_url='http://localhost:4566')
 
-    API_NAME = 'QuizAPI'
     response = apigateway_client.get_rest_apis()
     api_list = response.get('items', [])
     api = next((item for item in api_list if item['name'] == API_NAME), None)
@@ -19,7 +22,7 @@ def api_endpoint():
         raise Exception(f"API {API_NAME} not found.")
 
     API_ID = api['id']
-    API_ENDPOINT = f"http://localhost:4566/_aws/execute-api/{API_ID}/prod"
+    API_ENDPOINT = f"http://localhost:4566/_aws/execute-api/{API_ID}/{STAGE_NAME}"
 
     print(f"API Endpoint: {API_ENDPOINT}")
 

tests/test_outage.py

@@ -9,7 +9,8 @@
 from localstack.sdk.chaos.managers import fault_configuration
 
 LOCALSTACK_ENDPOINT = "http://localhost.localstack.cloud:4566"
-API_NAME = 'QuizAPI'
+API_NAME = "QuizAPI"
+STAGE_NAME = "prod"
 
 class TestLocalStackClient:
     client = localstack.sdk.chaos.ChaosClient()
@@ -28,7 +29,7 @@ def api_endpoint(apigateway_client):
         raise Exception(f"API {API_NAME} not found.")
 
     API_ID = api['id']
-    API_ENDPOINT = f"{LOCALSTACK_ENDPOINT}/_aws/execute-api/{API_ID}/test"
+    API_ENDPOINT = f"{LOCALSTACK_ENDPOINT}/_aws/execute-api/{API_ID}/{STAGE_NAME}"
 
     print(f"API Endpoint: {API_ENDPOINT}")