From aa69677928adc0467a44b8f47589fec00e898b11 Mon Sep 17 00:00:00 2001
From: blam <ben@blam.sh>
Date: Tue, 1 Mar 2022 14:03:08 +0100
Subject: [PATCH] chore: fixing tests

Signed-off-by: blam <ben@blam.sh>
---
 .../AllureReportComponent.test.tsx                    | 11 ++++++++++-
 plugins/todo/src/plugin.test.tsx                      | 11 ++++++++++-
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/plugins/allure/src/components/AllureReportComponent/AllureReportComponent.test.tsx b/plugins/allure/src/components/AllureReportComponent/AllureReportComponent.test.tsx
index 34d12aa032..e6355d861f 100644
--- a/plugins/allure/src/components/AllureReportComponent/AllureReportComponent.test.tsx
+++ b/plugins/allure/src/components/AllureReportComponent/AllureReportComponent.test.tsx
@@ -23,6 +23,7 @@ import {
   setupRequestMockHandlers,
   renderInTestApp,
 } from '@backstage/test-utils';
+import { EntityProvider } from '@backstage/plugin-catalog-react';
 
 describe('ExampleComponent', () => {
   const server = setupServer();
@@ -39,7 +40,15 @@ describe('ExampleComponent', () => {
   it('should render', async () => {
     const rendered = await renderInTestApp(
       <ThemeProvider theme={lightTheme}>
-        <AllureReportComponent />
+        <EntityProvider
+          entity={{
+            apiVersion: 'backstage.io/v1alpha1',
+            kind: 'Component',
+            metadata: { name: 'test' },
+          }}
+        >
+          <AllureReportComponent />
+        </EntityProvider>
       </ThemeProvider>,
     );
     expect(rendered.getByText('Missing Annotation')).toBeInTheDocument();
diff --git a/plugins/todo/src/plugin.test.tsx b/plugins/todo/src/plugin.test.tsx
index 57b9be291d..847ab7f65c 100644
--- a/plugins/todo/src/plugin.test.tsx
+++ b/plugins/todo/src/plugin.test.tsx
@@ -19,6 +19,7 @@ import { Route } from 'react-router';
 import { renderInTestApp, TestApiProvider } from '@backstage/test-utils';
 import { todoPlugin, EntityTodoContent } from './plugin';
 import { todoApiRef } from './api';
+import { EntityProvider } from '@backstage/plugin-catalog-react';
 
 describe('todo', () => {
   it('should export plugin', () => {
@@ -47,7 +48,15 @@ describe('todo', () => {
           ],
         ]}
       >
-        <Route path="/" element={<EntityTodoContent />} />
+        <EntityProvider
+          entity={{
+            apiVersion: 'backstage/v1alpha1 ',
+            kind: 'Component',
+            metadata: { name: 'Test TODO' },
+          }}
+        >
+          <Route path="/" element={<EntityTodoContent />} />
+        </EntityProvider>
       </TestApiProvider>,
     );