get back some test coverage

crccheck · Jan 25, 2017 · 7c5de4b · 7c5de4b
1 parent 69aa9bc
commit 7c5de4b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -73,7 +73,7 @@ module.exports._getShardIterator = getShardIterator
 module.exports._readShard = readShard
 
 module.exports.main = function (streamName, getShardIteratorOptions) {
-  getShardId(streamName)
+  return getShardId(streamName)
   .then((shardIds) => {
     const shardIterators = shardIds.map((shardId) =>
       getShardIterator(streamName, shardId, getShardIteratorOptions))

diff --git a/test/index.spec.js b/test/index.spec.js
@@ -149,4 +149,15 @@ describe('main', () => {
       clock.restore()
     })
   })
+
+  describe('main', () => {
+    it('logs when there is an error', () => {
+      AWS.Kinesis.prototype.describeStream = AWSPromise.reject('lol error')
+      const main = proxyquire('../index', {'aws-sdk': AWS})
+      return main.main('stream name', {})
+        .then(() => {
+          assert.equal(console.log.args[0][0], 'lol error')
+        })
+    })
+  })
 })