From 80b05b914ac7e521155feb2a6f0bb124337ae76a Mon Sep 17 00:00:00 2001
From: Scotty Bollinger <scotty.bollinger@elastic.co>
Date: Tue, 23 Mar 2021 17:39:57 -0500
Subject: [PATCH] [Workplace Search] Add missing tests to get 100% coverage
 (#95240)

---
 .../workplace_search/app_logic.test.ts        | 13 +++-
 .../workplace_search/routes.test.tsx          | 60 ++++++++++++++++++-
 2 files changed, 69 insertions(+), 4 deletions(-)

diff --git a/x-pack/plugins/enterprise_search/public/applications/workplace_search/app_logic.test.ts b/x-pack/plugins/enterprise_search/public/applications/workplace_search/app_logic.test.ts
index 8ba94e83d26cf..82fc00923202f 100644
--- a/x-pack/plugins/enterprise_search/public/applications/workplace_search/app_logic.test.ts
+++ b/x-pack/plugins/enterprise_search/public/applications/workplace_search/app_logic.test.ts
@@ -48,7 +48,7 @@ describe('AppLogic', () => {
     expect(AppLogic.values).toEqual(DEFAULT_VALUES);
   });
 
-  describe('initializeAppData()', () => {
+  describe('initializeAppData', () => {
     it('sets values based on passed props', () => {
       AppLogic.actions.initializeAppData(DEFAULT_INITIAL_APP_DATA);
 
@@ -66,11 +66,20 @@ describe('AppLogic', () => {
     });
   });
 
-  describe('setContext()', () => {
+  describe('setContext', () => {
     it('sets context', () => {
       AppLogic.actions.setContext(true);
 
       expect(AppLogic.values.isOrganization).toEqual(true);
     });
   });
+
+  describe('setSourceRestriction', () => {
+    it('sets property', () => {
+      mount(DEFAULT_INITIAL_APP_DATA);
+      AppLogic.actions.setSourceRestriction(true);
+
+      expect(AppLogic.values.account.canCreatePersonalSources).toEqual(true);
+    });
+  });
 });
diff --git a/x-pack/plugins/enterprise_search/public/applications/workplace_search/routes.test.tsx b/x-pack/plugins/enterprise_search/public/applications/workplace_search/routes.test.tsx
index 68bec94270a01..7d3e19dfe626a 100644
--- a/x-pack/plugins/enterprise_search/public/applications/workplace_search/routes.test.tsx
+++ b/x-pack/plugins/enterprise_search/public/applications/workplace_search/routes.test.tsx
@@ -13,8 +13,15 @@ import { EuiLink } from '@elastic/eui';
 
 import {
   getContentSourcePath,
+  getGroupPath,
+  getGroupSourcePrioritizationPath,
+  getReindexJobRoute,
+  getRoleMappingPath,
+  getSourcesPath,
+  GROUPS_PATH,
   SOURCES_PATH,
   PERSONAL_SOURCES_PATH,
+  ROLE_MAPPINGS_PATH,
   SOURCE_DETAILS_PATH,
 } from './routes';
 
@@ -24,17 +31,66 @@ const TestComponent = ({ id, isOrg }: { id: string; isOrg?: boolean }) => {
 };
 
 describe('getContentSourcePath', () => {
-  it('should format org route', () => {
+  it('should format org path', () => {
     const wrapper = shallow(<TestComponent id="123" isOrg />);
     const path = wrapper.find(EuiLink).prop('href');
 
     expect(path).toEqual(`${SOURCES_PATH}/123`);
   });
 
-  it('should format user route', () => {
+  it('should format user path', () => {
     const wrapper = shallow(<TestComponent id="123" />);
     const path = wrapper.find(EuiLink).prop('href');
 
     expect(path).toEqual(`${PERSONAL_SOURCES_PATH}/123`);
   });
 });
+
+describe('getGroupPath', () => {
+  it('should format path', () => {
+    expect(getGroupPath('123')).toEqual(`${GROUPS_PATH}/123`);
+  });
+});
+
+describe('getRoleMappingPath', () => {
+  it('should format path', () => {
+    expect(getRoleMappingPath('123')).toEqual(`${ROLE_MAPPINGS_PATH}/123`);
+  });
+});
+
+describe('getGroupSourcePrioritizationPath', () => {
+  it('should format path', () => {
+    expect(getGroupSourcePrioritizationPath('123')).toEqual(
+      `${GROUPS_PATH}/123/source_prioritization`
+    );
+  });
+});
+
+describe('getSourcesPath', () => {
+  const PATH = '/foo/123';
+
+  it('should format org path', () => {
+    expect(getSourcesPath(PATH, true)).toEqual(PATH);
+  });
+
+  it('should format user path', () => {
+    expect(getSourcesPath(PATH, false)).toEqual(`/p${PATH}`);
+  });
+});
+
+describe('getReindexJobRoute', () => {
+  const SOURCE_ID = '234';
+  const REINDEX_ID = '345';
+
+  it('should format org path', () => {
+    expect(getReindexJobRoute(SOURCE_ID, REINDEX_ID, true)).toEqual(
+      `/sources/${SOURCE_ID}/schema_errors/${REINDEX_ID}`
+    );
+  });
+
+  it('should format user path', () => {
+    expect(getReindexJobRoute(SOURCE_ID, REINDEX_ID, false)).toEqual(
+      `/p/sources/${SOURCE_ID}/schema_errors/${REINDEX_ID}`
+    );
+  });
+});