From 42dde2bf8618ed527735b3610860de9b4cabffca Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Sat, 1 Aug 2026 16:12:36 +0800
Subject: [PATCH] test(ui): align reconnect assertions with selected model

---
 .../pages/model-setup/model-setup-reconnect.test.ts  | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/ui/src/pages/model-setup/model-setup-reconnect.test.ts b/ui/src/pages/model-setup/model-setup-reconnect.test.ts
index 6e4c91b9f1a7..e01aa9fde9e8 100644
--- a/ui/src/pages/model-setup/model-setup-reconnect.test.ts
+++ b/ui/src/pages/model-setup/model-setup-reconnect.test.ts
@@ -104,6 +104,10 @@ async function mountPage(
   return page;
 }
 
+function selectedModelDetail(page: TestModelSetupPage): string | undefined {
+  return page.querySelector(".model-setup__current-copy > .muted")?.textContent?.trim();
+}
+
 describe("ModelSetupPage Gateway reconnect ownership", () => {
   beforeEach(async () => {
     await i18n.setLocale("en");
@@ -193,7 +197,7 @@ describe("ModelSetupPage Gateway reconnect ownership", () => {
 
     const page = await mountPage(context, client, staleHello);
 
-    await vi.waitFor(() => expect(page.textContent).toContain("provider/fresh-model"));
+    await vi.waitFor(() => expect(selectedModelDetail(page)).toBe("fresh-model"));
     expect(
       request.mock.calls.filter(([method]) => method === "openclaw.setup.detect"),
     ).toHaveLength(1);
@@ -252,7 +256,7 @@ describe("ModelSetupPage Gateway reconnect ownership", () => {
 
     setGatewayPhase("reconnecting");
     setGatewayPhase("connected");
-    await vi.waitFor(() => expect(page.textContent).toContain("provider/current-model"));
+    await vi.waitFor(() => expect(selectedModelDetail(page)).toBe("current-model"));
     page.routeData = {
       state: {
         phase: "ready",
@@ -267,8 +271,8 @@ describe("ModelSetupPage Gateway reconnect ownership", () => {
     };
     await page.updateComplete;
 
-    expect(page.textContent).not.toContain("provider/stale-model");
-    await vi.waitFor(() => expect(page.textContent).toContain("provider/current-model"));
+    expect(selectedModelDetail(page)).not.toBe("stale-model");
+    await vi.waitFor(() => expect(selectedModelDetail(page)).toBe("current-model"));
     expect(
       request.mock.calls.filter(([method]) => method === "openclaw.setup.detect"),
     ).toHaveLength(1);