Golang测试标准输出

Question

我正在尝试测试一些打印ANSI转义码的函数.例如

// Print a line in a color
func PrintlnColor(color string, a ...interface{}) {
    fmt.Print("\x1b[31m")
    fmt.Print(a...)
    fmt.Println("\x1b[0m")
}

我尝试使用Examples来做,但他们似乎不喜欢转义码.

有没有办法测试写入stdout的内容？

Answer 1

使用fmt.Fprint打印io.Writer可以控制输出的写入位置.

var out io.Writer = os.Stdout

func main() {
    // write to Stdout
    PrintlnColor("foo")

    buf := &bytes.Buffer{}
    out = buf

    // write  to buffer
    PrintlnColor("foo")

    fmt.Println(buf.String())
}

// Print a line in a color
func PrintlnColor(a ...interface{}) {
    fmt.Fprint(out, "\x1b[31m")
    fmt.Fprint(out, a...)
    fmt.Fprintln(out, "\x1b[0m")
}

进入游戏