在 Go 中模拟接收器函数

Question

我正在尝试对调用该结构中其他接收器函数的接收器函数进行单元测试。

假设我想测试 Three() 并在下面模拟对 two() 的调用：

type MyStruct struct {
    a string
    b string
}

func (m *MyStruct) one() int {
    return 2
}

func (m *MyStruct) two() int {
    return m.one() * 2
}

func (m *MyStruct) Three() int {
    return m.two() * 2
}

我正在遵循以下answer的方法二。

我为每个想要进行单元测试的函数创建了一个自定义构造函数，并用模拟版本覆盖了这些方法。但我想，一旦函数数量增加，代码维护可能并不容易。

是否有任何首选方法来模拟此类函数？我希望官方文档有一些关于如何在不同场景中模拟事物的指南，类似于 Python 上的 mox 提供的。

另外，请注意，我不想使用第三方模拟库。

Answer 1

这是测试你的东西的一种非常不习惯的方式。 在其他语言中可能需要所有这些嘲笑，但是 请不要在 Go 中这样做。

在您提供的示例中测试代码的自然方法 将是： 1) 为MyStruct.one 编写一个表驱动测试 并确保您测试所有案例。现在你知道one 工作得很好 2) 对MyStruct.two 做同样的事情。 请注意，测试未导出的东西是可能的、有用的并且 在 Go 中很常见。现在不再需要模拟 一些方法，只需 3) 编写一些表驱动测试 MyStruct.Three 并检查它是否有效。

但也许你的方法 one 和 two 可以做一些更有趣的事情，并且 访问环境（文件系统、数据库、网络）和 您不希望您对Three 的测试依赖于此吗？ 所以重构你的代码！也许Three 不应该是一种方法 MyStruct 的一个函数，它采用 interface OneAndTwoer 作为参数，您的生产代码使用“real”调用Three MyStructs，而您的测试代码使用 InMemoryMyStrcuts 调用它 哪些不依赖环境？你可以称之为 mock，我称之为接口的不同实现。

在您的示例中，提供建议很简单：使用表驱动 测试 one、two 和 Three 并且不要模拟。 对于更现实的问题，建议可能会有所不同，但 在不知道的情况下很难给出一般性的建议 情况。最好的一般建议是：看看 标准库，您可以在其中找到几乎适用于 每个测试场景。

Answer 2

您可以进行重构以将两个作为函数传递给Three，这样您就可以直接模拟输入。

func main() {
    var ms MyStruct
    fmt.Println(ms.one())
    fmt.Println(ms.two())
    fmt.Println(ms.Three(ms.two))
}

type MyStruct struct {}

func (m *MyStruct) one() int {
    return 2
}

func (m *MyStruct) two() int {
    return m.one() * 2
}

func (m *MyStruct) Three(two func() int) int {
    return two() * 2
}