Question

我正在尝试测试一些打印ANSI转义码的函数。 e.g。

// 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")
}

转到play

Golang测试标准输出

1 个答案: