From c0d61279af21ff72d5da138aed829a6916e3f743 Mon Sep 17 00:00:00 2001
From: Karan Shah <karan.shah@simplybusiness.co.uk>
Date: Fri, 4 Feb 2022 10:50:50 +0000
Subject: [PATCH] Fix extensions.test.tsx

Signed-off-by: Karan Shah <karan.shah@simplybusiness.co.uk>
---
 plugins/airbrake/dev/index.tsx            |  2 +-
 plugins/airbrake/src/api/mock/mock-api.ts |  8 +++++++-
 plugins/airbrake/src/extensions.test.tsx  | 18 +++++++++---------
 3 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/plugins/airbrake/dev/index.tsx b/plugins/airbrake/dev/index.tsx
index 7f32d30b4a..8fc43a73a1 100644
--- a/plugins/airbrake/dev/index.tsx
+++ b/plugins/airbrake/dev/index.tsx
@@ -37,7 +37,7 @@ createDevApp()
       <Page themeId="tool">
         <Header title="Airbrake demo application" subtitle="Mock API" />
         <Content>
-          <TestApiProvider apis={[[airbrakeApiRef, new MockAirbrakeApi()]]}>
+          <TestApiProvider apis={[[airbrakeApiRef, new MockAirbrakeApi(800)]]}>
             <EntityProvider entity={createEntity(1234)}>
               <EntityAirbrakeContent />
             </EntityProvider>
diff --git a/plugins/airbrake/src/api/mock/mock-api.ts b/plugins/airbrake/src/api/mock/mock-api.ts
index df0293da7c..4acfa51778 100644
--- a/plugins/airbrake/src/api/mock/mock-api.ts
+++ b/plugins/airbrake/src/api/mock/mock-api.ts
@@ -19,9 +19,15 @@ import { AirbrakeApi } from '../airbrake-api';
 import mockGroupsData from './airbrake-groups-api-mock.json';
 
 export class MockAirbrakeApi implements AirbrakeApi {
+  waitTimeInMillis: number;
+
+  constructor(waitTimeInMillis = 10) {
+    this.waitTimeInMillis = waitTimeInMillis;
+  }
+
   fetchGroups(): Promise<Groups> {
     return new Promise(resolve => {
-      setTimeout(() => resolve(mockGroupsData), 800);
+      setTimeout(() => resolve(mockGroupsData), this.waitTimeInMillis);
     });
   }
 }
diff --git a/plugins/airbrake/src/extensions.test.tsx b/plugins/airbrake/src/extensions.test.tsx
index 5532621848..3b2d1927d9 100644
--- a/plugins/airbrake/src/extensions.test.tsx
+++ b/plugins/airbrake/src/extensions.test.tsx
@@ -15,23 +15,23 @@
  */
 import React from 'react';
 import { EntityAirbrakeContent } from './extensions';
+import { Route } from 'react-router';
 import { renderInTestApp, TestApiProvider } from '@backstage/test-utils';
 import { airbrakeApiRef, MockAirbrakeApi } from './api';
-import { rootRouteRef } from './routes';
+import { createEntity } from './api/mock/mock-entity';
+import { EntityProvider } from '@backstage/plugin-catalog-react';
 
 describe('The Airbrake entity', () => {
   it('should render the content properly', async () => {
     const rendered = await renderInTestApp(
       <TestApiProvider apis={[[airbrakeApiRef, new MockAirbrakeApi()]]}>
-        <EntityAirbrakeContent />
+        <EntityProvider entity={createEntity(123)}>
+          <Route path="/" element={<EntityAirbrakeContent />} />
+        </EntityProvider>
       </TestApiProvider>,
-      {
-        mountedRoutes: {
-          '/': rootRouteRef,
-        },
-        routeEntries: ['/'],
-      },
     );
-    expect(rendered.getByText('ChunkLoadError')).toBeInTheDocument();
+    await expect(
+      rendered.findByText('ChunkLoadError'),
+    ).resolves.toBeInTheDocument();
   });
 });