测试 – Golang测试标准输出

发布时间：2020-12-16 09:28:18 所属栏目：大数据来源：网络整理

导读：我正在尝试测试一些打印ANSI转义码的函数.例如 // Print a line in a colorfunc PrintlnColor(color string,a ...interface{}) { fmt.Print("x1b[31m") fmt.Print(a...) fmt.Println("x1b[0m")} 我尝试使用Examples来做,但他们似乎不喜欢转义码. 有没有办

我正在尝试测试一些打印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的内容？

解决方法

使用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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!