sqs-consumer 6.0.0-alpha.2 → 6.0.1

package/.github/CODEOWNERS

@@ -0,0 +1,2 @@
+# All changes should be reviewed by codeowners
+* @bbc/ibl @bbc/detl-engineers

package/.github/CODE_OF_CONDUCT.md

@@ -0,0 +1,74 @@
+# Contributor Covenant Code of Conduct (CoC)
+
+## Our Pledge
+
+We as members, contributors, and leaders pledge to make participation in our community a harassment-free experience for everyone, regardless of age, body size, visible or invisible disability, ethnicity, sex characteristics, gender identity and expression, level of experience, education, socio-economic status, nationality, personal appearance, race, religion, or sexual identity and orientation.
+
+We pledge to act and interact in ways that contribute to an open, welcoming, diverse, inclusive, and healthy community.
+
+## Our Standards
+
+Examples of behaviour that contributes to a positive environment for our community include:
+
+- Respect different opinions, perspectives, and experiences
+- Giving and appreciating constructive feedback
+- Accepting responsibility and apologizing to those affected by our mistakes, and learning from the experience
+- Focusing on what is best for us as individuals and the overall community
+- Demonstrating kindness toward other people
+
+Examples of unacceptable behaviour include:
+
+- The use of sexualised language or imagery, and sexual attention or advances of any kind
+- Trolling, insulting or derogatory comments, and personal or political attacks
+- Public or private harassment
+- Publishing others’ private information, such as a physical or email address, without their explicit permission
+- Other conduct which could reasonably be considered inappropriate in a professional setting
+
+## Enforcement Responsibilities
+
+Project maintainers are responsible for clarifying and enforcing our standards of acceptable behaviour and will take appropriate and fair corrective action in response to any behaviour that they deem inappropriate, threatening, offensive, or harmful.
+
+Project maintainers have the right and responsibility to remove, edit, or reject comments, commits, code, wiki edits, issues, and other contributions that are not aligned to this Code of Conduct, and will communicate reasons for moderation decisions when appropriate.
+
+## Scope
+
+This Code of Conduct applies within all community spaces, and also applies when an individual is officially representing the community in public spaces. Examples of representing our community include using an official e-mail address, posting via an official social media account, or acting as an appointed representative at an online or offline event.
+
+## Enforcement
+
+Instances of abusive, harassing, or otherwise unacceptable behaviour may be reported to the community leaders responsible for enforcement. All complaints will be reviewed and investigated promptly and fairly.
+
+[Project maintainers](https://github.com/bbc/sqs-consumber/blob/main/.github/CODEOWNERS) are obligated to respect the privacy and security of the reporter of any incident.
+
+## Enforcement Guidelines
+
+### 1. Correction
+
+Community Impact: Use of inappropriate language or other behaviour deemed unprofessional or unwelcome in the community.
+
+Consequence: A private, written warning from community leaders, providing clarity around the nature of the violation and an explanation of why the behaviour was inappropriate. A public apology may be requested.
+
+### 2. Warning
+
+Community Impact: A violation through a single incident or series of actions.
+
+Consequence: A warning with consequences for continued behaviour. No interaction with the people involved, including unsolicited interaction with those enforcing the Code of Conduct, for a specified period of time. This includes avoiding interactions in community spaces as well as external channels like social media. Violating these terms may lead to a temporary or permanent ban.
+
+### 3. Temporary Ban
+
+Community Impact: A serious violation of community standards, including sustained inappropriate behaviour.
+
+Consequence: A temporary ban from any sort of interaction or public communication with the community for a specified period of time. No public or private interaction with the people involved, including unsolicited interaction with those enforcing the Code of Conduct, is allowed during this period. Violating these terms may lead to a permanent ban.
+
+### 4. Permanent Ban
+
+Community Impact: Demonstrating a pattern of violation of community standards, including sustained inappropriate behaviour, harassment of an individual, or aggression toward or disparagement of classes of individuals.
+
+Consequence: A permanent ban from any sort of public interaction within the community.
+
+## Attribution
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage], version 2.0, available at [https://www.contributor-covenant.org/version/2/0/code_of_conduct.html][version]
+
+[homepage]: https://contributor-covenant.org
+[version]: https://www.contributor-covenant.org/version/2/0/code_of_conduct/

package/.github/CONTRIBUTING.md

@@ -0,0 +1,41 @@
+# Contributing
+
+Thank you for your interest in contributing to the sqs-consumer.
+
+- If you're unsure if a feature would make a good addition, you can always [create an issue](https://github.com/bbc/sqs-consumer/issues/new) first. Raising an issue before creating a pull request is recommended.
+- We aim for 100% test coverage. Please write tests for any new functionality or changes.
+- Any API changes should be fully documented.
+- Make sure your code meets our linting standards. Run `npm run lint` to check your code.
+- Maintain the existing coding style. There are some settings in `.jsbeautifyrc` to help.
+- Be mindful of others when making suggestions and/or code reviewing.
+
+## Reporting Issues
+
+Before opening a new issue, first check that there is not already an [open issue or Pull Request](https://github.com/bbc/sqs-consumer/issues?utf8=%E2%9C%93&q=is%3Aopen) that addresses it.
+
+If there is, make relevant comments and add your reaction. Use a reaction in place of a "+1" comment:
+
+- 👍 - upvote
+- 👎 - downvote
+
+If you cannot find an existing issue that describes your bug or feature, create a new issue using the guidelines below.
+
+1. Pick an appropriate template for the type of issue [from here](https://github.com/bbc/sqs-consumer/issues/choose)
+2. Provide as much detail as possible
+3. Follow your issue in the issue tracking workflow
+
+## Contributing Code
+
+If you do not have push access to the repository, please [fork it](https://help.github.com/en/articles/fork-a-repo). You should then work on your own `main` branch.
+
+Otherwise, you may clone this repository and create a working branch with a _kebab-case_ name reflecting what you are working on (e.g. `fix-the-thing`).
+
+Follow the setup instructions in the [README](../README.md).
+
+Ensure all your code is thoroughly tested and that this testing is detailed in the pull request.
+
+## Pull Request Process
+
+1. Make sure you have opened an issue and it was approved by a project maintainer before working on a PR
+2. Read and complete all relevant sections of the PR template
+3. Wait for the PR get approved

package/.github/ISSUE_TEMPLATE/bug-report.md

@@ -0,0 +1,27 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: ''
+labels: bug
+assignees: ''
+---
+
+**Describe the bug**
+A clear and concise description of what the bug is.
+
+**To Reproduce**
+Steps to reproduce the behaviour:
+
+1. Go to '...'
+2. Select '....'
+3. Scroll down to '....'
+4. See error
+
+**Expected behaviour**
+A clear and concise description of what you expected to happen.
+
+**screenshots**
+If applicable, add screenshots to help explain your problem.
+
+**Additional context**
+Add any other context about the problem here, such as specific device information.

package/.github/ISSUE_TEMPLATE/feature-request.md

@@ -0,0 +1,19 @@
+---
+name: Feature request
+about: Suggest an idea for this project
+title: ''
+labels: feature-request
+assignees: ''
+---
+
+**The problem**
+A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
+
+**Suggested solution**
+A clear and concise description of what you want to happen.
+
+**Alternatives considered**
+A clear and concise description of any alternative solutions or features you've considered.
+
+**Additional context**
+Add any other context or screenshots about the feature request here.

package/.github/ISSUE_TEMPLATE/technical-question.md

@@ -0,0 +1,16 @@
+---
+name: Technical question
+about: Want to ask a technical question about the project
+title: ''
+labels: question
+assignees: ''
+---
+
+**Question**
+A clear technical question could be asked here.
+
+**screenshots**
+If applicable, add screenshots to help explain your question.
+
+**Additional context**
+Add any other context about the question here, such as specific device information, technology choice etc.

package/.github/SECURITY.md

@@ -0,0 +1,9 @@
+# Security Policy
+
+## Reporting a Vulnerability
+
+Our full security policy and vulnerability reporting procedure is documented on [this external website](https://www.bbc.com/backstage/security-disclosure-policy/#reportingavulnerability).
+
+Please note that this is a general BBC process. Communication will not be direct with the team responsible for this repo.
+
+If you would like to, you can also open an issue in this repo regarding your disclosure, but please never share any details of the vulnerability in the GitHub issue.

package/.github/pull_request_template.md

@@ -0,0 +1,29 @@
+Resolves #NUMBER
+
+**Description:**
+_A very high-level summary of easily-reproducible changes that can be understood by non-devs._
+
+**Type of change:**
+
+- [ ] Bug fix (non-breaking change which fixes an issue)
+- [ ] New feature (non-breaking change which adds functionality)
+- [ ] Breaking change (fix or feature that would cause existing functionality to change)
+
+**Why is this change required?:**
+_A simple explanation of what the problem is and how this PR solves it_
+
+**Code changes:**
+
+- _A bullet point list of key code changes that have been made._
+- _When describing code changes, try to communicate **how** and **why** you implemented something a specific way, not just **what** has changed._
+
+---
+
+**Checklist:**
+
+- [ ] My code follows the code style of this project.
+- [ ] My change requires a change to the documentation.
+- [ ] I have updated the documentation accordingly.
+- [ ] I have read the **CONTRIBUTING** document.
+- [ ] I have added tests to cover my changes.
+- [ ] All new and existing tests passed.

package/.github/workflows/coverage.yml

@@ -0,0 +1,35 @@
+name: Report Coverage
+on:
+  pull_request:
+    branches:
+      - 'main'
+  push:
+    branches:
+      - main
+permissions:
+  contents: read
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        node-version: [14.x]
+
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: Use Node.js ${{ matrix.node-version }}
+        uses: actions/setup-node@v3
+        with:
+          node-version: ${{ matrix.node-version }}
+
+      - name: Install Node Modules
+        run: npm ci
+
+      - name: Report Coverage
+        uses: paambaati/codeclimate-action@v3.2.0
+        env:
+          CC_TEST_REPORTER_ID: 2d851f8f3a9348ac4f43262305037f80a730c2660fda50af8ae4d445fd89333b
+        with:
+          coverageCommand: npm run lcov

package/.github/workflows/stale.yml

@@ -0,0 +1,20 @@
+name: 'Close stale issues and PRs'
+on:
+  schedule:
+    - cron: '30 1 * * *'
+
+jobs:
+  stale:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/stale@v6
+        with:
+          stale-issue-message: 'This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 5 days.'
+          stale-pr-message: 'This PR is stale because it has been open 45 days with no activity. Remove stale label or comment or this will be closed in 10 days.'
+          close-issue-message: 'This issue was closed because it has been stalled for 5 days with no activity.'
+          close-pr-message: 'This PR was closed because it has been stalled for 10 days with no activity.'
+          days-before-issue-stale: 30
+          days-before-pr-stale: 45
+          days-before-issue-close: 5
+          days-before-pr-close: 10
+          operations-per-run: 90

package/.github/workflows/test.yml

@@ -0,0 +1,34 @@
+name: Run Tests
+on:
+  pull_request:
+    branches:
+      - 'main'
+  push:
+    branches:
+      - main
+permissions:
+  contents: read
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        node-version: [14.x, 16.x, 18.x]
+
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: Use Node.js ${{ matrix.node-version }}
+        uses: actions/setup-node@v3
+        with:
+          node-version: ${{ matrix.node-version }}
+
+      - name: Install Node Modules
+        run: npm ci
+
+      - name: NPM Audit
+        run: npx audit-ci
+
+      - name: Run Tests and Linting
+        run: npm run test

package/README.md

@@ -71,8 +71,8 @@ const app = Consumer.create({
   sqs: new SQSClient({
     region: 'my-region',
     credentials: {
-      accessKeyId: process.env.AWS_ACCESS_KEY_ID,
-      secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY
+      accessKeyId: 'yourAccessKey',
+      secretAccessKey: 'yourSecret'
     }
   })
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sqs-consumer",
-  "version": "6.0.0-alpha.2",
+  "version": "6.0.1",
   "description": "Build SQS-based Node applications without the boilerplate",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/test/consumer.test.ts

@@ -0,0 +1,979 @@
+import {
+  ChangeMessageVisibilityBatchCommand,
+  ChangeMessageVisibilityCommand,
+  DeleteMessageBatchCommand,
+  DeleteMessageCommand,
+  ReceiveMessageCommand,
+  SQSClient
+} from '@aws-sdk/client-sqs';
+import { assert } from 'chai';
+import * as sinon from 'sinon';
+import * as pEvent from 'p-event';
+
+import { AWSError } from '../src/types';
+import { Consumer } from '../src/consumer';
+
+const sandbox = sinon.createSandbox();
+
+const AUTHENTICATION_ERROR_TIMEOUT = 20;
+const POLLING_TIMEOUT = 100;
+const QUEUE_URL = 'some-queue-url';
+const REGION = 'some-region';
+
+const mockReceiveMessage = sinon.match.instanceOf(ReceiveMessageCommand);
+const mockDeleteMessage = sinon.match.instanceOf(DeleteMessageCommand);
+const mockDeleteMessageBatch = sinon.match.instanceOf(
+  DeleteMessageBatchCommand
+);
+const mockChangeMessageVisibility = sinon.match.instanceOf(
+  ChangeMessageVisibilityCommand
+);
+const mockChangeMessageVisibilityBatch = sinon.match.instanceOf(
+  ChangeMessageVisibilityBatchCommand
+);
+
+class MockSQSError extends Error implements AWSError {
+  name: string;
+  $metadata: {
+    httpStatusCode: number;
+  };
+  $service: string;
+  $retryable: {
+    throttling: boolean;
+  };
+  $fault: 'client' | 'server';
+  time: Date;
+
+  constructor(message: string) {
+    super(message);
+    this.message = message;
+  }
+}
+
+describe('Consumer', () => {
+  let consumer;
+  let clock;
+  let handleMessage;
+  let handleMessageBatch;
+  let sqs;
+  const response = {
+    Messages: [
+      {
+        ReceiptHandle: 'receipt-handle',
+        MessageId: '123',
+        Body: 'body'
+      }
+    ]
+  };
+
+  beforeEach(() => {
+    clock = sinon.useFakeTimers();
+    handleMessage = sandbox.stub().resolves(null);
+    handleMessageBatch = sandbox.stub().resolves(null);
+    sqs = sinon.createStubInstance(SQSClient);
+    sqs.send = sinon.stub();
+
+    sqs.send.withArgs(mockReceiveMessage).resolves(response);
+    sqs.send.withArgs(mockDeleteMessage).resolves();
+    sqs.send.withArgs(mockDeleteMessageBatch).resolves();
+    sqs.send.withArgs(mockChangeMessageVisibility).resolves();
+    sqs.send.withArgs(mockChangeMessageVisibilityBatch).resolves();
+
+    consumer = new Consumer({
+      queueUrl: QUEUE_URL,
+      region: REGION,
+      handleMessage,
+      sqs,
+      authenticationErrorTimeout: AUTHENTICATION_ERROR_TIMEOUT
+    });
+  });
+
+  afterEach(() => {
+    clock.restore();
+    sandbox.restore();
+  });
+
+  it('requires a queueUrl to be set', () => {
+    assert.throws(() => {
+      Consumer.create({
+        region: REGION,
+        handleMessage
+      });
+    });
+  });
+
+  it('requires a handleMessage or handleMessagesBatch function to be set', () => {
+    assert.throws(() => {
+      new Consumer({
+        handleMessage: undefined,
+        region: REGION,
+        queueUrl: QUEUE_URL
+      });
+    });
+  });
+
+  it('requires the batchSize option to be no greater than 10', () => {
+    assert.throws(() => {
+      new Consumer({
+        region: REGION,
+        queueUrl: QUEUE_URL,
+        handleMessage,
+        batchSize: 11
+      });
+    });
+  });
+
+  it('requires the batchSize option to be greater than 0', () => {
+    assert.throws(() => {
+      new Consumer({
+        region: REGION,
+        queueUrl: QUEUE_URL,
+        handleMessage,
+        batchSize: -1
+      });
+    });
+  });
+
+  it('requires visibilityTimeout to be set with heartbeatInterval', () => {
+    assert.throws(() => {
+      new Consumer({
+        region: REGION,
+        queueUrl: QUEUE_URL,
+        handleMessage,
+        heartbeatInterval: 30
+      });
+    });
+  });
+
+  it('requires heartbeatInterval to be less than visibilityTimeout', () => {
+    assert.throws(() => {
+      new Consumer({
+        region: REGION,
+        queueUrl: QUEUE_URL,
+        handleMessage,
+        heartbeatInterval: 30,
+        visibilityTimeout: 30
+      });
+    });
+  });
+
+  describe('.create', () => {
+    it('creates a new instance of a Consumer object', () => {
+      const instance = Consumer.create({
+        region: REGION,
+        queueUrl: QUEUE_URL,
+        batchSize: 1,
+        visibilityTimeout: 10,
+        waitTimeSeconds: 10,
+        handleMessage
+      });
+
+      assert.instanceOf(instance, Consumer);
+    });
+  });
+
+  describe('.start', () => {
+    it('fires an error event when an error occurs receiving a message', async () => {
+      const receiveErr = new Error('Receive error');
+
+      sqs.send.withArgs(mockReceiveMessage).rejects(receiveErr);
+
+      consumer.start();
+
+      const err: any = await pEvent(consumer, 'error');
+
+      consumer.stop();
+      assert.ok(err);
+      assert.equal(err.message, 'SQS receive message failed: Receive error');
+    });
+
+    it('retains sqs error information', async () => {
+      const receiveErr = new MockSQSError('Receive error');
+      receiveErr.name = 'short code';
+      receiveErr.$retryable = {
+        throttling: false
+      };
+      receiveErr.$metadata = {
+        httpStatusCode: 403
+      };
+      receiveErr.time = new Date();
+      receiveErr.$service = 'service';
+
+      sqs.send.withArgs(mockReceiveMessage).rejects(receiveErr);
+
+      consumer.start();
+      const err: any = await pEvent(consumer, 'error');
+      consumer.stop();
+
+      assert.ok(err);
+      assert.equal(err.message, 'SQS receive message failed: Receive error');
+      assert.equal(err.code, receiveErr.name);
+      assert.equal(err.retryable, receiveErr.$retryable.throttling);
+      assert.equal(err.statusCode, receiveErr.$metadata.httpStatusCode);
+      assert.equal(err.time.toString(), receiveErr.time.toString());
+      assert.equal(err.service, receiveErr.$service);
+      assert.equal(err.fault, receiveErr.$fault);
+    });
+
+    it('fires a timeout event if handler function takes too long', async () => {
+      const handleMessageTimeout = 500;
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        region: REGION,
+        handleMessage: () =>
+          new Promise((resolve) => setTimeout(resolve, 1000)),
+        handleMessageTimeout,
+        sqs,
+        authenticationErrorTimeout: AUTHENTICATION_ERROR_TIMEOUT
+      });
+
+      consumer.start();
+      const [err]: any = await Promise.all([
+        pEvent(consumer, 'timeout_error'),
+        clock.tickAsync(handleMessageTimeout)
+      ]);
+      consumer.stop();
+
+      assert.ok(err);
+      assert.equal(
+        err.message,
+        `Message handler timed out after ${handleMessageTimeout}ms: Operation timed out.`
+      );
+    });
+
+    it('handles unexpected exceptions thrown by the handler function', async () => {
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        region: REGION,
+        handleMessage: () => {
+          throw new Error('unexpected parsing error');
+        },
+        sqs,
+        authenticationErrorTimeout: AUTHENTICATION_ERROR_TIMEOUT
+      });
+
+      consumer.start();
+      const err: any = await pEvent(consumer, 'processing_error');
+      consumer.stop();
+
+      assert.ok(err);
+      assert.equal(
+        err.message,
+        'Unexpected message handler failure: unexpected parsing error'
+      );
+    });
+
+    it('fires an error event when an error occurs deleting a message', async () => {
+      const deleteErr = new Error('Delete error');
+
+      handleMessage.resolves(null);
+      sqs.send.withArgs(mockDeleteMessage).rejects(deleteErr);
+
+      consumer.start();
+      const err: any = await pEvent(consumer, 'error');
+      consumer.stop();
+
+      assert.ok(err);
+      assert.equal(err.message, 'SQS delete message failed: Delete error');
+    });
+
+    it('fires a `processing_error` event when a non-`SQSError` error occurs processing a message', async () => {
+      const processingErr = new Error('Processing error');
+
+      handleMessage.rejects(processingErr);
+
+      consumer.start();
+      const [err, message] = await pEvent<
+        string | symbol,
+        { [key: string]: string }[]
+      >(consumer, 'processing_error', {
+        multiArgs: true
+      });
+      consumer.stop();
+
+      assert.equal(
+        err instanceof Error ? err.message : '',
+        'Unexpected message handler failure: Processing error'
+      );
+      assert.equal(message.MessageId, '123');
+    });
+
+    it('fires an `error` event when an `SQSError` occurs processing a message', async () => {
+      const sqsError = new Error('Processing error');
+      sqsError.name = 'SQSError';
+
+      handleMessage.resolves(sqsError);
+      sqs.send.withArgs(mockDeleteMessage).rejects(sqsError);
+
+      consumer.start();
+      const [err, message] = await pEvent<
+        string | symbol,
+        { [key: string]: string }[]
+      >(consumer, 'error', {
+        multiArgs: true
+      });
+      consumer.stop();
+
+      assert.equal(err.message, 'SQS delete message failed: Processing error');
+      assert.equal(message.MessageId, '123');
+    });
+
+    it('waits before repolling when a credentials error occurs', async () => {
+      const credentialsErr = {
+        name: 'CredentialsError',
+        message: 'Missing credentials in config'
+      };
+      sqs.send.withArgs(mockReceiveMessage).rejects(credentialsErr);
+      const errorListener = sandbox.stub();
+      consumer.on('error', errorListener);
+
+      consumer.start();
+      await clock.tickAsync(AUTHENTICATION_ERROR_TIMEOUT);
+      consumer.stop();
+
+      sandbox.assert.calledTwice(errorListener);
+      sandbox.assert.calledWithMatch(sqs.send.firstCall, mockReceiveMessage);
+      sandbox.assert.calledWithMatch(sqs.send.secondCall, mockReceiveMessage);
+    });
+
+    it('waits before repolling when a 403 error occurs', async () => {
+      const invalidSignatureErr = {
+        $metadata: {
+          httpStatusCode: 403
+        },
+        message: 'The security token included in the request is invalid'
+      };
+      sqs.send.withArgs(mockReceiveMessage).rejects(invalidSignatureErr);
+      const errorListener = sandbox.stub();
+      consumer.on('error', errorListener);
+
+      consumer.start();
+      await clock.tickAsync(AUTHENTICATION_ERROR_TIMEOUT);
+      consumer.stop();
+
+      sandbox.assert.calledTwice(errorListener);
+      sandbox.assert.calledWithMatch(sqs.send.firstCall, mockReceiveMessage);
+      sandbox.assert.calledWithMatch(sqs.send.secondCall, mockReceiveMessage);
+    });
+
+    it('waits before repolling when a UnknownEndpoint error occurs', async () => {
+      const unknownEndpointErr = {
+        name: 'UnknownEndpoint',
+        message:
+          'Inaccessible host: `sqs.eu-west-1.amazonaws.com`. This service may not be available in the `eu-west-1` region.'
+      };
+      sqs.send.withArgs(mockReceiveMessage).rejects(unknownEndpointErr);
+      const errorListener = sandbox.stub();
+      consumer.on('error', errorListener);
+
+      consumer.start();
+      await clock.tickAsync(AUTHENTICATION_ERROR_TIMEOUT);
+      consumer.stop();
+
+      sandbox.assert.calledTwice(errorListener);
+      sandbox.assert.calledTwice(sqs.send);
+      sandbox.assert.calledWithMatch(sqs.send.firstCall, mockReceiveMessage);
+      sandbox.assert.calledWithMatch(sqs.send.secondCall, mockReceiveMessage);
+    });
+
+    it('waits before repolling when a polling timeout is set', async () => {
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        region: REGION,
+        handleMessage,
+        sqs,
+        authenticationErrorTimeout: AUTHENTICATION_ERROR_TIMEOUT,
+        pollingWaitTimeMs: POLLING_TIMEOUT
+      });
+
+      consumer.start();
+      await clock.tickAsync(POLLING_TIMEOUT);
+      consumer.stop();
+
+      sandbox.assert.callCount(sqs.send, 4);
+      sandbox.assert.calledWithMatch(sqs.send.firstCall, mockReceiveMessage);
+      sandbox.assert.calledWithMatch(sqs.send.secondCall, mockDeleteMessage);
+      sandbox.assert.calledWithMatch(sqs.send.thirdCall, mockReceiveMessage);
+      sandbox.assert.calledWithMatch(sqs.send.getCall(3), mockDeleteMessage);
+    });
+
+    it('fires a message_received event when a message is received', async () => {
+      consumer.start();
+      const message = await pEvent(consumer, 'message_received');
+      consumer.stop();
+
+      assert.equal(message, response.Messages[0]);
+    });
+
+    it('fires a message_processed event when a message is successfully deleted', async () => {
+      handleMessage.resolves();
+
+      consumer.start();
+      const message = await pEvent(consumer, 'message_received');
+      consumer.stop();
+
+      assert.equal(message, response.Messages[0]);
+    });
+
+    it('calls the handleMessage function when a message is received', async () => {
+      consumer.start();
+      await pEvent(consumer, 'message_processed');
+      consumer.stop();
+
+      sandbox.assert.calledWith(handleMessage, response.Messages[0]);
+    });
+
+    it('deletes the message when the handleMessage function is called', async () => {
+      handleMessage.resolves();
+
+      consumer.start();
+      await pEvent(consumer, 'message_processed');
+      consumer.stop();
+
+      sandbox.assert.calledWith(sqs.send.secondCall, mockDeleteMessage);
+      sandbox.assert.match(
+        sqs.send.secondCall.args[0].input,
+        sinon.match({
+          QueueUrl: QUEUE_URL,
+          ReceiptHandle: 'receipt-handle'
+        })
+      );
+    });
+
+    it("doesn't delete the message when a processing error is reported", async () => {
+      handleMessage.rejects(new Error('Processing error'));
+
+      consumer.start();
+      await pEvent(consumer, 'processing_error');
+      consumer.stop();
+
+      sandbox.assert.neverCalledWithMatch(sqs.send, mockDeleteMessage);
+    });
+
+    it('consumes another message once one is processed', async () => {
+      handleMessage.resolves();
+
+      consumer.start();
+      await clock.runToLastAsync();
+      consumer.stop();
+
+      sandbox.assert.calledTwice(handleMessage);
+    });
+
+    it("doesn't consume more messages when called multiple times", () => {
+      sqs.send
+        .withArgs(mockReceiveMessage)
+        .resolves(new Promise((res) => setTimeout(res, 100)));
+      consumer.start();
+      consumer.start();
+      consumer.start();
+      consumer.start();
+      consumer.start();
+      consumer.stop();
+
+      sqs.send.calledOnceWith(mockReceiveMessage);
+    });
+
+    it('consumes multiple messages when the batchSize is greater than 1', async () => {
+      sqs.send.withArgs(mockReceiveMessage).resolves({
+        Messages: [
+          {
+            ReceiptHandle: 'receipt-handle-1',
+            MessageId: '1',
+            Body: 'body-1'
+          },
+          {
+            ReceiptHandle: 'receipt-handle-2',
+            MessageId: '2',
+            Body: 'body-2'
+          },
+          {
+            ReceiptHandle: 'receipt-handle-3',
+            MessageId: '3',
+            Body: 'body-3'
+          }
+        ]
+      });
+
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        messageAttributeNames: ['attribute-1', 'attribute-2'],
+        region: REGION,
+        handleMessage,
+        batchSize: 3,
+        sqs
+      });
+
+      consumer.start();
+      await pEvent(consumer, 'message_received');
+      consumer.stop();
+
+      sandbox.assert.callCount(handleMessage, 3);
+      sandbox.assert.calledWithMatch(sqs.send.firstCall, mockReceiveMessage);
+      sandbox.assert.match(
+        sqs.send.firstCall.args[0].input,
+        sinon.match({
+          QueueUrl: QUEUE_URL,
+          AttributeNames: [],
+          MessageAttributeNames: ['attribute-1', 'attribute-2'],
+          MaxNumberOfMessages: 3,
+          WaitTimeSeconds: AUTHENTICATION_ERROR_TIMEOUT,
+          VisibilityTimeout: undefined
+        })
+      );
+    });
+
+    it("consumes messages with message attribute 'ApproximateReceiveCount'", async () => {
+      const messageWithAttr = {
+        ReceiptHandle: 'receipt-handle-1',
+        MessageId: '1',
+        Body: 'body-1',
+        Attributes: {
+          ApproximateReceiveCount: 1
+        }
+      };
+
+      sqs.send.withArgs(mockReceiveMessage).resolves({
+        Messages: [messageWithAttr]
+      });
+
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        attributeNames: ['ApproximateReceiveCount'],
+        region: REGION,
+        handleMessage,
+        sqs
+      });
+
+      consumer.start();
+      const message = await pEvent(consumer, 'message_received');
+      consumer.stop();
+
+      sandbox.assert.calledWith(sqs.send, mockReceiveMessage);
+      sandbox.assert.match(
+        sqs.send.firstCall.args[0].input,
+        sinon.match({
+          QueueUrl: QUEUE_URL,
+          AttributeNames: ['ApproximateReceiveCount'],
+          MessageAttributeNames: [],
+          MaxNumberOfMessages: 1,
+          WaitTimeSeconds: AUTHENTICATION_ERROR_TIMEOUT,
+          VisibilityTimeout: undefined
+        })
+      );
+
+      assert.equal(message, messageWithAttr);
+    });
+
+    it('fires an emptyQueue event when all messages have been consumed', async () => {
+      sqs.send.withArgs(mockReceiveMessage).resolves({});
+
+      consumer.start();
+      await pEvent(consumer, 'empty');
+      consumer.stop();
+    });
+
+    it('terminate message visibility timeout on processing error', async () => {
+      handleMessage.rejects(new Error('Processing error'));
+
+      consumer.terminateVisibilityTimeout = true;
+
+      consumer.start();
+      await pEvent(consumer, 'processing_error');
+      consumer.stop();
+
+      sandbox.assert.calledWith(
+        sqs.send.secondCall,
+        mockChangeMessageVisibility
+      );
+      sandbox.assert.match(
+        sqs.send.secondCall.args[0].input,
+        sinon.match({
+          QueueUrl: QUEUE_URL,
+          ReceiptHandle: 'receipt-handle',
+          VisibilityTimeout: 0
+        })
+      );
+    });
+
+    it('does not terminate visibility timeout when `terminateVisibilityTimeout` option is false', async () => {
+      handleMessage.rejects(new Error('Processing error'));
+      consumer.terminateVisibilityTimeout = false;
+
+      consumer.start();
+      await pEvent(consumer, 'processing_error');
+      consumer.stop();
+
+      sqs.send.neverCalledWith(mockChangeMessageVisibility);
+    });
+
+    it('fires error event when failed to terminate visibility timeout on processing error', async () => {
+      handleMessage.rejects(new Error('Processing error'));
+
+      const sqsError = new Error('Processing error');
+      sqsError.name = 'SQSError';
+      sqs.send.withArgs(mockChangeMessageVisibility).rejects(sqsError);
+      consumer.terminateVisibilityTimeout = true;
+
+      consumer.start();
+      await pEvent(consumer, 'error');
+      consumer.stop();
+
+      sandbox.assert.calledWith(
+        sqs.send.secondCall,
+        mockChangeMessageVisibility
+      );
+      sandbox.assert.match(
+        sqs.send.secondCall.args[0].input,
+        sinon.match({
+          QueueUrl: QUEUE_URL,
+          ReceiptHandle: 'receipt-handle',
+          VisibilityTimeout: 0
+        })
+      );
+    });
+
+    it('fires response_processed event for each batch', async () => {
+      sqs.send.withArgs(mockReceiveMessage).resolves({
+        Messages: [
+          {
+            ReceiptHandle: 'receipt-handle-1',
+            MessageId: '1',
+            Body: 'body-1'
+          },
+          {
+            ReceiptHandle: 'receipt-handle-2',
+            MessageId: '2',
+            Body: 'body-2'
+          }
+        ]
+      });
+      handleMessage.resolves(null);
+
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        messageAttributeNames: ['attribute-1', 'attribute-2'],
+        region: REGION,
+        handleMessage,
+        batchSize: 2,
+        sqs
+      });
+
+      consumer.start();
+      await pEvent(consumer, 'response_processed');
+      consumer.stop();
+
+      sandbox.assert.callCount(handleMessage, 2);
+    });
+
+    it('calls the handleMessagesBatch function when a batch of messages is received', async () => {
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        messageAttributeNames: ['attribute-1', 'attribute-2'],
+        region: REGION,
+        handleMessageBatch,
+        batchSize: 2,
+        sqs
+      });
+
+      consumer.start();
+      await pEvent(consumer, 'response_processed');
+      consumer.stop();
+
+      sandbox.assert.callCount(handleMessageBatch, 1);
+    });
+
+    it('prefers handleMessagesBatch over handleMessage when both are set', async () => {
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        messageAttributeNames: ['attribute-1', 'attribute-2'],
+        region: REGION,
+        handleMessageBatch,
+        handleMessage,
+        batchSize: 2,
+        sqs
+      });
+
+      consumer.start();
+      await pEvent(consumer, 'response_processed');
+      consumer.stop();
+
+      sandbox.assert.callCount(handleMessageBatch, 1);
+      sandbox.assert.callCount(handleMessage, 0);
+    });
+
+    it('uses the correct visibility timeout for long running handler functions', async () => {
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        region: REGION,
+        handleMessage: () =>
+          new Promise((resolve) => setTimeout(resolve, 75000)),
+        sqs,
+        visibilityTimeout: 40,
+        heartbeatInterval: 30
+      });
+      const clearIntervalSpy = sinon.spy(global, 'clearInterval');
+
+      consumer.start();
+      await Promise.all([
+        pEvent(consumer, 'response_processed'),
+        clock.tickAsync(75000)
+      ]);
+      consumer.stop();
+
+      sandbox.assert.calledWith(
+        sqs.send.secondCall,
+        mockChangeMessageVisibility
+      );
+      sandbox.assert.match(
+        sqs.send.secondCall.args[0].input,
+        sinon.match({
+          QueueUrl: QUEUE_URL,
+          ReceiptHandle: 'receipt-handle',
+          VisibilityTimeout: 40
+        })
+      );
+      sandbox.assert.calledWith(
+        sqs.send.thirdCall,
+        mockChangeMessageVisibility
+      );
+      sandbox.assert.match(
+        sqs.send.thirdCall.args[0].input,
+        sinon.match({
+          QueueUrl: QUEUE_URL,
+          ReceiptHandle: 'receipt-handle',
+          VisibilityTimeout: 40
+        })
+      );
+      sandbox.assert.calledOnce(clearIntervalSpy);
+    });
+
+    it('passes in the correct visibility timeout for long running batch handler functions', async () => {
+      sqs.send.withArgs(mockReceiveMessage).resolves({
+        Messages: [
+          { MessageId: '1', ReceiptHandle: 'receipt-handle-1', Body: 'body-1' },
+          { MessageId: '2', ReceiptHandle: 'receipt-handle-2', Body: 'body-2' },
+          { MessageId: '3', ReceiptHandle: 'receipt-handle-3', Body: 'body-3' }
+        ]
+      });
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        region: REGION,
+        handleMessageBatch: () =>
+          new Promise((resolve) => setTimeout(resolve, 75000)),
+        batchSize: 3,
+        sqs,
+        visibilityTimeout: 40,
+        heartbeatInterval: 30
+      });
+      const clearIntervalSpy = sinon.spy(global, 'clearInterval');
+
+      consumer.start();
+      await Promise.all([
+        pEvent(consumer, 'response_processed'),
+        clock.tickAsync(75000)
+      ]);
+      consumer.stop();
+
+      sandbox.assert.calledWith(
+        sqs.send.secondCall,
+        mockChangeMessageVisibilityBatch
+      );
+      sandbox.assert.match(
+        sqs.send.secondCall.args[0].input,
+        sinon.match({
+          QueueUrl: QUEUE_URL,
+          Entries: sinon.match.array.deepEquals([
+            {
+              Id: '1',
+              ReceiptHandle: 'receipt-handle-1',
+              VisibilityTimeout: 40
+            },
+            {
+              Id: '2',
+              ReceiptHandle: 'receipt-handle-2',
+              VisibilityTimeout: 40
+            },
+            {
+              Id: '3',
+              ReceiptHandle: 'receipt-handle-3',
+              VisibilityTimeout: 40
+            }
+          ])
+        })
+      );
+      sandbox.assert.calledWith(
+        sqs.send.thirdCall,
+        mockChangeMessageVisibilityBatch
+      );
+      sandbox.assert.match(
+        sqs.send.thirdCall.args[0].input,
+        sinon.match({
+          QueueUrl: QUEUE_URL,
+          Entries: [
+            {
+              Id: '1',
+              ReceiptHandle: 'receipt-handle-1',
+              VisibilityTimeout: 40
+            },
+            {
+              Id: '2',
+              ReceiptHandle: 'receipt-handle-2',
+              VisibilityTimeout: 40
+            },
+            {
+              Id: '3',
+              ReceiptHandle: 'receipt-handle-3',
+              VisibilityTimeout: 40
+            }
+          ]
+        })
+      );
+      sandbox.assert.calledOnce(clearIntervalSpy);
+    });
+
+    it('emit error when changing visibility timeout fails', async () => {
+      sqs.send.withArgs(mockReceiveMessage).resolves({
+        Messages: [
+          { MessageId: '1', ReceiptHandle: 'receipt-handle-1', Body: 'body-1' }
+        ]
+      });
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        region: REGION,
+        handleMessage: () =>
+          new Promise((resolve) => setTimeout(resolve, 75000)),
+        sqs,
+        visibilityTimeout: 40,
+        heartbeatInterval: 30
+      });
+
+      const receiveErr = new MockSQSError('failed');
+      sqs.send.withArgs(mockChangeMessageVisibility).rejects(receiveErr);
+
+      consumer.start();
+      const [err]: any[] = await Promise.all([
+        pEvent(consumer, 'error'),
+        clock.tickAsync(75000)
+      ]);
+      consumer.stop();
+
+      assert.ok(err);
+      assert.equal(err.message, 'Error changing visibility timeout: failed');
+    });
+
+    it('emit error when changing visibility timeout fails for batch handler functions', async () => {
+      sqs.send.withArgs(mockReceiveMessage).resolves({
+        Messages: [
+          { MessageId: '1', ReceiptHandle: 'receipt-handle-1', Body: 'body-1' },
+          { MessageId: '2', ReceiptHandle: 'receipt-handle-2', Body: 'body-2' }
+        ]
+      });
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        region: REGION,
+        handleMessageBatch: () =>
+          new Promise((resolve) => setTimeout(resolve, 75000)),
+        sqs,
+        batchSize: 2,
+        visibilityTimeout: 40,
+        heartbeatInterval: 30
+      });
+
+      const receiveErr = new MockSQSError('failed');
+      sqs.send.withArgs(mockChangeMessageVisibilityBatch).rejects(receiveErr);
+
+      consumer.start();
+      const [err]: any[] = await Promise.all([
+        pEvent(consumer, 'error'),
+        clock.tickAsync(75000)
+      ]);
+      consumer.stop();
+
+      assert.ok(err);
+      assert.equal(err.message, 'Error changing visibility timeout: failed');
+    });
+  });
+
+  describe('.stop', () => {
+    it('stops the consumer polling for messages', async () => {
+      consumer.start();
+      consumer.stop();
+
+      await Promise.all([pEvent(consumer, 'stopped'), clock.runAllAsync()]);
+
+      sandbox.assert.calledOnce(handleMessage);
+    });
+
+    it('fires a stopped event when last poll occurs after stopping', async () => {
+      consumer.start();
+      consumer.stop();
+      await Promise.all([pEvent(consumer, 'stopped'), clock.runAllAsync()]);
+    });
+
+    it('fires a stopped event only once when stopped multiple times', async () => {
+      const handleStop = sandbox.stub().returns(null);
+
+      consumer.on('stopped', handleStop);
+
+      consumer.start();
+      consumer.stop();
+      consumer.stop();
+      consumer.stop();
+      await clock.runAllAsync();
+
+      sandbox.assert.calledOnce(handleStop);
+    });
+
+    it('fires a stopped event a second time if started and stopped twice', async () => {
+      const handleStop = sandbox.stub().returns(null);
+
+      consumer.on('stopped', handleStop);
+
+      consumer.start();
+      consumer.stop();
+      consumer.start();
+      consumer.stop();
+      await clock.runAllAsync();
+
+      sandbox.assert.calledTwice(handleStop);
+    });
+  });
+
+  describe('isRunning', async () => {
+    it('returns true if the consumer is polling', () => {
+      consumer.start();
+      assert.isTrue(consumer.isRunning);
+      consumer.stop();
+    });
+
+    it('returns false if the consumer is not polling', () => {
+      consumer.start();
+      consumer.stop();
+      assert.isFalse(consumer.isRunning);
+    });
+  });
+
+  describe('delete messages property', () => {
+    beforeEach(() => {
+      consumer = new Consumer({
+        queueUrl: QUEUE_URL,
+        region: REGION,
+        handleMessage,
+        sqs,
+        authenticationErrorTimeout: AUTHENTICATION_ERROR_TIMEOUT,
+        shouldDeleteMessages: false
+      });
+    });
+
+    it('do not deletes the message when the handleMessage function is called', async () => {
+      handleMessage.resolves();
+
+      consumer.start();
+      await pEvent(consumer, 'message_processed');
+      consumer.stop();
+
+      sandbox.assert.neverCalledWithMatch(sqs.send, mockDeleteMessage);
+    });
+  });
+});