Cobra:如何在测试中以编程方式设置标志

Question

我正在使用cobra来构建CLI，并希望模拟使用不同选项/标志集运行的命令。我一直在试图弄清楚如何使用眼镜蛇 API 在我的测试中设置标志，但还没有真正得到它。

我有这个：

// NewFooCmd returns a cobra.Command fitted to print output to the buffer for easier testing.
buf := &bytes.Buffer{}
cmd := package.NewFooCmd(buf)
cmd.Execute()
// some validations on the content of buf

到目前为止，我发现的最接近的东西是：

cmd.Flags().Set(name string, value string)

。但这似乎不对，因为虽然标志的名称都是字符串，但它们并不都将字符串作为值。此外，即使我有一个int标志并通过string(1)，它似乎也不起作用.

我在这里缺少一些简单的东西吗？

Answer 1

您可以使用 (c *Command) SetArgs(a []string) 函数来实现此目的。您的某些参数是整数或布尔值这一事实在这里无关紧要 - 毕竟，这就是用户将在命令行上输入的内容！

在

开发测试时，可以使用(c *Command) DebugFlags()函数，以确保传递的标志也正确插入。

我与 Cobra 的集成测试最终往往如下所示：

...
cmd := cli.RootCmd()
buf := new(bytes.Buffer)
cmd.SetOutput(buf)
cmd.SetArgs([]string{
    "--some-flag",
    fmt.Sprintf("--some-string=%s", value),
    fmt.Sprintf("--some-integer=%d", integer),
})
err := cmd.Execute()
...