fix mock

MetaMask · Sep 23, 2024 · 07b9037 · 07b9037
1 parent a42b3ad
commit 07b9037
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 21 deletions.
diff --git a/app/core/Analytics/MetaMetricsPrivacySegmentPlugin.test.ts b/app/core/Analytics/MetaMetricsPrivacySegmentPlugin.test.ts
@@ -1,21 +1,3 @@
-jest.mock('@segment/analytics-react-native', () => ({
-  PluginType: {
-    enrichment: 'enrichment',
-    utility: 'utility',
-  },
-  EventType: {
-    TrackEvent: 'track',
-    IdentifyEvent: 'identify',
-  },
-  Plugin: class Plugin {
-    type = 'utility';
-    analytics: SegmentClient | undefined;
-    configure(analytics: SegmentClient) {
-      this.analytics = analytics;
-    }
-  },
-}));
-
 import MetaMetricsPrivacySegmentPlugin from './MetaMetricsPrivacySegmentPlugin';
 import METAMETRICS_ANONYMOUS_ID from './MetaMetrics.constants';
 

diff --git a/app/util/test/testSetup.js b/app/util/test/testSetup.js
@@ -5,6 +5,7 @@ import mockClipboard from '@react-native-clipboard/clipboard/jest/clipboard-mock
 import { mockTheme } from '../theme';
 import Adapter from 'enzyme-adapter-react-16';
 import Enzyme from 'enzyme';
+import { SegmentClient } from '@segment/analytics-react-native';
 
 Enzyme.configure({ adapter: new Adapter() });
 
@@ -283,13 +284,34 @@ const initializeMockClient = () => {
     group: jest.fn(),
     alias: jest.fn(),
     reset: jest.fn(),
+    add: jest.fn(),
   };
   return global.segmentMockClient;
 };
 
-jest.mock('@segment/analytics-react-native', () => ({
-  createClient: jest.fn(() => initializeMockClient()),
-}));
+jest.mock('@segment/analytics-react-native', () => {
+  class Plugin {
+    type = 'utility';
+    analytics = undefined;
+
+    configure(analytics) {
+      this.analytics = analytics;
+    }
+  }
+
+  return {
+    createClient: jest.fn(() => initializeMockClient()),
+    PluginType: {
+      enrichment: 'enrichment',
+      utility: 'utility',
+    },
+    EventType: {
+      TrackEvent: 'track',
+      IdentifyEvent: 'identify',
+    },
+    Plugin,
+  };
+});
 
 jest.mock('@notifee/react-native', () =>
   require('@notifee/react-native/jest-mock'),