testing - 如何在单元测试中测试函数的输出(stdout/stderr)

Question

我想测试一个简单的函数:

func (t *Thing) print(min_verbosity int, message string) {
    if t.verbosity >= minv {
        fmt.Print(message)
    }
}

但是我怎样才能测试函数实际发送到标准输出的内容呢？ Test::Output在 Perl 中做我想要的。我知道我可以编写自己的所有样板在 Go 中做同样的事情(如 here 所述):

orig = os.Stdout
r,w,_ = os.Pipe()
thing.print("Some message")
var buf bytes.Buffer
io.Copy(&buf, r)
w.Close()
os.Stdout = orig
if(buf.String() != "Some message") {
    t.Error("Failure!")
}

但是对于每一个测试来说，这都是很多额外的工作。我希望有一种更标准的方法，或者可能是一个抽象库来处理这个问题。

Answer 1

还有一点要记住，没有什么能阻止你编写函数以避免样板。

例如，我有一个使用 log 的命令行应用程序，我编写了这个函数:

func captureOutput(f func()) string {
    var buf bytes.Buffer
    log.SetOutput(&buf)
    f()
    log.SetOutput(os.Stderr)
    return buf.String()
}

然后这样使用:

output := captureOutput(func() {
    client.RemoveCertificate("www.example.com")
})
assert.Equal(t, "removed certificate www.example.com\n", output)

使用这个断言库:http://godoc.org/github.com/stretchr/testify/assert .