Cobra + Viper Golang如何测试子命令？

Question

我正在使用Go开发一个Web应用程序.到目前为止一切顺利,但现在我正在将Wercker整合为CI工具并开始关注测试.但我的应用程序在很大程度上依赖于Cobra/Viper配置/ flags/environment_variables方案,我不知道在运行我的测试套件之前如何正确初始化Viper值.任何帮助将非常感激.

Answer 1

当我使用Cobra/Viper或任何其他CLI助手组合时,我这样做的方法是让CLI工具运行一个函数,其唯一目的是获取参数并将它们传递给另一个将执行实际工作的方法.

这是使用Cobra的简短(和愚蠢)示例:

package main

import (
        "fmt"
        "os"

        "github.com/spf13/cobra"
)

func main() {
        var Cmd = &cobra.Command{
                Use:   "boom",
                Short: "Explode all the things!",
                Run:   Boom,
        }

        if err := Cmd.Execute(); err != nil {
                fmt.Println(err)
                os.Exit(-1)
        }
}

func Boom(cmd *cobra.Command, args []string) {
        boom(args...)
}

func boom(args ...string) {
        for _, arg := range args {
                println("boom " + arg)
        }
}

这里的Boom功能很难测试,但boom很容易.

你可以看到这另一个(非哑)例子这里(和对应的测试在这里).