Refactor code

Signed-off-by: Bandini Bhopi <bandinib@amazon.com>

src/plugins/data_source_management/public/auth_registry/authentication_methods_registry.test.ts

@@ -3,22 +3,9 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import {
-  AuthenticationMethodRegistery,
-  AuthenticationMethod,
-} from './authentication_methods_registry';
+import { AuthenticationMethodRegistery } from './authentication_methods_registry';
 import React from 'react';
-
-export const createAuthenticationMethod = (
-  authMethod: Partial<AuthenticationMethod>
-): AuthenticationMethod => ({
-  name: 'unknown',
-  credentialForm: React.createElement('div', {}, 'Hello, world!'),
-  credentialSourceOption: {
-    value: 'unknown',
-  },
-  ...authMethod,
-});
+import { createAuthenticationMethod } from '../mocks';
 
 describe('AuthenticationMethodRegistery', () => {
   let registry: AuthenticationMethodRegistery;

src/plugins/data_source_management/public/auth_registry/index.ts

@@ -8,5 +8,3 @@ export {
   AuthenticationMethod,
   AuthenticationMethodRegistery,
 } from './authentication_methods_registry';
-
-export { createAuthenticationMethod } from './authentication_methods_registry.test';

src/plugins/data_source_management/public/mocks.ts

@@ -3,6 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
+import React from 'react';
 import { throwError } from 'rxjs';
 import { SavedObjectsClientContract } from 'opensearch-dashboards/public';
 import { AuthType } from './types';
@@ -14,6 +15,7 @@ import {
 } from './plugin';
 import { managementPluginMock } from '../../management/public/mocks';
 import { mockManagementPlugin as indexPatternManagementPluginMock } from '../../index_pattern_management/public/mocks';
+import { AuthenticationMethod } from './auth_registry';
 
 /* Mock Types */
 
@@ -225,3 +227,14 @@ export const testDataSourceManagementPlugin = (
   };
   return { setup, doStart };
 };
+
+export const createAuthenticationMethod = (
+  authMethod: Partial<AuthenticationMethod>
+): AuthenticationMethod => ({
+  name: 'unknown',
+  credentialForm: React.createElement('div', {}, 'Hello, world!'),
+  credentialSourceOption: {
+    value: 'unknown',
+  },
+  ...authMethod,
+});

src/plugins/data_source_management/public/plugin.test.ts

@@ -4,8 +4,7 @@
  */
 import { coreMock } from '../../../core/public/mocks';
 import { DataSourceManagementPluginStart } from './plugin';
-import { testDataSourceManagementPlugin } from './mocks';
-import { createAuthenticationMethod } from './auth_registry';
+import { testDataSourceManagementPlugin, createAuthenticationMethod } from './mocks';
 
 describe('#dataSourceManagement', () => {
   let coreSetup: any;