more tests

Author: mdjastrzebski

src/__tests__/cleanup.test.tsx

@@ -1,7 +1,7 @@
 import * as React from 'react';
 import { View } from 'react-native';
 
-import { cleanupAsync, render } from '../pure';
+import { cleanupAsync, render, renderAsync } from '../pure';
 
 class Test extends React.Component<{ onUnmount: () => void }> {
   componentWillUnmount() {
@@ -14,7 +14,7 @@ class Test extends React.Component<{ onUnmount: () => void }> {
   }
 }
 
-test('cleanup', async () => {
+test('cleanup after render', async () => {
   const fn = jest.fn();
 
   render(<Test onUnmount={fn} />);
@@ -24,3 +24,14 @@ test('cleanup', async () => {
   await cleanupAsync();
   expect(fn).toHaveBeenCalledTimes(2);
 });
+
+test('cleanup after renderAsync', async () => {
+  const fn = jest.fn();
+
+  await renderAsync(<Test onUnmount={fn} />);
+  await renderAsync(<Test onUnmount={fn} />);
+  expect(fn).not.toHaveBeenCalled();
+
+  await cleanupAsync();
+  expect(fn).toHaveBeenCalledTimes(2);
+});