英文:
Golang test stdout
问题
我正在尝试测试一些打印 ANSI 转义码的函数，例如：
// 打印带颜色的行
func PrintlnColor(color string, a ...interface{}) {
fmt.Print("\x1b[31m")
fmt.Print(a...)
fmt.Println("\x1b[0m")
}
我尝试使用 Examples 来测试，但它们似乎不支持转义码。
有没有办法测试输出到标准输出(stdout)的内容？
英文:
I am trying to test some functions that print ANSI escape codes. 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")
}
I tried using Examples to do it, but they don't seem to like escape codes.
Is there any way to test what is written to stdout?
答案1
得分: 13
使用fmt.Fprint将内容打印到io.Writer可以控制输出的位置。
var out io.Writer = os.Stdout
func main() {
// 输出到标准输出
PrintlnColor("foo")
buf := &bytes.Buffer{}
out = buf
// 输出到缓冲区
PrintlnColor("foo")
fmt.Println(buf.String())
}
// 以彩色打印一行
func PrintlnColor(a ...interface{}) {
fmt.Fprint(out, "\x1b[31m")
fmt.Fprint(out, a...)
fmt.Fprintln(out, "\x1b[0m")
}
英文:
Using fmt.Fprint to print to io.Writer lets you control where the output is written.
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")
}
Go play
通过集体智慧和协作来改善编程学习和解决问题的方式。致力于成为全球开发者共同参与的知识库，让每个人都能够通过互相帮助和分享经验来进步。
评论