Tracking the removed messages in the mock and fixing a JSDoc tag

adobe · Aug 18, 2023 · aa7bbe2 · aa7bbe2
1 parent 558231f
commit aa7bbe2
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/packages/queue-client/src/index.js b/packages/queue-client/src/index.js
@@ -68,7 +68,7 @@ export class QueueClient {
   /**
    * Gets the queue records from the context
    * @param {import('@adobe/helix-universal').UniversalContext} context
-   * @returns {Array<import('./queue.js').QueueRecord>} the queue records
+   * @returns {Array<QueueRecord>} the queue records
    */
   static extractQueueRecords(context) {
     return context.invocation?.event?.Records || [];

diff --git a/packages/queue-client/src/mock.js b/packages/queue-client/src/mock.js
@@ -12,17 +12,29 @@
 export class MockQueueClient {
   messages = [];
 
+  removed = [];
+
   async sendMessage(message) {
     const id = this.messages.push(message);
     return id.toString();
   }
 
+  /**
+   * Reads the message, reading from blob storage if required
+   * @param {string} messageBody
+   * @returns {Promise<Object>}
+   */
   // eslint-disable-next-line class-methods-use-this
-  async removeMessage() {
-    // do nothing
+  async readMessageBody(messageBody) {
+    return JSON.parse(messageBody);
+  }
+
+  async removeMessage(receiptHandle) {
+    this.removed.push(receiptHandle);
   }
 
   reset() {
     this.messages = [];
+    this.removed = [];
   }
 }