Question

我正在尝试模拟在file.js中定义的SQS调用。它是文件中的全局实例。因此，在我需要file.js进行测试时，设置了它的实例，并且未调用我的模拟方法。但是，如果我在所需的函数中本地设置了该SQS实例，则可以进行模拟。但这是错误的，因为每次调用该方法时都会设置该实例。如何在测试中模拟SQS？我尝试了所有在问题中给出的方法。他们都没有为我工作。

// file.js

const AWS = require('aws-sdk');
const sqs = new AWS.SQS();
const queueURL = config.sqs_connect.queue_url;
const params = {
    MaxNumberOfMessages: 10,
    QueueUrl: queueURL
};
exports.receiveMessages = async function () {
    // let sqs = new AWS.SQS();
    return new Promise((resolve, reject) => {
        sqs.receiveMessage(params, function (err, data) {
        if (err) {
            console.log("error")
            reject(err);
        } else if (data.Messages) {
            try {
                consumeAndDeleteMessages(data.Messages, err => {
                    if (err) reject(err);
                    else resolve();
                });
            } catch (error) {
                reject(error);
            }
        } else {
            // logger.log("No data in queue");
            resolve();
        }
    });
})
}

// file.test.js

const AWS = require('aws-sdk');
const consumer = require('path-to-file');

describe("foo", () => {
 it("updates all info", async () => {
    let delete_stack = [];
    AWSMock.setSDKInstance(AWS);
    AWSMock.mock('SQS', 'receiveMessage', (params, callback) => {
        callback(null, { Messages: [{ MessageId: '1234', ReceiptHandle: 'qwertyu', Body: JSON.stringify(update_payload) }] });
    });
    AWSMock.mock('SQS', 'deleteMessageBatch', (params, callback) => {
        delete_stack.push(params.Entries);
        callback(null, {});
    });

    await consumer.receiveMessages();
    AWSMock.restore('SQS');
    expect(delete_stack).toStrictEqual([
        [{ "Id": "1234", "ReceiptHandle": "qwertyu" }]
    ]);
});
});

如果我在receiveMessage中本地定义sqs，则测试将起作用。我尝试了提供的所有方法，但没有一个起作用。我在做错什么吗？

用玩笑嘲笑SQS

0 个答案: