use the catalog mock in more tests

Signed-off-by: Fredrik Adelöw <freben@gmail.com>

plugins/techdocs/src/home/components/DefaultTechDocsHome.test.tsx

@@ -21,11 +21,11 @@ import {
   storageApiRef,
 } from '@backstage/core-plugin-api';
 import {
-  CatalogApi,
   catalogApiRef,
   starredEntitiesApiRef,
   MockStarredEntitiesApi,
 } from '@backstage/plugin-catalog-react';
+import { catalogApiMock } from '@backstage/plugin-catalog-react/testUtils';
 import {
   MockStorageApi,
   renderInTestApp,
@@ -36,22 +36,18 @@ import React from 'react';
 import { rootDocsRouteRef } from '../../routes';
 import { DefaultTechDocsHome } from './DefaultTechDocsHome';
 
-const mockCatalogApi: Partial<CatalogApi> = {
-  getEntityFacets: async () => ({ facets: { 'relations.ownedBy': [] } }),
-  getEntitiesByRefs: () => Promise.resolve({ items: [] }),
-  getEntities: async () => ({
-    items: [
-      {
-        apiVersion: 'version',
-        kind: 'User',
-        metadata: {
-          name: 'owned',
-          namespace: 'default',
-        },
+const mockCatalogApi = catalogApiMock({
+  entities: [
+    {
+      apiVersion: 'version',
+      kind: 'User',
+      metadata: {
+        name: 'owned',
+        namespace: 'default',
       },
-    ],
-  }),
-};
+    },
+  ],
+});
 
 describe('TechDocs Home', () => {
   const configApi: ConfigApi = new ConfigReader({

plugins/techdocs/src/home/components/Grids/EntityListDocsGrid.test.tsx

@@ -21,12 +21,14 @@ import {
   storageApiRef,
 } from '@backstage/core-plugin-api';
 import {
-  CatalogApi,
   catalogApiRef,
   starredEntitiesApiRef,
   MockStarredEntitiesApi,
 } from '@backstage/plugin-catalog-react';
-import { MockEntityListContextProvider } from '@backstage/plugin-catalog-react/testUtils';
+import {
+  MockEntityListContextProvider,
+  catalogApiMock,
+} from '@backstage/plugin-catalog-react/testUtils';
 import {
   MockStorageApi,
   renderInTestApp,
@@ -62,12 +64,7 @@ const entities = [
   },
 ];
 
-const mockCatalogApi = {
-  getEntityByRef: () => Promise.resolve(),
-  getEntities: async () => ({
-    items: entities,
-  }),
-} as Partial<CatalogApi>;
+const mockCatalogApi = catalogApiMock({ entities });
 
 describe('Entity List Docs Grid', () => {
   beforeEach(() => {

plugins/techdocs/src/home/components/TechDocsCustomHome.test.tsx

@@ -14,7 +14,8 @@
  * limitations under the License.
  */
 
-import { CatalogApi, catalogApiRef } from '@backstage/plugin-catalog-react';
+import { catalogApiRef } from '@backstage/plugin-catalog-react';
+import { catalogApiMock } from '@backstage/plugin-catalog-react/testUtils';
 import { renderInTestApp, TestApiRegistry } from '@backstage/test-utils';
 import { screen } from '@testing-library/react';
 import React from 'react';
@@ -22,21 +23,18 @@ import { TechDocsCustomHome, PanelType } from './TechDocsCustomHome';
 import { ApiProvider } from '@backstage/core-app-api';
 import { rootDocsRouteRef } from '../../routes';
 
-const mockCatalogApi = {
-  getEntityByRef: jest.fn(),
-  getEntities: async () => ({
-    items: [
-      {
-        apiVersion: 'version',
-        kind: 'User',
-        metadata: {
-          name: 'owned',
-          namespace: 'default',
-        },
+const mockCatalogApi = catalogApiMock({
+  entities: [
+    {
+      apiVersion: 'version',
+      kind: 'User',
+      metadata: {
+        name: 'owned',
+        namespace: 'default',
       },
-    ],
-  }),
-} as Partial<CatalogApi>;
+    },
+  ],
+});
 
 describe('TechDocsCustomHome', () => {
   const apiRegistry = TestApiRegistry.from([catalogApiRef, mockCatalogApi]);