fix: change the way mocks work

Author: Brandon Carroll

src/preset/mock-components.js renamed to src/preset/mock-component.js

@@ -1,12 +1,9 @@
-'use strict';
-
 import React from 'react';
 
-import { getConfig } from '../lib';
+import { mockNativeMethods } from './mock-native-methods';
 
 function mockComponent(component, path = component) {
   const RealComponent = jest.requireActual(path);
-  const { mockNativeMethods } = jest.requireActual('./mock-native-methods');
 
   const displayName =
     RealComponent.displayName ||
@@ -49,18 +46,4 @@ function mockComponent(component, path = component) {
   return Component;
 }
 
-getConfig('coreComponents').forEach(component => {
-  try {
-    jest.doMock(component, () => mockComponent(component));
-  } catch (e) {
-    //
-  }
-});
-
-jest.doMock('Picker', () => {
-  const Picker = mockComponent('Picker');
-  Picker.Item = ({ children, ...props }) => React.createElement('Picker.Item', props, children);
-  return Picker;
-});
-
 export { mockComponent };

src/preset/mock-modules.js

@@ -0,0 +1,45 @@
+import { getConfig } from '../lib';
+import { mockComponent } from './mock-component';
+import { mockNativeMethods } from './mock-native-methods';
+
+// Un-mock the react-native components so we can do it ourselves
+getConfig('coreComponents').forEach(component => {
+  try {
+    // try to un-mock the component
+    jest.unmock(component);
+  } catch (e) {
+    // do nothing if we can't
+  }
+});
+
+// Un-mock ReactNative so we can hide annoying `console.warn`s
+jest.unmock('ReactNative');
+
+// Mock the components we want mocked
+getConfig('coreComponents').forEach(component => {
+  try {
+    jest.doMock(component, () => mockComponent(component));
+  } catch (e) {
+    //
+  }
+});
+
+// Mock the Picker one-off because it's kinda weird
+jest.doMock('Picker', () => {
+  const React = jest.requireActual('react');
+  const Picker = mockComponent('Picker');
+  Picker.Item = ({ children, ...props }) => React.createElement('Picker.Item', props, children);
+  return Picker;
+});
+
+// Re-mock ReactNative with native methods mocked
+jest.mock('NativeAnimatedHelper').mock('ReactNative', () => {
+  const ReactNative = jest.requireActual('ReactNative');
+  const NativeMethodsMixin =
+    ReactNative.__SECRET_INTERNALS_DO_NOT_USE_OR_YOU_WILL_BE_FIRED.NativeMethodsMixin;
+
+  Object.assign(NativeMethodsMixin, mockNativeMethods);
+  Object.assign(ReactNative.NativeComponent.prototype, mockNativeMethods);
+
+  return ReactNative;
+});

src/preset/mock-native-methods.js

@@ -1,5 +1,3 @@
-'use strict';
-
 const mockNativeMethods = {
   measure: jest.fn(),
   measureInWindow: jest.fn(),

src/preset/setup.js

@@ -2,7 +2,4 @@
 import './configure';
 
 // Un-mock the things that we'll be mocking
-import './unmock';
-
-// Mock lean core components
-import './mock-components';
+import './mock-modules';