使用参数来执行测试代码的方法

今日不肯埋头，明日何以抬头！每日一句努力自己的话哈哈~哈喽，今天我将给大家带来一篇《使用参数来执行测试代码的方法》，主要内容是讲解等等，感兴趣的朋友可以收藏或者有更好的建议在评论提出，我都会认真看的！大家一起进步，一起学习！

我有两个文件 main.go 和 main_test.go

在main.go下

package main

import (
    "fmt"
    "os"
    "strconv"
)

func sum(a, b int) int {
    return a + b
}

func main() {
    a, _ := strconv.atoi(os.args[1])
    b, _ := strconv.atoi(os.args[2])

    fmt.println(sum(a, b))
}

在 main_test.go 下我有

package main

import (
    "flag"
    "fmt"
    "testing"
)

func TestMain(t *testing.M) {
    args1 := flag.Arg(0)
    args2 := flag.Arg(1)

    fmt.Print(args1, args2)

    os.Args = []string{args1, args2}

    t.Run()


}

我正在尝试通过 go test main_test.go -args 1 2 -v 运行 go test，但我没有得到正确的输出，任何人都可以指导我如何编写用于测试 main 函数的命令，以便它运行正确地。

正确答案

使用恒定示例（测试用例）进行的测试比在测试中使用任何交互式内容更有效率，因为您可能需要它自动运行多次。你为什么不像我的例子那样做呢？

main_test.go：

package main

import (
    "testing"
)

func testmain(t *testing.t) {

    for _, testcase := range []struct {
        a, b, result int
    }{
        {1, 2, 3},
        {5, 6, 11},
        {10, 2, 12},
    } {
        if sum(testcase.a, testcase.b) != testcase.result {
            t.fail()
        }
    }

}

看看这个例子也很好： Go By Example: Testing

这里是测试 main 函数并传递参数的代码：

func TestMain(m *testing.M) {
    a := make([]string, len(os.Args)+2)
    a[0] = os.Args[0]
    a[1] = "first arg"
    a[2] = "second arg"
    copy(a[3:], os.Args[1:])

    os.Args = a

    main()
}

理论要掌握，实操不能落！以上关于《使用参数来执行测试代码的方法》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！