From 0606d5bf783ee869a25883346ce079970d584193 Mon Sep 17 00:00:00 2001
From: Min Kim <minkimcello@gmail.com>
Date: Mon, 20 Mar 2023 13:15:15 -0400
Subject: [PATCH] Add router tests for the new scaffolder context menu

Signed-off-by: Min Kim <minkimcello@gmail.com>
---
 .../TemplateListPage.test.tsx                 | 50 +++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/plugins/scaffolder/src/next/TemplateListPage/TemplateListPage.test.tsx b/plugins/scaffolder/src/next/TemplateListPage/TemplateListPage.test.tsx
index d5a76d2a47..ee36d876cf 100644
--- a/plugins/scaffolder/src/next/TemplateListPage/TemplateListPage.test.tsx
+++ b/plugins/scaffolder/src/next/TemplateListPage/TemplateListPage.test.tsx
@@ -135,4 +135,54 @@ describe('TemplateListPage', () => {
 
     expect(getByText('Tags')).toBeInTheDocument();
   });
+
+  describe('scaffolder page context menu', () => {
+    it('should render if context menu props are not set to false', async () => {
+      const { queryByTestId } = await renderInTestApp(
+        <TestApiProvider
+          apis={[
+            [catalogApiRef, mockCatalogApi],
+            [
+              starredEntitiesApiRef,
+              new DefaultStarredEntitiesApi({
+                storageApi: MockStorageApi.create(),
+              }),
+            ],
+            [permissionApiRef, {}],
+          ]}
+        >
+          <TemplateListPage />
+        </TestApiProvider>,
+        { mountedRoutes: { '/': rootRouteRef } },
+      );
+      expect(queryByTestId('menu-button')).toBeInTheDocument();
+    });
+
+    it('should not render if context menu props are set to false', async () => {
+      const { queryByTestId } = await renderInTestApp(
+        <TestApiProvider
+          apis={[
+            [catalogApiRef, mockCatalogApi],
+            [
+              starredEntitiesApiRef,
+              new DefaultStarredEntitiesApi({
+                storageApi: MockStorageApi.create(),
+              }),
+            ],
+            [permissionApiRef, {}],
+          ]}
+        >
+          <TemplateListPage
+            contextMenu={{
+              editor: false,
+              actions: false,
+              tasks: false,
+            }}
+          />
+        </TestApiProvider>,
+        { mountedRoutes: { '/': rootRouteRef } },
+      );
+      expect(queryByTestId('menu-button')).not.toBeInTheDocument();
+    });
+  });
 });