From f04f38dbcd0bf873a97643d7b7e6e1832937d5d4 Mon Sep 17 00:00:00 2001
From: Nataliya Issayeva <nissayeva@users.noreply.github.com>
Date: Fri, 12 Nov 2021 16:09:08 -0500
Subject: [PATCH] Fix collator tests

Signed-off-by: Nataliya Issayeva <nissayeva@users.noreply.github.com>
---
 .../src/search/DefaultCatalogCollator.test.ts   | 17 +++++++++++++++--
 .../src/search/DefaultTechDocsCollator.test.ts  | 14 ++++++++++++++
 2 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/plugins/catalog-backend/src/search/DefaultCatalogCollator.test.ts b/plugins/catalog-backend/src/search/DefaultCatalogCollator.test.ts
index 20e141b921..49235584e1 100644
--- a/plugins/catalog-backend/src/search/DefaultCatalogCollator.test.ts
+++ b/plugins/catalog-backend/src/search/DefaultCatalogCollator.test.ts
@@ -14,7 +14,10 @@
  * limitations under the License.
  */
 
-import { PluginEndpointDiscovery } from '@backstage/backend-common';
+import {
+  PluginEndpointDiscovery,
+  TokenManager,
+} from '@backstage/backend-common';
 import { Entity } from '@backstage/catalog-model';
 import { DefaultCatalogCollator } from './DefaultCatalogCollator';
 import { setupServer } from 'msw/node';
@@ -55,6 +58,7 @@ const expectedEntities: Entity[] = [
 
 describe('DefaultCatalogCollator', () => {
   let mockDiscoveryApi: jest.Mocked<PluginEndpointDiscovery>;
+  let mockTokenManager: jest.Mocked<TokenManager>;
   let collator: DefaultCatalogCollator;
 
   beforeAll(() => {
@@ -62,7 +66,14 @@ describe('DefaultCatalogCollator', () => {
       getBaseUrl: jest.fn().mockResolvedValue('http://localhost:7000'),
       getExternalBaseUrl: jest.fn(),
     };
-    collator = new DefaultCatalogCollator({ discovery: mockDiscoveryApi });
+    mockTokenManager = {
+      getServerToken: jest.fn().mockResolvedValue({ token: '' }),
+      validateServerToken: jest.fn().mockResolvedValue(true),
+    };
+    collator = new DefaultCatalogCollator({
+      discovery: mockDiscoveryApi,
+      tokenManager: mockTokenManager,
+    });
     server.listen();
   });
   beforeEach(() => {
@@ -118,6 +129,7 @@ describe('DefaultCatalogCollator', () => {
     // Provide an alternate location template.
     collator = new DefaultCatalogCollator({
       discovery: mockDiscoveryApi,
+      tokenManager: mockTokenManager,
       locationTemplate: '/software/:name',
     });
 
@@ -131,6 +143,7 @@ describe('DefaultCatalogCollator', () => {
     // Provide an alternate location template.
     collator = DefaultCatalogCollator.fromConfig(new ConfigReader({}), {
       discovery: mockDiscoveryApi,
+      tokenManager: mockTokenManager,
       filter: {
         kind: ['Foo', 'Bar'],
       },
diff --git a/plugins/techdocs-backend/src/search/DefaultTechDocsCollator.test.ts b/plugins/techdocs-backend/src/search/DefaultTechDocsCollator.test.ts
index 21be228636..e689cd9f61 100644
--- a/plugins/techdocs-backend/src/search/DefaultTechDocsCollator.test.ts
+++ b/plugins/techdocs-backend/src/search/DefaultTechDocsCollator.test.ts
@@ -17,6 +17,7 @@
 import {
   PluginEndpointDiscovery,
   getVoidLogger,
+  TokenManager,
 } from '@backstage/backend-common';
 import { Entity } from '@backstage/catalog-model';
 import { DefaultTechDocsCollator } from './DefaultTechDocsCollator';
@@ -87,6 +88,7 @@ const expectedEntities: Entity[] = [
 
 describe('DefaultTechDocsCollator with legacyPathCasing configuration', () => {
   let mockDiscoveryApi: jest.Mocked<PluginEndpointDiscovery>;
+  let mockTokenManager: jest.Mocked<TokenManager>;
   let collator: DefaultTechDocsCollator;
 
   const worker = setupServer();
@@ -96,6 +98,10 @@ describe('DefaultTechDocsCollator with legacyPathCasing configuration', () => {
       getBaseUrl: jest.fn().mockResolvedValue('http://test-backend'),
       getExternalBaseUrl: jest.fn(),
     };
+    mockTokenManager = {
+      getServerToken: jest.fn().mockResolvedValue({ token: '' }),
+      validateServerToken: jest.fn().mockResolvedValue(true),
+    };
     const mockConfig = new ConfigReader({
       techdocs: {
         legacyUseCaseSensitiveTripletPaths: true,
@@ -103,6 +109,7 @@ describe('DefaultTechDocsCollator with legacyPathCasing configuration', () => {
     });
     collator = DefaultTechDocsCollator.fromConfig(mockConfig, {
       discovery: mockDiscoveryApi,
+      tokenManager: mockTokenManager,
       logger,
       legacyPathCasing: true,
     });
@@ -147,6 +154,7 @@ describe('DefaultTechDocsCollator with legacyPathCasing configuration', () => {
 
 describe('DefaultTechDocsCollator', () => {
   let mockDiscoveryApi: jest.Mocked<PluginEndpointDiscovery>;
+  let mockTokenManager: jest.Mocked<TokenManager>;
   let collator: DefaultTechDocsCollator;
 
   const worker = setupServer();
@@ -156,8 +164,13 @@ describe('DefaultTechDocsCollator', () => {
       getBaseUrl: jest.fn().mockResolvedValue('http://test-backend'),
       getExternalBaseUrl: jest.fn(),
     };
+    mockTokenManager = {
+      getServerToken: jest.fn().mockResolvedValue({ token: '' }),
+      validateServerToken: jest.fn().mockResolvedValue(true),
+    };
     collator = DefaultTechDocsCollator.fromConfig(new ConfigReader({}), {
       discovery: mockDiscoveryApi,
+      tokenManager: mockTokenManager,
       logger,
     });
 
@@ -195,6 +208,7 @@ describe('DefaultTechDocsCollator', () => {
     // Provide an alternate location template.
     collator = new DefaultTechDocsCollator({
       discovery: mockDiscoveryApi,
+      tokenManager: mockTokenManager,
       locationTemplate: '/software/:name',
       logger,
     });