Golang 测试标准输出

Question

我正在尝试测试一些打印 ANSI 转义码的函数。例如// Print a line in a colorfunc PrintlnColor(color string, a ...interface{}) {    fmt.Print("\x1b[31m")    fmt.Print(a...)    fmt.Println("\x1b[0m")}我尝试使用示例来做到这一点，但他们似乎不喜欢转义码。有什么方法可以测试写入标准输出的内容吗？

Answer 1

使用fmt.Fprintto print toio.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")

}