From 469877ba24056cfea7b6cc8816fe9c44f882cf1a Mon Sep 17 00:00:00 2001
From: Rik Claessens <rhmclaessens@gmail.com>
Date: Sat, 13 Jul 2024 11:32:19 +0200
Subject: [PATCH] Add icons as an optional property of renderInTestApp

Signed-off-by: Rik Claessens <rhmclaessens@gmail.com>
---
 packages/test-utils/src/testUtils/appWrappers.tsx | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/packages/test-utils/src/testUtils/appWrappers.tsx b/packages/test-utils/src/testUtils/appWrappers.tsx
index f2c1ff592e..3a55389b74 100644
--- a/packages/test-utils/src/testUtils/appWrappers.tsx
+++ b/packages/test-utils/src/testUtils/appWrappers.tsx
@@ -23,13 +23,14 @@ import React, {
 import { MemoryRouter, Route } from 'react-router-dom';
 import { themes, UnifiedThemeProvider } from '@backstage/theme';
 import MockIcon from '@material-ui/icons/AcUnit';
-import { createSpecializedApp } from '@backstage/core-app-api';
+import { AppIcons, createSpecializedApp } from '@backstage/core-app-api';
 import {
   AppComponents,
   attachComponentData,
   BootErrorPageProps,
   createRouteRef,
   ExternalRouteRef,
+  IconComponent,
   RouteRef,
 } from '@backstage/core-plugin-api';
 import { MatcherFunction, RenderResult } from '@testing-library/react';
@@ -107,6 +108,13 @@ export type TestAppOptions = {
    * Components to be forwarded to the `components` option of `createApp`.
    */
   components?: Partial<AppComponents>;
+
+  /**
+   * Icons to be forwarded to the `icons` option of `createApp`.
+   */
+  icons?: Partial<AppIcons> & {
+    [key in string]: IconComponent;
+  };
 };
 
 function isExternalRouteRef(
@@ -145,7 +153,10 @@ export function createTestAppWrapper(
       ),
       ...options.components,
     },
-    icons: mockIcons,
+    icons: {
+      ...mockIcons,
+      ...options.icons,
+    },
     plugins: [],
     themes: [
       {