feat: unmount

RaoHai · Mar 18, 2019 · 6df6d74 · 6df6d74
1 parent 64028dc
commit 6df6d74
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/test-hooks.tsx b/src/test-hooks.tsx
@@ -1,5 +1,5 @@
 import React from 'react';
-import { render, act } from 'react-testing-library'
+import { render, act, cleanup } from 'react-testing-library'
 
 export default function testHooks<T>(wrapper: () => T) {
   let hooks: T;
@@ -20,6 +20,9 @@ export default function testHooks<T>(wrapper: () => T) {
     },
     hooks() {
       return hooks;
+    },
+    unmount() {
+      return cleanup();
     }
   }
 }
diff --git a/test/test-hooks.test.ts b/test/test-hooks.test.ts
@@ -3,11 +3,12 @@ import { useState } from 'react';
 
 describe('test-hooks', () => {
   it('basic usage', () => {
-    const { act, hooks } = testHooks(() => useState(true));
+    const { act, hooks, unmount } = testHooks(() => useState(true));
     expect(hooks()[0]).toEqual(true);
     act(hook => {
       hook[1](false);
     });
     expect(hooks()[0]).toEqual(false);
+    unmount();
   });
 });