如何模拟 *exec.Cmd / exec.Command()？

当前位置： > > > > 如何模拟 *exec.Cmd / exec.Command()？

来源：stackoverflow
2024-04-25 10:24:35
0浏览
收藏

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《如何模拟 *exec.Cmd / exec.Command()？》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

问题内容

我需要模拟 exec.command()。

我可以使用以下方式模拟它：

var rname string
var rargs []string

mockexeccommand := func(name string, arg ...string) *exec.cmd {
    rname = name
    rargs = arg

    return nil
}

但是，这在实际代码中不起作用，因为它会抱怨 nil 指针，因为返回的 exec.cmd 调用 run()。

我试图像这样嘲笑它：

type mock exec.cmd

func (m *mock) run() error {
    return nil
}

var rname string
var rargs []string

mockexeccommand := func(name string, arg ...string) *exec.cmd {
    rname = name
    rargs = arg

    m := mock{}

    return &m
}

但它抱怨：不能使用 &m （*mock 类型的值）作为 return 语句中的 *exec.cmd 值compilerincompleteassign。

有什么办法可以解决这个问题吗？有没有更好的方法来模拟 exec.command()？

如果我返回“模拟”命令，模拟函数就会起作用，尽管我也更喜欢控制 run() 函数：

var rName string
var rArgs []string

mockExecCommand := func(name string, arg ...string) *exec.Cmd {
    rName = name
    rArgs = arg

    return exec.Command("echo")
}

正确答案

实际上有一种方法可以做到这一点。所有功劳均归于 文章。请查看以下内容的解释：

func fakeExecCommand(command string, args...string) *exec.Cmd {
    cs := []string{"-test.run=TestHelperProcess", "--", command}
    cs = append(cs, args...)
    cmd := exec.Command(os.Args[0], cs...)
    cmd.Env = []string{"GO_WANT_HELPER_PROCESS=1"}
    return cmd
}

func TestHelperProcess(t *testing.T){
    if os.Getenv("GO_WANT_HELPER_PROCESS") != "1" {
        return
    }
    os.Exit(0)
}

虽然劫持测试可执行文件以运行特定函数是有效的，但仅使用常规依赖项注入会更直接。不需要魔法。

设计一个可以运行命令的接口（例如 commandexecutor），然后将其中之一作为运行命令所需的任何函数的输入。然后，您可以在测试期间提供满足接口的模拟实现（手工制作或使用您选择的工具（如 gomock）生成）。为您的生产代码提供真正的实现（调用 exec 包）。您的模拟实现甚至可以对参数进行断言，以便您知道命令正在正确“执行”。

到这里，我们也就讲完了《如何模拟 *exec.Cmd / exec.Command()？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注公众号，带你了解更多关于的知识点！