如何测试如何在main()函数中调用特定的方法

Question

我想测试newrelic.NewConfig和newrelic.NewApplication是否在main()函数中被调用。

import (
    "github.com/newrelic/go-agent"  
)

func main() {
    /* NewRelic configuration */
    newRelicConfig := newrelic.NewConfig("my-app-name", 
    os.Getenv("NEW_RELIC_LICENSE_KEY"))
    app, err := newrelic.NewApplication(newRelicConfig)

    // followed by other code
}

我是否应该将该代码移动到main包中的一个单独的函数中，例如：

func SetupNewRelicConfig() Application {
    newRelicConfig := newrelic.NewConfig("my-app-name", 
    os.Getenv("NEW_RELIC_LICENSE_KEY"))
    app, err := newrelic.NewApplication(newRelicConfig)

    if err != nil {
      log.Fatal(err)
    }

    return app
}

这样，我就可以检查是否调用了SetupNewRelicConfig。

，测试这个的正确方法是什么？

Answer 1

您希望通过自动测试或某种类型的运行时断言对此进行测试吗？

假设您希望在您的套件中添加一个自动测试：

您需要找到一种方法来模拟NewRelic包导出的函数。

这里描述了一种非常便宜的方法(“在Golang修补猴子”)：

https://husobee.github.io/golang/testing/unit-test/2015/06/08/golang-unit-testing.html

更全面的方法要求您将这些函数调用添加到可由测试套件交换的结构中。参见依赖注入，如下所述：

4342/dependency-injection-in-golang-e587c69478a8

最后，看看如何使用一个模拟框架。我对担架上的嘲弄非常幸运。