如何在 Go 中实现 CLI 命令的单元测试

Question

我正在启动一个使用 spf13/cobra 的新 OSS CLI 工具。作为 golang 的新手，我很难找出孤立测试命令的最佳方法。谁能给我一个如何测试命令的例子？几个警告：

1. 你不能从你的 init 函数返回一个 cobra.Command
2. 你不能get_test.go在 cmd 目录中......我的印象是 golang 最佳实践。
3. 我是 golang 的新手，请放轻松 :sweat_smile:

如果我错了，请纠正我。

这是我要测试的 cmd：https : //github.com/sahellebusch/raider/blob/3-add-get-alerts/cmd/get.go。

对想法、建议、批评持开放态度，无论什么。

Answer 1

有多种使用 go 实现 CLI 的方法。这是我开发的 CLI 的基本结构，它主要受 docker CLI 的影响，我还添加了单元测试。

您需要做的第一件事是将 CLI 作为界面。这将在名为“cli”的包内。

package cli

type Cli interface {
     // Have interface functions here
     sayHello() error
}

这将由 2 个 cli 实现：HelloCli（我们真正的 CLI）和 MockCli（用于单元测试）

package cli

type HelloCli struct {
}

func NewHelloCli() *HelloCli {
    cli := &HelloCli{
    }
    return cli
}

这里 HelloCli 将实现 sayHello 函数如下。

package cli

func (cli *HelloCli) SayHello() error {
    // Implement here
}

类似地，在名为test实现 cli 接口的包中会有一个模拟 cli ，它还将实现 sayHello 函数。

package test

type MockCli struct {
    }

    func NewMockCli() *HelloCli {
        cli := &MockCli{
        }
        return cli
    }

func (cli *MockCli) SayHello() error {
        // Mock implementation here
    }

现在我将展示如何添加命令。首先，我将拥有主包，这是我将添加所有新命令的地方。

package main

func newCliCommand(cli cli.Cli) *cobra.Command {
    cmd := &cobra.Command{
        Use:   "foo <command>"
    }

    cmd.AddCommand(
        newHelloCommand(cli),
    )
    return cmd
}

func main() {
    helloCli := cli.NewHelloCli()
    cmd := newCliCommand(helloCli)
    if err := cmd.Execute(); err != nil {
        // Do something here if execution fails
    }
}

func newHelloCommand(cli cli.Cli) *cobra.Command {
    cmd := &cobra.Command{
        Use:   "hello",
        Short: "Prints hello",
        Run: func(cmd *cobra.Command, args []string) {
            if err := pkg.RunHello(cli, args[0]); err != nil {
                // Do something if command fails
            }
        },
        Example: "  foo hello",
    }
    return cmd
}

在这里，我有一个名为hello. 接下来，我将在一个名为“pkg”的单独包中实现。

package pkg

func RunHello(cli cli.Cli) error {
    // Do something in this function
    cli.SayHello()
    return nil
}

单元测试也将包含在此包中名为hello_test.

package pkg

func TestRunHello(t *testing.T) {
    mockCli := test.NewMockCli()

    tests := []struct {
        name     string
    }{
        {
            name:     "my test 1",
        },
        {
            name:     "my test 2"
        },
    }
    for _, tst := range tests {
        t.Run(tst.name, func(t *testing.T) {
            err := SayHello(mockCli)
            if err != nil {
                t.Errorf("error in SayHello, %v", err)
            }
        })
    }
}

当您执行时foo hello，HelloCli将被传递给 sayHello() 函数，当您运行单元测试时，MockCli将被传递。