refactor: address pr feedback

Author: Belco90

src/__tests__/wait-for.js

@@ -1,3 +1,4 @@
+import {renderIntoDocument} from './helpers/test-utils'
 import {waitFor} from '../'
 
 test('waits callback to not throw an error', async () => {
@@ -34,7 +35,11 @@ test('uses generic error if there was no last error', async () => {
 })
 
 test('throws nice error if provided callback is not a function', () => {
-  expect(() => waitFor('not a function')).toThrow(
+  const {queryByTestId} = renderIntoDocument(`
+    <div data-testid="div"></div>
+  `)
+  const someElement = queryByTestId('div')
+  expect(() => waitFor(someElement)).toThrow(
     'Received `callback` arg must be a function',
   )
 })

src/wait-for.js

@@ -23,7 +23,7 @@ function waitFor(
   } = {},
 ) {
   if (typeof callback !== 'function') {
-    throw new Error('Received `callback` arg must be a function')
+    throw new TypeError('Received `callback` arg must be a function')
   }
 
   // created here so we get a nice stacktrace