diff --git a/apps/desktop/src/app/shell/model-catalog-menu.test.tsx b/apps/desktop/src/app/shell/model-catalog-menu.test.tsx index 407a1fb764..a777e1e269 100644 --- a/apps/desktop/src/app/shell/model-catalog-menu.test.tsx +++ b/apps/desktop/src/app/shell/model-catalog-menu.test.tsx @@ -111,12 +111,15 @@ describe('the reasoning-effort badge (#51833)', () => { expect(badge.className).toContain('rounded-sm') // …as a SIBLING of the truncating model-name span, so it can never read as - // part of a differently-named model. - const nameSpan = badge.previousElementSibling + // part of a differently-named model. The `-flash` variant tag is its own + // chip between them (#118083); the name itself stays free of both. + const nameSpan = badge.parentElement?.querySelector('.truncate') expect(nameSpan?.className).toContain('truncate') expect(nameSpan?.contains(badge)).toBe(false) - expect(nameSpan?.textContent?.toLowerCase()).toContain('gemini 2.5 flash') + expect(nameSpan?.textContent?.toLowerCase()).toContain('gemini 2.5') + expect(nameSpan?.textContent?.toLowerCase()).not.toContain('flash') + expect(nameSpan?.textContent?.toLowerCase()).not.toContain('high') }) it('drops the effort badge entirely when the model has no reasoning support', async () => { @@ -133,7 +136,7 @@ describe('the reasoning-effort badge (#51833)', () => { renderMenu({ effort: 'high', model: 'gemini-2.5-flash', provider: 'google' }) - await screen.findByText(/Gemini 2\.5 Flash/i) + await screen.findByText('Gemini 2.5') await waitFor(() => { expect(screen.queryByText('High')).toBeNull()