From a58b25198e093a55fcb103b6b84c654e49ae8a1a Mon Sep 17 00:00:00 2001
From: John Philip <jphilip@spotify.com>
Date: Mon, 17 Feb 2025 02:37:31 -0500
Subject: [PATCH] update tests

Signed-off-by: John Philip <jphilip@spotify.com>
---
 .../TechDocsReaderPage.test.tsx                  | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/plugins/techdocs/src/reader/components/TechDocsReaderPage/TechDocsReaderPage.test.tsx b/plugins/techdocs/src/reader/components/TechDocsReaderPage/TechDocsReaderPage.test.tsx
index afb57e22ba..93b5fdc880 100644
--- a/plugins/techdocs/src/reader/components/TechDocsReaderPage/TechDocsReaderPage.test.tsx
+++ b/plugins/techdocs/src/reader/components/TechDocsReaderPage/TechDocsReaderPage.test.tsx
@@ -17,7 +17,10 @@
 import React from 'react';
 import { scmIntegrationsApiRef } from '@backstage/integration-react';
 
-import { entityRouteRef } from '@backstage/plugin-catalog-react';
+import {
+  entityPresentationApiRef,
+  entityRouteRef,
+} from '@backstage/plugin-catalog-react';
 import {
   mockApis,
   renderInTestApp,
@@ -81,6 +84,16 @@ const techdocsStorageApiMock: jest.Mocked<typeof techdocsStorageApiRef.T> = {
   syncEntityDocs: jest.fn(),
 };
 
+const entityPresentationApiMock: jest.Mocked<
+  typeof entityPresentationApiRef.T
+> = {
+  forEntity: jest.fn().mockReturnValue({
+    snapshot: {
+      primaryTitle: 'Test Entity',
+    },
+  }),
+};
+
 const fetchApiMock = {
   fetch: jest.fn().mockResolvedValue({
     ok: true,
@@ -115,6 +128,7 @@ const Wrapper = ({ children }: { children: React.ReactNode }) => {
         [configApiRef, configApi],
         [techdocsApiRef, techdocsApiMock],
         [techdocsStorageApiRef, techdocsStorageApiMock],
+        [entityPresentationApiRef, entityPresentationApiMock],
       ]}
     >
       {children}