@kiwicom/smart-faq/src/services/TestRenderer.js

// @flow strict

import * as React from 'react';
import { mount } from 'enzyme';
import { QueryRenderer, graphql } from 'react-relay';
import { createMockEnvironment, MockPayloadGenerator } from 'relay-test-utils';

const getTestRenderer = (renderWithData: () => React.Node) => {
  const environment = createMockEnvironment();
  const TestRenderer = ({ renderWithData }) => (
    <QueryRenderer
      environment={environment}
      query={graphql`
        query TestRendererQuery @relay_test_operation {
          ignoreThis: FAQArticle(id: "test-id") {
            ...ArticleContent_article
          }
        }
      `}
      variables={{}}
      render={renderWithData}
    />
  );
  const wrapper = mount(<TestRenderer renderWithData={renderWithData} />);
  environment.mock.resolveMostRecentOperation(operation =>
    MockPayloadGenerator.generate(operation),
  );

  return wrapper;
};

export default getTestRenderer;