[PRME-169] pre-prod domain fix for mock login (#736)

Author: NogaNHS

lambdas/services/login_redirect_service.py

@@ -25,7 +25,7 @@ def configure_oidc(self) -> WebApplicationClient:
             client_id=self.oidc_parameters[self.ssm_prefix + "OIDC_CLIENT_ID"],
         )
 
-    def prepare_redirect_response(self):
+    def prepare_redirect_response(self, *args, **kwargs):
         try:
             self.oidc_parameters = self.ssm_service.get_ssm_parameters(
                 [

lambdas/services/mock_login_redirect_service.py

@@ -1,3 +1,4 @@
+import os
 import random
 import re
 import string
@@ -24,8 +25,11 @@ def prepare_redirect_response(self, event):
 
         state = "".join(random.choices(string.ascii_letters + string.digits, k=30))
         self.save_state_in_dynamo_db(state)
+        if os.getenv("WORKSPACE") == "pre-prod":
+            clean_url = re.sub(r"^api.", "", host)
+        else:
+            clean_url = re.sub(r"^api-", "", host)
 
-        clean_url = re.sub(r"^api-", "", host)
         url = f"https://{clean_url}{MOCK_LOGIN_ROUTE}?state={state}"
 
         location_header = {"Location": url}