cleanup.test.tsx

import * as React from 'react';
import { View } from 'react-native';

import { cleanup, render, unsafe_renderSync } from '../pure';

class Test extends React.Component<{ onUnmount: () => void }> {
  componentWillUnmount() {
    if (this.props.onUnmount) {
      this.props.onUnmount();
    }
  }
  render() {
    return <View />;
  }
}

test('cleanup after render', async () => {
  const fn = jest.fn();

  await render(<Test onUnmount={fn} />);
  await render(<Test onUnmount={fn} />);
  expect(fn).not.toHaveBeenCalled();

  await cleanup();
  expect(fn).toHaveBeenCalledTimes(2);
});

test('cleanup after unsafe_renderSync', async () => {
  const fn = jest.fn();

  unsafe_renderSync(<Test onUnmount={fn} />);
  unsafe_renderSync(<Test onUnmount={fn} />);
  expect(fn).not.toHaveBeenCalled();

  await cleanup();
  expect(fn).toHaveBeenCalledTimes(2);
});