From f7ad6f0795c7f20c53856b9027aa72881cfe3796 Mon Sep 17 00:00:00 2001
From: Karan Shah <karan.shah@simplybusiness.co.uk>
Date: Fri, 4 Feb 2022 11:59:40 +0000
Subject: [PATCH] Fix EntityAirbrakeWidget.test.tsx

Signed-off-by: Karan Shah <karan.shah@simplybusiness.co.uk>
---
 .../EntityAirbrakeWidget/EntityAirbrakeWidget.test.tsx     | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/plugins/airbrake/src/components/EntityAirbrakeWidget/EntityAirbrakeWidget.test.tsx b/plugins/airbrake/src/components/EntityAirbrakeWidget/EntityAirbrakeWidget.test.tsx
index 5fd1aa9bf5..7b3bd6595b 100644
--- a/plugins/airbrake/src/components/EntityAirbrakeWidget/EntityAirbrakeWidget.test.tsx
+++ b/plugins/airbrake/src/components/EntityAirbrakeWidget/EntityAirbrakeWidget.test.tsx
@@ -17,13 +17,16 @@
 import React from 'react';
 import { EntityAirbrakeWidget } from './EntityAirbrakeWidget';
 import exampleData from '../../api/mock/airbrake-groups-api-mock.json';
-import { renderInTestApp } from '@backstage/test-utils';
+import { renderInTestApp, TestApiProvider } from '@backstage/test-utils';
 import { createEntity } from '../../api/mock/mock-entity';
+import { airbrakeApiRef, MockAirbrakeApi } from '../../api';
 
 describe('EntityAirbrakeContent', () => {
   it('renders all errors sent from Airbrake', async () => {
     const table = await renderInTestApp(
-      <EntityAirbrakeWidget entity={createEntity(123)} />,
+      <TestApiProvider apis={[[airbrakeApiRef, new MockAirbrakeApi()]]}>
+        <EntityAirbrakeWidget entity={createEntity(123)} />
+      </TestApiProvider>,
     );
     expect(exampleData.groups.length).toBeGreaterThan(0);
     for (const group of exampleData.groups) {