test: fix QueueMessageNotFoundError checking

weyoss · Nov 20, 2023 · 6053cee · 6053cee
1 parent 9ef8ade
commit 6053cee
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/tests/deleting-messages/test00003.test.ts b/tests/tests/deleting-messages/test00003.test.ts
@@ -16,6 +16,7 @@ import { getQueueMessages } from '../../common/queue-messages';
 import { getQueueDeadLetteredMessages } from '../../common/queue-dead-lettered-messages';
 import { getQueuePendingMessages } from '../../common/queue-pending-messages';
 import { getQueueAcknowledgedMessages } from '../../common/queue-acknowledged-messages';
+import { QueueMessageNotFoundError } from '../../../src/lib/queue/errors';
 
 test('Combined test: Delete an acknowledged message. Check pending, acknowledged, and dead-letter message. Check queue metrics.', async () => {
   await createQueue(defaultQueue, false);
@@ -67,5 +68,5 @@ test('Combined test: Delete an acknowledged message. Check pending, acknowledged
       queue,
       message.getRequiredId(),
     );
-  }).rejects.toThrow('Message not found');
+  }).rejects.toThrow(QueueMessageNotFoundError);
 });
diff --git a/tests/tests/deleting-messages/test00004.test.ts b/tests/tests/deleting-messages/test00004.test.ts
@@ -16,6 +16,7 @@ import { getQueueDeadLetteredMessages } from '../../common/queue-dead-lettered-m
 import { getQueuePendingMessages } from '../../common/queue-pending-messages';
 import { getQueueAcknowledgedMessages } from '../../common/queue-acknowledged-messages';
 import { getQueueMessages } from '../../common/queue-messages';
+import { QueueMessageNotFoundError } from '../../../src/lib/queue/errors';
 
 test('Combined test: Delete a dead-letter message. Check pending, acknowledged, and dead-letter message. Check queue metrics.', async () => {
   await createQueue(defaultQueue, false);
@@ -69,5 +70,5 @@ test('Combined test: Delete a dead-letter message. Check pending, acknowledged,
       queue,
       message.getRequiredId(),
     );
-  }).rejects.toThrow('Message not found');
+  }).rejects.toThrow(QueueMessageNotFoundError);
 });