Bug-13143-001 Added tests for function exports

packages/jest-mock/src/__tests__/__fixtures__/class-mocks-types.ts

@@ -52,3 +52,16 @@ export default class SuperTestClass {
 }
 
 export class TestClass extends SuperTestClass {}
+
+export function testFunction1() {
+  return 'testFunction1';
+}
+
+function testFunction() {
+  return 'testFunction2';
+}
+export const testFunction2 = testFunction;
+
+export const testFunction3 = () => {
+  return 'testFunction3';
+};

packages/jest-mock/src/__tests__/__fixtures__/index.ts

@@ -0,0 +1,5 @@
+export const f1 = () => {
+  console.log("Im f1 calling f2");
+  f2();
+};
+export const f2 = () => console.log("Im f2");

packages/jest-mock/src/__tests__/class-mocks.test.ts

@@ -5,6 +5,37 @@
  * LICENSE file in the root directory of this source tree.
  *
  */
+import * as testTypes from './__fixtures__/class-mocks-types';
+import * as m from './__fixtures__/index';
+
+describe('Testing the mocking of exported functions', () => {
+  test('adds 1 + 2 to equal 3', () => {
+    const spy = jest.spyOn(m, 'f2').mockImplementationOnce(jest.fn());
+    m.f2();
+    expect(spy).toHaveBeenCalled();
+  });
+
+  it('can mock a directly exported function', () => {
+    jest.spyOn(testTypes, 'testFunction1').mockImplementation(() => {
+      return 'mockTestFunction';
+    });
+    expect(testTypes.testFunction1()).toBe('mockTestFunction');
+  });
+
+  it('can mock an indirectly exported function', () => {
+    jest.spyOn(testTypes, 'testFunction2').mockImplementation(() => {
+      return 'mockTestFunction';
+    });
+    expect(testTypes.testFunction2()).toBe('mockTestFunction');
+  });
+
+  it('can mock an indirectly exported anonymous function', () => {
+    jest.spyOn(testTypes, 'testFunction3').mockImplementation(() => {
+      return 'mockTestFunction';
+    });
+    expect(testTypes.testFunction3()).toBe('mockTestFunction');
+  });
+});
 
 describe('Testing the mocking of a class', () => {
   it('can call an instance method', () => {

packages/jest-mock/src/__tests__/index2.test.ts

@@ -0,0 +1,7 @@
+import * as m from './__fixtures__/index';
+
+test('adds 1 + 2 to equal 3', () => {
+  const spy = jest.spyOn(m, 'f2').mockImplementationOnce(jest.fn());
+  m.f2();
+  expect(spy).toHaveBeenCalled();
+});