From 8dcf469c14aae508485bcc31ad413f13a619aca3 Mon Sep 17 00:00:00 2001
From: Patrik Oldsberg <poldsberg@gmail.com>
Date: Wed, 2 Mar 2022 19:15:02 +0100
Subject: [PATCH] home: fixed starred entities initialization in stories

Signed-off-by: Patrik Oldsberg <poldsberg@gmail.com>
---
 .../StarredEntities.stories.tsx               | 24 ++++++-------------
 .../src/templates/DefaultTemplate.stories.tsx | 18 ++++++--------
 2 files changed, 14 insertions(+), 28 deletions(-)

diff --git a/plugins/home/src/homePageComponents/StarredEntities/StarredEntities.stories.tsx b/plugins/home/src/homePageComponents/StarredEntities/StarredEntities.stories.tsx
index 315c0d0d16..1b01820482 100644
--- a/plugins/home/src/homePageComponents/StarredEntities/StarredEntities.stories.tsx
+++ b/plugins/home/src/homePageComponents/StarredEntities/StarredEntities.stories.tsx
@@ -15,11 +15,7 @@
  */
 
 import { HomePageStarredEntities } from '../../plugin';
-import {
-  wrapInTestApp,
-  TestApiProvider,
-  MockStorageApi,
-} from '@backstage/test-utils';
+import { wrapInTestApp, TestApiProvider } from '@backstage/test-utils';
 import {
   starredEntitiesApiRef,
   MockStarredEntitiesApi,
@@ -28,24 +24,18 @@ import {
 import { Grid } from '@material-ui/core';
 import React, { ComponentType } from 'react';
 
-const mockStorageApi = MockStorageApi.create();
-mockStorageApi
-  .forBucket('starredEntities')
-  .set('entityRefs', [
-    'component:default/example-starred-entity',
-    'component:default/example-starred-entity-2',
-    'component:default/example-starred-entity-3',
-    'component:default/example-starred-entity-4',
-  ]);
+const starredEntitiesApi = new MockStarredEntitiesApi();
+starredEntitiesApi.toggleStarred('component:default/example-starred-entity');
+starredEntitiesApi.toggleStarred('component:default/example-starred-entity-2');
+starredEntitiesApi.toggleStarred('component:default/example-starred-entity-3');
+starredEntitiesApi.toggleStarred('component:default/example-starred-entity-4');
 
 export default {
   title: 'Plugins/Home/Components/StarredEntities',
   decorators: [
     (Story: ComponentType<{}>) =>
       wrapInTestApp(
-        <TestApiProvider
-          apis={[[starredEntitiesApiRef, new MockStarredEntitiesApi()]]}
-        >
+        <TestApiProvider apis={[[starredEntitiesApiRef, starredEntitiesApi]]}>
           <Story />
         </TestApiProvider>,
         {
diff --git a/plugins/home/src/templates/DefaultTemplate.stories.tsx b/plugins/home/src/templates/DefaultTemplate.stories.tsx
index 5dc4ff61a0..c726294ba6 100644
--- a/plugins/home/src/templates/DefaultTemplate.stories.tsx
+++ b/plugins/home/src/templates/DefaultTemplate.stories.tsx
@@ -21,7 +21,7 @@ import {
   HomePageCompanyLogo,
   HomePageStarredEntities,
 } from '../plugin';
-import { wrapInTestApp, TestApiProvider, MockStorageApi} from '@backstage/test-utils';
+import { wrapInTestApp, TestApiProvider} from '@backstage/test-utils';
 import { Content, Page, InfoCard } from '@backstage/core-components';
 import {
   starredEntitiesApiRef,
@@ -37,15 +37,11 @@ import {
 import { Grid, makeStyles } from '@material-ui/core';
 import React, { ComponentType } from 'react';
 
-const mockStorageApi = MockStorageApi.create();
-mockStorageApi
-  .forBucket('starredEntities')
-  .set('entityRefs', [
-    'component:default/example-starred-entity',
-    'component:default/example-starred-entity-2',
-    'component:default/example-starred-entity-3',
-    'component:default/example-starred-entity-4'
-  ]);
+const starredEntitiesApi = new MockStarredEntitiesApi();
+starredEntitiesApi.toggleStarred('component:default/example-starred-entity');
+starredEntitiesApi.toggleStarred('component:default/example-starred-entity-2');
+starredEntitiesApi.toggleStarred('component:default/example-starred-entity-3');
+starredEntitiesApi.toggleStarred('component:default/example-starred-entity-4');
 
 export default {
   title: 'Plugins/Home/Templates',
@@ -57,7 +53,7 @@ export default {
             apis={[
               [
                 starredEntitiesApiRef,
-                new MockStarredEntitiesApi(),
+                starredEntitiesApi,
               ],
               [searchApiRef, { query: () => Promise.resolve({ results: [] }) }],
             ]}