From cccfa039e53911573d02b47c5e0f45b941083d23 Mon Sep 17 00:00:00 2001
From: Patrik Oldsberg <poldsberg@gmail.com>
Date: Wed, 15 Sep 2021 00:11:19 +0200
Subject: [PATCH] catalog: fix broken test

Signed-off-by: Patrik Oldsberg <poldsberg@gmail.com>
---
 .../EntityLayout/EntityLayout.test.tsx        | 22 ++++++++-----------
 1 file changed, 9 insertions(+), 13 deletions(-)

diff --git a/plugins/catalog/src/components/EntityLayout/EntityLayout.test.tsx b/plugins/catalog/src/components/EntityLayout/EntityLayout.test.tsx
index 218e55fb5f..ad0d3bceb4 100644
--- a/plugins/catalog/src/components/EntityLayout/EntityLayout.test.tsx
+++ b/plugins/catalog/src/components/EntityLayout/EntityLayout.test.tsx
@@ -62,27 +62,23 @@ describe('EntityLayout', () => {
   });
 
   it('renders the entity title if defined', async () => {
-    const mockEntityDataWithTitle = {
-      loading: false,
-      error: undefined,
-      entity: {
-        kind: 'MyKind',
-        metadata: {
-          name: 'my-entity',
-          title: 'My Entity',
-        },
-      } as Entity,
-    };
+    const mockEntityWithTitle = {
+      kind: 'MyKind',
+      metadata: {
+        name: 'my-entity',
+        title: 'My Entity',
+      },
+    } as Entity;
 
     const rendered = await renderInTestApp(
       <ApiProvider apis={mockApis}>
-        <EntityContext.Provider value={mockEntityDataWithTitle}>
+        <EntityProvider entity={mockEntityWithTitle}>
           <EntityLayout>
             <EntityLayout.Route path="/" title="tabbed-test-title">
               <div>tabbed-test-content</div>
             </EntityLayout.Route>
           </EntityLayout>
-        </EntityContext.Provider>
+        </EntityProvider>
       </ApiProvider>,
     );