Golang多次调用AWS服务的函数如何编写单元测试Mock？

你遇到的空指针panic和测试不生效问题，核心由以下几个错误导致：

• Mock结构体仅嵌入了iamiface.IAMAPI接口，没有重载GetUser、CreateUser两个被测方法，调用时会直接触发nil接口的空指针panic
• 测试代码存在多处笔误：待测试方法是CreateMyUser，测试时错误调用了ResetCredentials；mockedIamCall结构体定义缺少type关键字；mock变量类型前后不统一（mockSomeOutput和mockOutput不一致）
• 原函数存在语法缺陷：声明了(string, error)返回值但没有return语句；log.Fatalf格式化字符串缺少错误占位符%v，且log.Fatal会直接终止进程，导致测试无法正常捕获错误

1. Mock实现要覆盖所有调用的接口方法 #

对于多次调用第三方接口的场景，不要仅嵌入SDK提供的接口就结束，必须显式实现所有被测函数会调用的方法，可在mock结构体中增加调用计数、请求/返回配置字段，适配多调用的差异化返回需求：

type MockIAMClient struct {
  iamiface.IAMAPI
  // GetUser调用配置与调用记录
  GetUserExpectedErr error
  GetUserCalledTimes int
  GetUserLastInput   *iam.GetUserInput
  // CreateUser调用配置与调用记录
  CreateUserExpectedErr error
  CreateUserCalledTimes int
  CreateUserLastInput   *iam.CreateUserInput
}

// 重载GetUser方法
func (m *MockIAMClient) GetUser(input *iam.GetUserInput) (*iam.GetUserOutput, error) {
  m.GetUserCalledTimes++
  m.GetUserLastInput = input
  return nil, m.GetUserExpectedErr
}

// 重载CreateUser方法
func (m *MockIAMClient) CreateUser(input *iam.CreateUserInput) (*iam.CreateUserOutput, error) {
  m.CreateUserCalledTimes++
  m.CreateUserLastInput = input
  return nil, m.CreateUserExpectedErr
}

2. 单个测试用例仅覆盖一个分支场景 #

拆分不同业务分支为独立的子测试用例，每个用例单独配置mock返回值，避免多场景耦合：

• 测试用户已存在的场景：配置GetUserExpectedErr = nil，校验CreateUserCalledTimes = 0
• 测试用户不存在、创建成功的场景：配置GetUserExpectedErr = errors.New("user not found")、CreateUserExpectedErr = nil，校验两个接口调用次数均为1，且入参用户名符合预期
• 测试用户不存在、创建失败的场景：配置CreateUserExpectedErr = errors.New("create failed")，校验返回的错误符合预期

3. 必须验证调用参数和调用次数 #

测试断言不要仅判断返回值，还要校验mock中记录的调用次数、入参是否符合业务逻辑，避免出现“返回值正确但逻辑执行路径错误”的漏测情况。

4. 依赖统一通过构造函数注入 #

被测结构体中的第三方客户端不要在内部初始化，必须通过构造函数传入，确保测试时可以无侵入替换为mock实例，你当前iamService用iamInstance字段持有IAM客户端的设计是符合规范的。

5. 禁止在被测函数中使用进程终止类方法 #

不要在被测函数中使用log.Fatal、os.Exit这类会直接终止进程的方法，所有错误都应该向上返回，由调用方（包括测试代码）处理，避免测试进程意外终止。

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