Vitest vi.mock与require：模块导入机制对测试模拟的影响

// src/client-authenticator.js (被测试模块)
const { ssmClient, getParameterCommand } = require('./helpers/aws'); // 使用 require

// test/client-authenticator.test.js (测试文件)
import { it, describe, expect, vi, beforeEach } from 'vitest';
const ClientAuthenticator = require('../src/client-authenticator'); // 测试文件也使用 require
const { ssmClient, getParameterCommand } = require('../src/helpers/aws'); // 再次 require

const ssmClientMock = vi.fn();
const getParameterCommandMock = vi.fn();

vi.mock('../src/helpers/aws', () => {
    return {
        ssmClient: ssmClientMock,
        getParameterCommand: getParameterCommandMock,
    };
});

describe('ClientAuthenticator.authenticator Tests', () => {
    it('Should set correct client name', async () => {
        // Arrange
        console.log(ssmClient); // 此时输出的仍然是真实的 ssmClient 实现，而非 ssmClientMock
        // ... rest of the test ...
    });
});

// test/client-authenticator.test.js
import { it, describe, expect, vi, beforeEach } from 'vitest';
import ClientAuthenticator from '../src/client-authenticator'; // 将 require 转换为 import
import { ssmClient, getParameterCommand } from '../src/helpers/aws'; // 将 require 转换为 import

const ssmClientMock = vi.fn();
const getParameterCommandMock = vi.fn();

vi.mock('../src/helpers/aws', () => {
    return {
        ssmClient: ssmClientMock,
        getParameterCommand: getParameterCommandMock,
    };
});

describe('ClientAuthenticator.authenticator Tests', () => {
    it('Should set correct client name', async () => {
        // Arrange
        console.log(ssmClient); // 此时，如果 src/helpers/aws 也是 ESM，则会输出模拟版本
        // ... rest of the test ...
    });
});

// src/client-authenticator.js (修改后)
import { ssmClient, getParameterCommand } from './helpers/aws'; // 将 require 转换为 import

class ClientAuthenticator {
    constructor() {
        // ...
    }

    async authenticate() {
        // 使用 ssmClient 和 getParameterCommand
        const command = new getParameterCommand({ Name: 'some-param' });
        const data = await ssmClient.send(command);
        return data.Parameter.Value;
    }
}

export default ClientAuthenticator; // 导出为 ES 模块

// package.json
{
  "name": "my-project",
  "version": "1.0.0",
  "type": "module", // 添加此行，指示项目使用 ES 模块
  "main": "index.js",
  "scripts": {
    "test": "vitest"
  },
  "devDependencies": {
    "vitest": "^1.x.x"
  }
}

// test/client-authenticator.test.js (最终版本)
import { it, describe, expect, vi, beforeEach } from 'vitest';
import ClientAuthenticator from '../src/client-authenticator'; // ESM 导入
import { ssmClient, getParameterCommand } from '../src/helpers/aws'; // ESM 导入

const ssmClientMock = vi.fn();
const getParameterCommandMock = vi.fn();

vi.mock('../src/helpers/aws', () => {
    return {
        ssmClient: ssmClientMock,
        getParameterCommand: getParameterCommandMock,
    };
});

describe('ClientAuthenticator.authenticator Tests', () => {
    beforeEach(() => {
        // 重置 mock 状态，确保每个测试用例独立
        ssmClientMock.mockClear();
        getParameterCommandMock.mockClear();
    });

    it('Should call ssmClient and getParameterCommand with correct parameters', async () => {
        // Arrange
        ssmClientMock.mockReturnValueOnce({
            send: vi.fn().mockResolvedValueOnce({
                Parameter: { Value: 'mocked-secret' }
            })
        });
        // getParameterCommandMock 应该被实例化，所以我们模拟它的构造函数
        getParameterCommandMock.mockImplementation((params) => ({
            params, // 存储传入的参数，以便后续断言
            // 真实的 Command 类会有更多逻辑，这里仅为模拟
        }));

        const authenticator = new ClientAuthenticator();

        // Act
        const result = await authenticator.authenticate();

        // Assert
        expect(getParameterCommandMock).toHaveBeenCalledWith({ Name: 'some-param' });
        expect(ssmClientMock).toHaveBeenCalledTimes(1);
        expect(result).toBe('mocked-secret');
    });
});