@@ -45,7 +45,7 @@ async function seedConversation(input: {
4545 . toBe ( true )
4646
4747 if ( ! userMessageID ) throw new Error ( "Expected a user message id" )
48- await expect ( input . page . locator ( `[data-message-id="${ userMessageID } "]` ) . first ( ) ) . toBeVisible ( { timeout : 30_000 } )
48+ await expect ( input . page . locator ( `[data-message-id="${ userMessageID } "]` ) ) . toHaveCount ( 1 , { timeout : 30_000 } )
4949 return { prompt, userMessageID }
5050}
5151
@@ -123,7 +123,7 @@ test("slash redo clears revert and restores latest state", async ({ page, withPr
123123 . toBeUndefined ( )
124124
125125 await expect ( seeded . prompt ) . not . toContainText ( token )
126- await expect ( page . locator ( `[data-message-id="${ seeded . userMessageID } "]` ) . first ( ) ) . toBeVisible ( )
126+ await expect ( page . locator ( `[data-message-id="${ seeded . userMessageID } "]` ) ) . toHaveCount ( 1 )
127127 } )
128128 } )
129129} )
@@ -158,8 +158,8 @@ test("slash undo/redo traverses multi-step revert stack", async ({ page, withPro
158158 const firstMessage = page . locator ( `[data-message-id="${ first . userMessageID } "]` )
159159 const secondMessage = page . locator ( `[data-message-id="${ second . userMessageID } "]` )
160160
161- await expect ( firstMessage . first ( ) ) . toBeVisible ( )
162- await expect ( secondMessage . first ( ) ) . toBeVisible ( )
161+ await expect ( firstMessage ) . toHaveCount ( 1 )
162+ await expect ( secondMessage ) . toHaveCount ( 1 )
163163
164164 await second . prompt . click ( )
165165 await page . keyboard . press ( `${ modKey } +A` )
@@ -176,7 +176,7 @@ test("slash undo/redo traverses multi-step revert stack", async ({ page, withPro
176176 } )
177177 . toBe ( second . userMessageID )
178178
179- await expect ( firstMessage . first ( ) ) . toBeVisible ( )
179+ await expect ( firstMessage ) . toHaveCount ( 1 )
180180 await expect ( secondMessage ) . toHaveCount ( 0 )
181181
182182 await second . prompt . click ( )
@@ -210,7 +210,7 @@ test("slash undo/redo traverses multi-step revert stack", async ({ page, withPro
210210 } )
211211 . toBe ( second . userMessageID )
212212
213- await expect ( firstMessage . first ( ) ) . toBeVisible ( )
213+ await expect ( firstMessage ) . toHaveCount ( 1 )
214214 await expect ( secondMessage ) . toHaveCount ( 0 )
215215
216216 await second . prompt . click ( )
@@ -226,8 +226,8 @@ test("slash undo/redo traverses multi-step revert stack", async ({ page, withPro
226226 } )
227227 . toBeUndefined ( )
228228
229- await expect ( firstMessage . first ( ) ) . toBeVisible ( )
230- await expect ( secondMessage . first ( ) ) . toBeVisible ( )
229+ await expect ( firstMessage ) . toHaveCount ( 1 )
230+ await expect ( secondMessage ) . toHaveCount ( 1 )
231231 } )
232232 } )
233233} )
0 commit comments