go - 测试是否调用了函数

Question

鉴于此结构和功能：

type ExampleModule struct {
  DB                   *database.Store
  AnotherModule        AnotherModuleInterface
}

func(m *ExampleModule) A (i int, id int[]) error{
  err := m.AnotherModuke.SomeFunc(i, id)
}

如何进行单元测试以确保SomeFunc在运行该函数时调用它A？

Answer 1

1. 你可以模拟接口的实现，比如

globalIndex
type Mock struct{}

func (m Mock) SomeFunc(){
    globalIndex++
}

func testA(t *testing.T) {
    a := ExampleModule{
        AnotherModule: Mock{},
    }
    a.A()
    assert(globalIndex == 1)
}

2. 试试testify。AssertExpectations能帮你

https://github.com/stretchr/testify#mock-package