Golang：在调用 Init 方法之前模拟接口方法

Question

如何模拟在包 init() 方法中调用的内容？

例如：

ma​​in.go

var myService MyService = myservicepkg.New()

func init(){
   response := myService.get()
}

func otherMethod(){
   //do something
}

ma​​intest.go

func Test_otherMethod(){
  ctrl := NewController(t)
  defer ctrl.Finish()
  myServiceMock = myservicepkg.NewMock(myService)

  myServiceMock.EXPECT().get().return("success")
}

问题是在服务被mock替换之前调用了init()。

Answer 1

这是与可变的全球国家合作的问题。

我的建议是添加一个标志，以便在某些条件下不运行它，或者公开一个可以在内部测试中恢复/更新此全局变量的私有函数。

这反映了您的设计：如果测试很复杂，也许您应该重构。

创建一个带有字段Service的对象Application，在构造函数/构建器/工厂上创建将更容易测试。

init的使用非常微妙。除了将驱动程序注册到 sql 包中之外，我从不使用它（可能是为了处理标志）。

也许你可以在你的设计中添加更多的面向对象

Answer 2

您需要在 init() 内调用 otherMethod()。否则不能在 init() 之前调用它。

Answer 3

我找到了一个解决方法，您可以阻止 init 代码在测试中执行，并像这样单独测试该方法：

func init(){
  if len(os.Args) > 0 && strings.HasSuffix(os.Args[0], ".test") {
    log.Printf("skipping jwt.init() for testing")
  } else if len(os.Args) > 1 && strings.HasSuffix(os.Args[1], "-test.run") {
    log.Printf("skipping jwt.init() for testing")
  } else { 
    response := myService.get()
  }
}

这将阻止调用 init 服务调用。