From cfd261d6795481d457bb002e4139d7cbf9c44ead Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Tue, 10 Feb 2026 16:30:31 -0800
Subject: [PATCH] Split e2e ui testing for browser

---
 .circleci/config.yml                          | 29 +++++++++++++++----
 .../scripts/e2e_tests/neonHelperScripts.ts    | 17 ++++++++---
 2 files changed, 37 insertions(+), 9 deletions(-)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index e171759f1c..ddd224eca6 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -3933,6 +3933,9 @@ jobs:
       image: ubuntu-2204:2023.10.1
     resource_class: xlarge
     working_directory: ~/project
+    parameters:
+      browser:
+        type: string
     steps:
       - checkout
       - setup_google_dns
@@ -3962,7 +3965,7 @@ jobs:
             echo "Expires at: $EXPIRES_AT"
             neon branches create \
               --project-id $NEON_PROJECT_ID \
-              --name preview/commit-${CIRCLE_SHA1:0:7} \
+              --name preview/commit-${CIRCLE_SHA1:0:7}-<< parameters.browser >> \
               --expires-at $EXPIRES_AT \
               --parent br-fancy-paper-ad1olsb3 \
               --api-key $NEON_API_KEY || true
@@ -3972,7 +3975,7 @@ jobs:
             E2E_UI_TEST_DATABASE_URL=$(neon connection-string \
               --project-id $NEON_PROJECT_ID \
               --api-key $NEON_API_KEY \
-              --branch preview/commit-${CIRCLE_SHA1:0:7} \
+              --branch preview/commit-${CIRCLE_SHA1:0:7}-<< parameters.browser >> \
               --database-name yuneng-trial-db \
               --role neondb_owner)
             echo $E2E_UI_TEST_DATABASE_URL
@@ -3984,7 +3987,7 @@ jobs:
               -e UI_USERNAME="admin" \
               -e UI_PASSWORD="gm" \
               -e LITELLM_LICENSE=$LITELLM_LICENSE \
-              --name litellm-docker-database \
+              --name litellm-docker-database-<< parameters.browser >> \
               -v $(pwd)/litellm/proxy/example_config_yaml/simple_config.yaml:/app/config.yaml \
               litellm-docker-database:ci \
               --config /app/config.yaml \
@@ -4000,7 +4003,7 @@ jobs:
             sudo rm dockerize-linux-amd64-v0.6.1.tar.gz
       - run:
           name: Start outputting logs
-          command: docker logs -f litellm-docker-database
+          command: docker logs -f litellm-docker-database-<< parameters.browser >>
           background: true
       - run:
           name: Wait for app to be ready
@@ -4009,6 +4012,7 @@ jobs:
           name: Run Playwright Tests
           command: |
             npx playwright test \
+              --project << parameters.browser >> \
               --config ui/litellm-dashboard/e2e_tests/playwright.config.ts \
               --reporter=html \
               --output=test-results
@@ -4214,6 +4218,20 @@ workflows:
                 - main
                 - /litellm_.*/
       - e2e_ui_testing:
+          name: e2e_ui_testing_chromium
+          browser: chromium
+          context: e2e_ui_tests
+          requires:
+            - ui_build
+            - build_docker_database_image
+          filters:
+            branches:
+              only:
+                - main
+                - /litellm_.*/
+      - e2e_ui_testing:
+          name: e2e_ui_testing_firefox
+          browser: firefox
           context: e2e_ui_tests
           requires:
             - ui_build
@@ -4525,7 +4543,8 @@ workflows:
             - litellm_assistants_api_testing
             - auth_ui_unit_tests
             - db_migration_disable_update_check
-            - e2e_ui_testing
+            - e2e_ui_testing_chromium
+            - e2e_ui_testing_firefox
             - litellm_proxy_unit_testing_key_generation
             - litellm_proxy_unit_testing_part1
             - litellm_proxy_unit_testing_part2
diff --git a/ui/litellm-dashboard/scripts/e2e_tests/neonHelperScripts.ts b/ui/litellm-dashboard/scripts/e2e_tests/neonHelperScripts.ts
index 3078a0d90d..089ad4e792 100644
--- a/ui/litellm-dashboard/scripts/e2e_tests/neonHelperScripts.ts
+++ b/ui/litellm-dashboard/scripts/e2e_tests/neonHelperScripts.ts
@@ -1,4 +1,4 @@
-import { createApiClient } from "@neondatabase/api-client";
+import { createApiClient, EndpointType } from "@neondatabase/api-client";
 import { config } from "dotenv";
 import { resolve } from "path";
 
@@ -27,6 +27,13 @@ export async function createNeonE2ETestingBranch(projectId: string, parentBranch
         parent_id: parentBranchId,
         expires_at: expireAt ?? new Date(Date.now() + 1000 * 60 * 30).toISOString(),
       },
+      endpoints: [
+        {
+          type: EndpointType.ReadWrite,
+          autoscaling_limit_min_cu: 0.25,
+          autoscaling_limit_max_cu: 1,
+        },
+      ],
     });
     return response;
   } catch (error) {
@@ -35,13 +42,15 @@ export async function createNeonE2ETestingBranch(projectId: string, parentBranch
 }
 
 export async function getNeonE2ETestingBranchConnectionString() {
-  await createNeonE2ETestingBranch(PROJECT_ID, PARENT_BRANCH);
-
+  const createBranchResponse = await createNeonE2ETestingBranch(PROJECT_ID, PARENT_BRANCH);
+  const projectId = createBranchResponse.data.branch.project_id;
   const response = await apiClient.getConnectionUri({
     database_name: NEON_E2E_UI_TEST_DB_NAME,
     role_name: "neondb_owner",
-    projectId: PROJECT_ID,
+    projectId: projectId,
   });
   console.log("connection string:", response.data.uri);
   return response.data.uri;
 }
+
+getNeonE2ETestingBranchConnectionString();