const jestExtensions = require('jest-extensions')
jestExtensions.extend(expect)
 
test('success', () => {
  const mock = jest.fn()
  mock('arg1', 'arg2')
  mock('arg3', 'arg4')
  expect(mock).toBeCalledTimes(2)
  expcet(mock).toBeCalledNthWith(0, 'arg1', 'arg2')
  expcet(mock).toBeCalledNthWith(1, 'arg3', 'arg4')
})
 
test('fail', () => {
  const mock = jest.fn()
  expect(mock).toBeCalledOnce()
})

export default class MyClass {
  
  now() {
    return new Date
  }
 
  async getFromApi(date) {
    // asynchronously request external API
  }
 
  async getNowData() {
    return await this.getFromApi(this.now())
  }
 
}

import { mockModuleClass } from 'jest-extensions'
 
function mockMyClass(...unmockedMethods) {
  return mockModuleClass(__dirname + '/../MyClass.js', ...unmockedMethods)
}
 
describe('MyClass', () => {
 
  describe('getNowData', () => {
 
    it('succeeds', async () => {
      const sut = new mockMyClass('getNowData')()
 
      // All other methods are already mocked.
      // Even if mock implementations aren't defined,
      // no accidental side effect evaluations or state changes can occur.
      sut.now.mockImplementationOnce(() => new Date('2017-01-16'))
      sut.getFromApi.mockImplementation(() => ({ data: 'value' }))
 
      const result = await sut.getNowData()
 
      expect(sut.getFromApi).toBeCalledWith(new Date('2017-01-16'))
      expect(result).toEqual({ data: 'value' })
    })
 
  })
 
})