Testing 戈朗试验标准

我试图测试一些打印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")
}

我试着用它，但他们似乎不喜欢逃跑密码

---

有没有办法测试写入标准输出的内容？

使用

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

---

Go

测试标准输出的目的是什么？在将函数写入标准输出之前，验证函数是否生成所需的输出要容易得多。问题是标准输出是唯一的输出。该函数将返回

n int，err error

，就像

fmt.Println

一样，您可以用类似于

io.MultiWriter

的东西来替换

os.Stdout

，它返回一个

*os.File

，但只需重构代码以使其可测试性更容易。您不需要测试

fmt.Println

和

os.Stdout

，它们有自己的单元测试。是的，编写一个

FprintlnColor

，然后让

PrintlnColor

用

Stdout

调用它。它可以写入一个

bytes.Buffer

进行测试。好的，如果你把它作为答案添加，我会接受它。