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

时间:2017-07-11 15:22:07

标签: function unit-testing go methods newrelic

我想测试newrelic.NewConfig函数中是否调用了newrelic.NewApplicationmain()

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是否被调用。

测试此功能的正确方法是什么?

1 个答案:

答案 0 :(得分:1)

您是希望通过自动化测试或某种类型的运行时断言来测试它吗?

假设您要为自己的套件添加自动化测试:

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

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

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

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

https://medium.com/@zach_4342/dependency-injection-in-golang-e587c69478a8

最后,考虑使用模拟框架。我在拉伸器的证明项目中嘲笑我很幸运。