From 6aba9d113d7ef824582238e31d0ac939fe7647c7 Mon Sep 17 00:00:00 2001
From: Camila Belo <camilaibs@gmail.com>
Date: Sat, 28 Jan 2023 11:06:01 +0100
Subject: [PATCH] fix(search): result extension stories

Signed-off-by: Camila Belo <camilaibs@gmail.com>
---
 .../components/SearchResult/SearchResult.stories.tsx | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/plugins/search-react/src/components/SearchResult/SearchResult.stories.tsx b/plugins/search-react/src/components/SearchResult/SearchResult.stories.tsx
index b491d66c83..f3086fc40a 100644
--- a/plugins/search-react/src/components/SearchResult/SearchResult.stories.tsx
+++ b/plugins/search-react/src/components/SearchResult/SearchResult.stories.tsx
@@ -15,14 +15,13 @@
  */
 
 import React, { ComponentType } from 'react';
-import { MemoryRouter } from 'react-router-dom';
 
 import { List, ListItem } from '@material-ui/core';
 import DefaultIcon from '@material-ui/icons/InsertDriveFile';
 import CustomIcon from '@material-ui/icons/NoteAdd';
 
 import { Link } from '@backstage/core-components';
-import { TestApiProvider } from '@backstage/test-utils';
+import { TestApiProvider, wrapInTestApp } from '@backstage/test-utils';
 import { createPlugin } from '@backstage/core-plugin-api';
 import { SearchDocument } from '@backstage/plugin-search-common';
 
@@ -71,15 +70,14 @@ export default {
   title: 'Plugins/Search/SearchResult',
   component: SearchResult,
   decorators: [
-    (Story: ComponentType<{}>) => (
-      <MemoryRouter>
+    (Story: ComponentType<{}>) =>
+      wrapInTestApp(
         <TestApiProvider apis={[[searchApiRef, searchApiMock]]}>
           <SearchContextProvider>
             <Story />
           </SearchContextProvider>
-        </TestApiProvider>
-      </MemoryRouter>
-    ),
+        </TestApiProvider>,
+      ),
   ],
 };