3box-react-hooks/test/testHelper.js

React hooks wrapping the 3box API

import React from 'react';
import { mount } from 'enzyme';

export const testHook = (hook, ...args) => {
  return testHookN(hook, null, ...args);
};

export const testHookN = (hook, n, ...args) => {
  const Component = ({ resolve }) => {
    let result = hook(...args);
    if (n != null) {
      result = result[n];
    }

    if (result != null) {
      resolve(result);
    }

    return null;
  }

  return asyncHookPromise(Component);
};

export const asyncHookPromise = Component => {
  return new Promise(resolve => {
    mount(<Component resolve={resolve} />);
  });
};