0

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

//文件.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();
        }
    });
})
}

// 文件.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，测试将工作文件。我已经尝试了所有提供的方法，它们都不起作用。难道我做错了什么？

4

0 回答 0