代码之家 › 专栏 › 技术社区 › johann

模拟“aws sdk”的“S3”功能(nodeJS和Jest)

aws-sdk jestjs node.js

johann · 技术社区 · 2 年前

我需要测试一个文件,该文件负责通过“aws sdk”(nodeJs+Jest)从S3检索数据。文件是:

const AWS = require('aws-sdk');

let S3 = null;

const getS3 = async () => {
  if (S3 === null) {
    const config = {
      endpoint: new AWS.Endpoint(process.env.S3_URL),
      s3ForcePathStyle: true,
    };
    S3 = new AWS.S3(config);
  }
  return S3;
};

module.exports.getObjectList = async (prefix) => {
  const params = {
    Bucket: process.env.S3_BUCKET,
    Delimiter: '/',
    Prefix: prefix,
  };
  const s3 = await getS3();
  const objectList = s3
    .listObjects(params)
    .promise()
    .then((data) => {
      const keys = data.Contents.map((c) => c.Key);
      return keys;
    })
    .catch((err) => {
      console.error(err);
      return null;
    });
  return objectList;
};

测试文件如下:

const s3Client = require('./s3Client');

const mockS3Instance = {
  listObjects: jest.fn().mockReturnThis(),
  promise: jest.fn().mockReturnThis(),
  catch: jest.fn(),
};

jest.mock('aws-sdk', () => {
  return {
    S3: jest.fn(() => mockS3Instance),
    Endpoint: jest.fn(() => {
      'blabla';
    }),
  };
});

describe('s3Client tests', () => {
  it('basic test', async () => {
    const getObjectListResult = await s3Client.getObjectList('test');
    expect(1).toBe(1);
  });
});

但会返回一条错误消息:

ypeError: s3.listObjects(...).promise(...).then is not a function

2 回复 | 直到 2 年前

gear4 2 年前

你需要添加一个 then 嘲笑自己 mockS3Instance 对象

johann 2 年前

感谢@gear4 reply,我能够修复我的代码: (主要部分是关于模拟返回的承诺)

const s3Client = require('./s3Client');

const mockS3Instance = {
  listObjects: jest.fn().mockReturnThis(),
  promise: jest.fn(() => {
    return new Promise((resolve, reject) => {
      return resolve({
        Contents: [{ Key: 'test-file-1' }, { Key: 'test-file-2' }],
      });
    });
  }),
  catch: jest.fn(),
};

jest.mock('aws-sdk', () => {
  return {
    S3: jest.fn(() => mockS3Instance),
    Endpoint: jest.fn(() => {
      'blabla';
    }),
  };
});

describe('s3Client tests', () => {
  it('Dummy tests: to be implemented', async () => {
    const getObjectListResult = await s3Client.getObjectList('test');
    expect(JSON.stringify(getObjectListResult)).toBe(
      JSON.stringify(['test-file-1', 'test-file-2']),
    );
  });
});

推荐文章

Ekaterina M · 如何使用Jest和测试库在JavaScript中进行单元测试?

1 年前

Gpack · 是否可以使用Jest收集vm.runInContext加载的代码的覆盖率?

1 年前

johann · 模拟“aws sdk”的“S3”功能(nodeJS和Jest)

2 年前

hikky36 · 为什么开玩笑。fn()可以分配给Typescript中的任何函数?

2 年前

kyrolos magdy · 用jest和

2 年前

Parth Chitroda · 无法读取未定义的属性“displayName”。反应+玩笑+酶

2 年前

Cazs · 在React onChange中测试功能组件

2 年前

Anshu Verma · 使用Jest和酶测试内部使用效果

2 年前

Kristesiashvili · 如果不能捕获源代码中的更改,那么用mock编写单元测试有什么意义呢?

2 年前

Alex Barbosa · 开玩笑说一个功能失败了,另一个功能却没有

2 年前