more tweaks

test/ai.test.tsx

@@ -34,7 +34,8 @@ describe('RewriteButton', () => {
     generate.mockResolvedValue('Better text')
     const onAccept = vi.fn()
     render(<RewriteButton schema={{}} value="Bad txt" onAccept={onAccept} />)
-    fireEvent.change(screen.getByLabelText('Rewrite with AI'), { target: { value: 'fix' } })
+    fireEvent.click(screen.getByLabelText('Rewrite with AI'))
+    fireEvent.click(screen.getByText('Fix spelling & grammar'))
     await screen.findByLabelText('AI suggestion')
     expect(onAccept).not.toHaveBeenCalled()
     fireEvent.click(screen.getByText('Accept'))
@@ -44,7 +45,8 @@ describe('RewriteButton', () => {
     generate.mockResolvedValue('Other')
     const onAccept = vi.fn()
     render(<RewriteButton schema={{}} value="x" onAccept={onAccept} />)
-    fireEvent.change(screen.getByLabelText('Rewrite with AI'), { target: { value: 'improve' } })
+    fireEvent.click(screen.getByLabelText('Rewrite with AI'))
+    fireEvent.click(screen.getByText('Improve'))
     fireEvent.click(await screen.findByText('Reject'))
     expect(onAccept).not.toHaveBeenCalled()
     await waitFor(() => screen.getByLabelText('Rewrite with AI'))
@@ -52,7 +54,8 @@ describe('RewriteButton', () => {
   it('surfaces engine errors', async () => {
     generate.mockRejectedValue(new Error('boom'))
     render(<RewriteButton schema={{}} value="x" onAccept={() => {}} />)
-    fireEvent.change(screen.getByLabelText('Rewrite with AI'), { target: { value: 'improve' } })
+    fireEvent.click(screen.getByLabelText('Rewrite with AI'))
+    fireEvent.click(screen.getByText('Improve'))
     expect((await screen.findByRole('alert')).textContent).toBe('boom')
   })
 })