@@ -73,7 +73,7 @@ describe('AiToastContent', () => {
7373 const title = target . querySelector ( '.ai-title' )
7474 expect ( title ?. textContent ) . toBe ( 'AI features available' )
7575
76- const buttons = target . querySelectorAll ( '.ai-button' )
76+ const buttons = target . querySelectorAll ( '.ai-actions button' )
7777 expect ( buttons ) . toHaveLength ( 3 )
7878 expect ( buttons [ 0 ] . textContent ) . toBe ( 'Download' )
7979 expect ( buttons [ 1 ] . textContent ) . toBe ( 'Not now' )
@@ -85,7 +85,7 @@ describe('AiToastContent', () => {
8585 const target = document . createElement ( 'div' )
8686 mount ( AiToastContent , { target } )
8787
88- const downloadButton = target . querySelector ( '.ai-button. primary' ) as HTMLButtonElement
88+ const downloadButton = target . querySelector ( '.btn- primary' ) as HTMLButtonElement
8989 downloadButton . click ( )
9090 flushSync ( )
9191
@@ -97,7 +97,7 @@ describe('AiToastContent', () => {
9797 const target = document . createElement ( 'div' )
9898 mount ( AiToastContent , { target } )
9999
100- const dismissButton = target . querySelector ( '.ai-button. secondary' ) as HTMLButtonElement
100+ const dismissButton = target . querySelector ( '.btn- secondary' ) as HTMLButtonElement
101101 dismissButton . click ( )
102102 flushSync ( )
103103
@@ -109,7 +109,7 @@ describe('AiToastContent', () => {
109109 const target = document . createElement ( 'div' )
110110 mount ( AiToastContent , { target } )
111111
112- const optOutButton = target . querySelector ( '.ai-button. tertiary' ) as HTMLButtonElement
112+ const optOutButton = target . querySelector ( '.tertiary-link ' ) as HTMLButtonElement
113113 optOutButton . click ( )
114114 flushSync ( )
115115
@@ -149,7 +149,7 @@ describe('AiToastContent', () => {
149149 const target = document . createElement ( 'div' )
150150 mount ( AiToastContent , { target } )
151151
152- const cancelButton = target . querySelector ( '.ai-button. secondary' ) as HTMLButtonElement
152+ const cancelButton = target . querySelector ( '.btn- secondary' ) as HTMLButtonElement
153153 cancelButton . click ( )
154154 flushSync ( )
155155
@@ -169,7 +169,7 @@ describe('AiToastContent', () => {
169169 expect ( description ?. textContent ) . toBe ( 'Starting inference server' )
170170
171171 // No buttons in installing state
172- expect ( target . querySelectorAll ( '.ai-button' ) ) . toHaveLength ( 0 )
172+ expect ( target . querySelectorAll ( '.ai-actions button' ) ) . toHaveLength ( 0 )
173173 } )
174174
175175 it ( 'renders ready state with Got it button' , ( ) => {
@@ -181,7 +181,7 @@ describe('AiToastContent', () => {
181181 const title = target . querySelector ( '.ai-title' )
182182 expect ( title ?. textContent ) . toBe ( 'AI ready' )
183183
184- const button = target . querySelector ( '.ai-button. primary' ) as HTMLButtonElement
184+ const button = target . querySelector ( '.btn- primary' ) as HTMLButtonElement
185185 expect ( button . textContent ) . toBe ( 'Got it' )
186186 } )
187187
@@ -191,7 +191,7 @@ describe('AiToastContent', () => {
191191 const target = document . createElement ( 'div' )
192192 mount ( AiToastContent , { target } )
193193
194- const button = target . querySelector ( '.ai-button. primary' ) as HTMLButtonElement
194+ const button = target . querySelector ( '.btn- primary' ) as HTMLButtonElement
195195 button . click ( )
196196 flushSync ( )
197197
0 commit comments