chore: improving tests

bbc · Oct 16, 2023 · e73df6d · e73df6d
1 parent 63d146a
commit e73df6d
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/test/tests/consumer.test.ts b/test/tests/consumer.test.ts
@@ -480,27 +480,25 @@ describe('Consumer', () => {
     });
 
     it('calls the preReceiveMessageCallback and postReceiveMessageCallback function before receiving a message', async () => {
-      let callbackCalls = 0;
+      const preReceiveMessageCallbackStub = sandbox.stub().resolves(null);
+      const postReceiveMessageCallbackStub = sandbox.stub().resolves(null);
 
       consumer = new Consumer({
         queueUrl: QUEUE_URL,
         region: REGION,
         handleMessage,
         sqs,
         authenticationErrorTimeout: AUTHENTICATION_ERROR_TIMEOUT,
-        preReceiveMessageCallback: async () => {
-          callbackCalls++;
-        },
-        postReceiveMessageCallback: async () => {
-          callbackCalls++;
-        }
+        preReceiveMessageCallback: preReceiveMessageCallbackStub,
+        postReceiveMessageCallback: postReceiveMessageCallbackStub
       });
 
       consumer.start();
       await pEvent(consumer, 'message_processed');
       consumer.stop();
 
-      assert.equal(callbackCalls, 2);
+      sandbox.assert.calledOnce(preReceiveMessageCallbackStub);
+      sandbox.assert.calledOnce(postReceiveMessageCallbackStub);
     });
 
     it('deletes the message when the handleMessage function is called', async () => {