Support custom amplitude instance (#160)

rangle · Nov 21, 2017 · 6058769 · 6058769
1 parent 436bd54
commit 6058769
Show file tree

Hide file tree

Showing 4 changed files with 251 additions and 193 deletions.
diff --git a/src/targets/amplitude/__tests__/amplitude.mocks.js b/src/targets/amplitude/__tests__/amplitude.mocks.js
@@ -10,6 +10,7 @@ const instanceMock = {
   identify: jest.fn(),
   logRevenueV2: jest.fn(),
 };
+
 const identityMock = {
   add: jest.fn(),
   set: jest.fn(),
@@ -33,4 +34,19 @@ const amplitudeSDKMock = {
   logRevenueV2: jest.fn(),
 };
 
+export function resetAllMocks() {
+  [
+    instanceMock,
+    identityMock,
+    revenueMock,
+    amplitudeSDKMock,
+  ].forEach(mockObj => {
+    Object.keys(mockObj).forEach(key => {
+      if (typeof mockObj[key].mockReset === 'function') {
+        mockObj[key].mockReset();
+      }
+    });
+  });
+}
+
 export default amplitudeSDKMock;