test: mock diagnosticProvider

amplitude · Oct 12, 2023 · 860bde2 · 860bde2
1 parent 3ad286e
commit 860bde2
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/packages/analytics-browser/test/helpers/mock.ts b/packages/analytics-browser/test/helpers/mock.ts
@@ -1,6 +1,5 @@
 import { Logger, MemoryStorage, UUID } from '@amplitude/analytics-core';
 import { BrowserClient, BrowserConfig, LogLevel, UserSession } from '@amplitude/analytics-types';
-import { BrowserDiagnostic } from '../../src/diagnostics/diagnostic';
 
 export const createAmplitudeMock = (): jest.MockedObject<BrowserClient> => ({
   init: jest.fn(),
@@ -79,7 +78,12 @@ export const createConfigurationMock = (options?: Partial<BrowserConfig>) => {
     },
     trackingSessionEvents: false,
     userId: undefined,
-    diagnosticProvider: new BrowserDiagnostic(),
+    diagnosticProvider: {
+      isDisabled: false,
+      serverUrl: undefined,
+      apiKey: undefined,
+      track: jest.fn(),
+    },
     ...options,
   };
 };
diff --git a/packages/analytics-core/test/config.test.ts b/packages/analytics-core/test/config.test.ts
@@ -17,6 +17,7 @@ describe('config', () => {
       apiKey: API_KEY,
       storageProvider: defaultConfig.storageProvider,
       transportProvider: defaultConfig.transportProvider,
+      diagnosticProvider: defaultConfig.diagnosticProvider,
     });
     expect(config).toEqual({
       apiKey: 'apiKey',

diff --git a/packages/analytics-core/test/helpers/default.ts b/packages/analytics-core/test/helpers/default.ts
@@ -2,8 +2,6 @@ import { Config } from '@amplitude/analytics-types';
 import { getDefaultConfig } from '../../src/config';
 
 export const useDefaultConfig = (): Config => {
-  const defaultConfig = getDefaultConfig();
-  defaultConfig.diagnosticProvider.apiKey = API_KEY;
   return {
     apiKey: API_KEY,
     transportProvider: {
@@ -17,7 +15,13 @@ export const useDefaultConfig = (): Config => {
       reset: async () => undefined,
       getRaw: async () => undefined,
     },
-    ...defaultConfig,
+    ...getDefaultConfig(),
+    diagnosticProvider: {
+      isDisabled: false,
+      serverUrl: undefined,
+      apiKey: undefined,
+      track: jest.fn(),
+    },
   };
 };