From 1ba7c606e679bdcd1212c23a46146aed5646c504 Mon Sep 17 00:00:00 2001
From: Adam <2363879+adamdotdevin@users.noreply.github.com>
Date: Mon, 19 Jan 2026 08:38:42 -0600
Subject: chore: cleanup

---
 packages/app/e2e/context.spec.ts | 25 +++++++++++++++++++++----
 1 file changed, 21 insertions(+), 4 deletions(-)

(limited to 'packages/app/e2e/context.spec.ts')

diff --git a/packages/app/e2e/context.spec.ts b/packages/app/e2e/context.spec.ts
index dbfeda8d8..beabd2eb7 100644
--- a/packages/app/e2e/context.spec.ts
+++ b/packages/app/e2e/context.spec.ts
@@ -9,12 +9,29 @@ test("context panel can be opened from the prompt", async ({ page, sdk, gotoSess
   const sessionID = created.id
 
   try {
+    await sdk.session.promptAsync({
+      sessionID,
+      noReply: true,
+      parts: [
+        {
+          type: "text",
+          text: "seed context",
+        },
+      ],
+    })
+
+    await expect
+      .poll(async () => {
+        const messages = await sdk.session.messages({ sessionID, limit: 1 }).then((r) => r.data ?? [])
+        return messages.length
+      })
+      .toBeGreaterThan(0)
+
     await gotoSession(sessionID)
 
-    const promptForm = page.locator("form").filter({ has: page.locator(promptSelector) }).first()
-    const contextButton = promptForm
-      .locator("button")
-      .filter({ has: promptForm.locator('[data-component="progress-circle"]').first() })
+    const contextButton = page
+      .locator('[data-component="button"]')
+      .filter({ has: page.locator('[data-component="progress-circle"]').first() })
       .first()
 
     await expect(contextButton).toBeVisible()
-- 
cgit v1.2.3