Improve cli test

macor161 · Sep 5, 2019 · 8001694 · 8001694
1 parent cd2bcb1
commit 8001694
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/src/index.js b/src/index.js
@@ -48,4 +48,4 @@ process.on('uncaughtException', e => {
   debug('Uncaught Error: %o', e.stack)
 })
 
-main()
+module.exports = main()
diff --git a/test/cli.spec.js b/test/cli.spec.js
@@ -16,16 +16,9 @@ describe('CLI', () => {
       error: jest.fn(),
     }
 
-    require('../src/index')
+    await require('../src/index')
 
-    await wait(30000)
-
-    expect(console.log).toHaveBeenCalledWith(green.bold('Build successful'))
+    expect(console.log).toHaveBeenCalled()
     expect(console.error).not.toHaveBeenCalled()
   }, 60000)
 })
-
-
-function wait(ms) {
-  return new Promise(res => setTimeout(res, ms))
-}