department-of-veterans-affairs/vets-website

import React from 'react';
import { render } from '@testing-library/react';

export function renderHook(renderCallback, options = {}) {
  const { initialProps, ...renderOptions } = options;
  const result = React.createRef();

  function TestComponent({ renderCallbackProps }) {
    const pendingResult = renderCallback(renderCallbackProps);

    React.useEffect(() => {
      result.current = pendingResult;
    });

    return null;
  }

  const { rerender: baseRerender, unmount } = render(
    <TestComponent renderCallbackProps={initialProps} />,
    renderOptions,
  );

  function rerender(rerenderCallbackProps) {
    return baseRerender(
      <TestComponent renderCallbackProps={rerenderCallbackProps} />,
    );
  }

  return { result, rerender, unmount };
}

export const getVaButtonByText = (text, view) => {
  return view.container.querySelector(`va-button[text="${text}"]`);
};

export const getVaLinkByText = (text, view) => {
  return view.container.querySelector(`va-link[text="${text}"]`);
};