有没有办法使用 gomock 模拟类型断言？

Question

我是 golang 新手，使用 Gomock 进行测试。我已经为接口foo 生成了模拟，但是在我的代码中有一段使用foo.(type) 的逻辑。

我可以知道是否有办法模拟这个并返回我选择的类型吗？如果没有，那么做这件事的好方法是什么？谢谢！

例如。代码sn-p：

// assume structs A and B implements the interface `foo` 
type foo interface {...}
type A struct {...}
type B struct {...} 

func DoSomething(i foo) {
  // Is there a way to mock this type assertion for i here? 
  switch currentType := i.(type) {
  case *A: 
    ...
  case *B:
    ...
  default:
    ...
  }
}

Answer 1

gomock 为接口生成新的实现。 （对于接口Foo 实现是MockFoo 结构）。在您的场景中，此类模拟仅涵盖 default 案例。

处理DoSomething(i foo) 的最简单方法是通过具体实现（基于您的场景的测试数据）