Azure服务连接函数单元测试：应使用Mock还是Stub？

对于这类依赖外部云服务SDK的代码，必须用Mock来模拟Azure相关的SDK类和方法，单元测试的核心是隔离外部依赖，绝对不能真的建立云服务连接——既浪费资源，也会让测试不稳定。

具体实现步骤如下：

1. Mock核心依赖类 #

用测试框架（比如Jest）Mock掉Azure SDK的导出类，避免加载真实的SDK代码：

// Mock Azure Blob存储SDK
jest.mock('@azure/storage-blob', () => ({
  BlobServiceClient: {
    fromConnectionString: jest.fn()
  }
}));

// Mock IoT Hub SDK及MQTT传输
jest.mock('azure-iot-device', () => ({
  Client: {
    fromConnectionString: jest.fn()
  }
}));
jest.mock('azure-iot-device-mqtt', () => ({
  Mqtt: jest.fn()
}));

2. 模拟方法返回值 #

给Mock的静态方法设置返回值，同时模拟容器客户端的getContainerClient方法：

const { BlobServiceClient } = require('@azure/storage-blob');
const { Client: DeviceClient } = require('azure-iot-device');

// 模拟Blob服务客户端的返回结构
const mockBlobServiceClient = {
  getContainerClient: jest.fn((containerName) => ({
    // 可以根据测试需求，添加容器客户端的必要属性/方法
  }))
};
BlobServiceClient.fromConnectionString.mockReturnValue(mockBlobServiceClient);

// 模拟IoT Hub客户端的返回结构
const mockIotClient = {};
DeviceClient.fromConnectionString.mockReturnValue(mockIotClient);

3. 编写测试用例并断言调用逻辑 #

实例化你的类，调用_connectToCloud后，验证Mock方法是否被正确调用：

test('_connectToCloud 正确初始化所有云服务客户端', () => {
  // 假设你的类名为CloudConnector
  const connector = new CloudConnector();
  connector.Blobcredentials = 'mock-blob-connection-string';
  connector.IOTcredentials = 'mock-iot-connection-string';

  connector._connectToCloud();

  // 验证Blob连接字符串是否正确传入
  expect(BlobServiceClient.fromConnectionString).toHaveBeenCalledWith('mock-blob-connection-string');
  // 验证两次调用getContainerClient，参数都是'name1'
  expect(mockBlobServiceClient.getContainerClient).toHaveBeenCalledTimes(2);
  expect(mockBlobServiceClient.getContainerClient).toHaveBeenCalledWith('name1');
  // 验证IoT Hub客户端初始化时传入了正确的连接字符串和MQTT传输
  expect(DeviceClient.fromConnectionString).toHaveBeenCalledWith(
    'mock-iot-connection-string',
    expect.any(Function)
  );
});

为什么选Mock而非Stub？ #

Stub的作用只是替换方法返回固定值，而我们需要验证SDK方法的调用逻辑是否符合预期：比如连接字符串是否传对、getContainerClient是否被调用了两次、IoT客户端是否用了MQTT传输。Mock既能提供模拟的返回值，又能做调用断言，完全匹配这个场景的需求。

内容的提问来源于stack exchange，提问作者user15183037