From 11d17dafa443b8d32dee66f63ef15834ba926ffa Mon Sep 17 00:00:00 2001
From: Patrik Oldsberg <poldsberg@gmail.com>
Date: Sun, 14 Sep 2025 02:41:02 +0200
Subject: [PATCH] bui-themer: fix test

Signed-off-by: Patrik Oldsberg <poldsberg@gmail.com>
---
 .../BuiThemerPage/BuiThemePreview.test.tsx           | 12 +++++++-----
 .../components/BuiThemerPage/ThemeContent.test.tsx   |  2 +-
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/plugins/bui-themer/src/components/BuiThemerPage/BuiThemePreview.test.tsx b/plugins/bui-themer/src/components/BuiThemerPage/BuiThemePreview.test.tsx
index d42c648677..79a2f9e83e 100644
--- a/plugins/bui-themer/src/components/BuiThemerPage/BuiThemePreview.test.tsx
+++ b/plugins/bui-themer/src/components/BuiThemerPage/BuiThemePreview.test.tsx
@@ -14,8 +14,9 @@
  * limitations under the License.
  */
 
-import { render, screen } from '@testing-library/react';
+import { screen } from '@testing-library/react';
 import '@testing-library/jest-dom';
+import { renderInTestApp } from '@backstage/test-utils';
 import { BuiThemePreview } from './BuiThemePreview';
 
 describe('BuiThemePreview', () => {
@@ -34,8 +35,8 @@ describe('BuiThemePreview', () => {
       })),
     });
   });
-  it('renders headings and sample components', () => {
-    const { container } = render(
+  it('renders headings and sample components', async () => {
+    const { container } = await renderInTestApp(
       <BuiThemePreview
         mode="light"
         styleObject={{
@@ -53,7 +54,7 @@ describe('BuiThemePreview', () => {
     );
 
     expect(container.firstChild).toHaveAttribute('data-theme-mode', 'light');
-    expect(screen.getByText('Theme Preview')).toBeInTheDocument();
+    expect(await screen.findByText('Button Variants')).toBeInTheDocument();
     expect(screen.getByText('Button Variants')).toBeInTheDocument();
     expect(screen.getByRole('button', { name: 'Primary' })).toBeInTheDocument();
     expect(
@@ -63,6 +64,7 @@ describe('BuiThemePreview', () => {
       screen.getByRole('button', { name: 'Tertiary' }),
     ).toBeInTheDocument();
     expect(screen.getByText('Form Inputs')).toBeInTheDocument();
-    expect(screen.getByText('Surface Variations')).toBeInTheDocument();
+    expect(screen.getByText('Tag Variants')).toBeInTheDocument();
+    expect(screen.getByText('Text Variants')).toBeInTheDocument();
   });
 });
diff --git a/plugins/bui-themer/src/components/BuiThemerPage/ThemeContent.test.tsx b/plugins/bui-themer/src/components/BuiThemerPage/ThemeContent.test.tsx
index 720c46d12d..8ece77ad0e 100644
--- a/plugins/bui-themer/src/components/BuiThemerPage/ThemeContent.test.tsx
+++ b/plugins/bui-themer/src/components/BuiThemerPage/ThemeContent.test.tsx
@@ -94,6 +94,6 @@ describe('ThemeContent', () => {
 
     fireEvent.click(screen.getByText('Live Preview'));
     // The preview renders content from BuiThemePreview; basic smoke check:
-    expect(screen.getByText('Theme Preview')).toBeInTheDocument();
+    expect(await screen.findByText('Button Variants')).toBeInTheDocument();
   });
 });