Returns unmount function on testHook

andre-matulionis-ifood · Andrewmat · commit 9f286058da77 · 2019-02-09T21:17:24.000-02:00
diff --git a/src/index.js b/src/index.js
@@ -67,7 +67,8 @@ function TestHook({callback}) {
 }
 
 function testHook(callback) {
-  render(<TestHook callback={callback} />)
+  const { unmount } = render(<TestHook callback={callback} />)
+  return unmount
 }
 
 function cleanup() {

Original file line number	Diff line number	Diff line change
`@@ -67,7 +67,8 @@ function TestHook({callback}) {`
`67`	`67`	`}`
`68`	`68`
`69`	`69`	`function testHook(callback) {`
`70`		`- render(<TestHook callback={callback} />)`
	`70`	`+ const { unmount } = render(<TestHook callback={callback} />)`
	`71`	`+ return unmount`
`71`	`72`	`}`
`72`	`73`
`73`	`74`	`function cleanup() {`