Recover test-coverage in lib/comsumers-manager

lib/__mocks__/aws-sdk.js

@@ -1,40 +1,20 @@
 'use strict';
 
-const createTable = jest.fn(() => {
-  return { promise: () => Promise.resolve({}) };
-});
-
-const deleteMock = jest.fn(() => {
-  return { promise: () => Promise.resolve({}) };
-});
-
-const describeTable = jest.fn(() => {
-  return { promise: () => Promise.resolve({}) };
-});
-
-const get = jest.fn(() => {
-  return { promise: () => Promise.resolve({}) };
-});
-
-const listTagsOfResource = jest.fn(() => {
-  return { promise: () => Promise.resolve({}) };
-});
-
-const put = jest.fn(() => {
-  return { promise: () => Promise.resolve({}) };
-});
-
-const tagResource = jest.fn(() => {
-  return { promise: () => Promise.resolve({}) };
-});
-
-const update = jest.fn(() => {
-  return { promise: () => Promise.resolve({}) };
-});
+function createResponseMock() {
+  return jest.fn(() => {
+    return { promise: () => Promise.resolve({}) };
+  });
+}
 
-const waitFor = jest.fn(() => {
-  return { promise: () => Promise.resolve({}) };
-});
+const createTable = createResponseMock();
+const deleteMock = createResponseMock();
+const describeTable = createResponseMock();
+const get = createResponseMock();
+const listTagsOfResource = createResponseMock();
+const put = createResponseMock();
+const tagResource = createResponseMock();
+const update = createResponseMock();
+const waitFor = createResponseMock();
 
 const DynamoDB = jest.fn(() => ({
   createTable,

lib/consumers-manager.test.js

@@ -149,4 +149,11 @@ describe('lib/consumers-manager', () => {
       ['Stopping the consumer for "foo"…']
     ]);
   });
+
+  test('the manager should stop all consumers when asked to stop', async () => {
+    manager = new ConsumersManager({ logger, stateStore });
+    await manager.reconcile();
+    manager.stop();
+    expect(PollingConsumer.getMocks().stop).toHaveBeenCalled();
+  });
 });

lib/index.test.js

@@ -303,7 +303,8 @@ describe('lib/index', () => {
     await stateStore.registerEnhancedConsumer('bar-2', 'arn:bar-2');
     stateStore.registerEnhancedConsumer.mockClear();
 
-    const kinesis = new Kinesis({ ...options, useEnhancedFanOut: true });
+    const debug = jest.fn();
+    const kinesis = new Kinesis({ ...options, logger: { debug }, useEnhancedFanOut: true });
     await expect(kinesis.startConsumer()).resolves.toBeUndefined();
     expect(stream.registerEnhancedConsumer.mock.calls).toEqual([
       [expect.objectContaining({ consumerName: 'lifion-kinesis-0001' })],
@@ -327,6 +328,11 @@ describe('lib/index', () => {
     };
     expect(await stream.getEnhancedConsumers()).toEqual(enhancedConsumers);
     expect(await stateStore.getEnhancedConsumers()).toEqual(enhancedConsumers);
+    expect(debug.mock.calls).toEqual([
+      ['Trying to start the consumer…'],
+      ['Cleaning up enhanced consumers for "test-stream"…'],
+      ['The consumer is now ready.']
+    ]);
     kinesis.stopConsumer();
   });