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)
}

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

Answer 1

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

globalIndex
type Mock struct{}

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

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

1. 尝试作证。 AssertExpectations 可以帮到你

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