亚马逊网络服务 - 如何在 Jest 和 TypeScript 中正确模拟/伪造 AWS SecretManager

我正在尝试使用 Jest 对 aws-sdk 的 SecretsManager 进行单元测试，但遇到了一些问题，我有一个简单的缓存客户端

import { SecretsManager } from 'aws-sdk';
import { SecretsManagerIntegrationError } from '../errors';
const cache = new Map();

interface SecretsManagerConfig {
  region?: string;
}

export class SecretsManagerClient {
  secretsManager: SecretsManager;

  constructor(config: SecretsManagerConfig) {
    this.secretsManager = new SecretsManager(config);
  }

  async getSecretFromSecretsManager(secretName: string) {
    try {
      const params = { SecretId: secretName };
      const data = await this.secretsManager.getSecretValue(params).promise();

      if (data.SecretString) {
        return JSON.parse(data.SecretString);
      } else {
        return Buffer.from(data.SecretBinary as any, 'base64').toString('ascii');
      }
    } catch (error) {
      throw new SecretsManagerIntegrationError(`Error retrieving secrets: ${error}`);
    }
  }

  async getSecret(secretName: string) {
    if (!cache.get(secretName)) {
      const secret = await this.getSecretFromSecretsManager(secretName);
      cache.set(secretName, secret);
      return secret;
    }

    return cache.get(secretName);
  }
}

在我的测试文件中，我已经像这样模拟了 SDK……看起来很冗长，而 sinon 处理起来相当直接，这个断言失败了，我不确定为什么……

import { SecretsManagerClient } from '../../src/clients/secretsManagerClient';

jest.mock('aws-sdk', () => {
  return {
    SecretsManager: jest.fn(() => ({
      getSecretValue: jest.fn(() => ({
        promise: jest.fn(() => ({
          SecretString: JSON.stringify('cool-secrets-here'),
        })),
      })),
    })),
  };
});

describe('SecretsManagerClient', () => {
  let secretsManager: SecretsManagerClient;

  beforeEach(() => {
    secretsManager = new SecretsManagerClient({});
  });

  describe('when a cached value does not exist', () => {
    it('queries the secretsManager', async () => {
      const getSecretFromSecretsManager = jest.spyOn(secretsManager, 'getSecretFromSecretsManager');

      await secretsManager.getSecret('cool-secret');

      expect(getSecretFromSecretsManager).toHaveBeenCalledTimes(1);
    });
  });

  <!-- fails --->
  describe('when a cached value exists', () => {
    it('uses the cached value', async () => {
      const getSecretFromSecretsManager = jest.spyOn(secretsManager, 'getSecretFromSecretsManager');

      await secretsManager.getSecret('cool-secret');
      await secretsManager.getSecret('cool-secret');

      expect(getSecretFromSecretsManager).toHaveBeenCalledTimes(1);
    });
  });
});

答案＃1

我已经使用 sinon 存根实现了对 AWS 服务的模拟。下面是代码示例。

import AWS from 'aws-sdk'
import sinon from 'sinon'

let sinonSandbox

const beforeEach = (done) => {
   sinonSandbox = sinon.sandbox.create()
   done()
}

const afterEach = done => {
   sinonSandbox.restore()
   done()
} 
lab.test('test name', (done) => {
    sinonSandbox.stub(AWS, 'SecretsManager')
      .returns({
        getSecretValue: () => {
          return {
            SecretValue: 'Secret-Value'
          }
        }
    })
    done()
})

你还可以创建一个 aws-sdk-mock npm 模块，该模块模拟所有 AWS SDK 服务和方法。

它真的非常容易使用。只需调用 AWS.mock，传入服务、方法和一个存根函数即可。

AWS.mock('SecretsManager', 'getSecretValue', function(params, callback) {
    callback(null, 'success');
});

然后通过调用以下方法在测试后恢复这些方法：

AWS.restore('SecretsManager', 'getSecretValue');

Sinon 链接：

Sinon Stub 最新版本

存根-承诺-AWS-SDK-节点-JS

aws-sdk-mock 链接：

aws-sdk-模拟

https://www.npmjs.com/package/aws-sdk-mock