测试 golang 命令行工具的核心在于模拟输入和捕获输出。1. 使用 bytes.buffer 模拟 os.stdin，将输入内容写入缓冲区并替换 os.stdin；2. 利用 os.pipe 更精确控制输入输出流，提升测试健壮性；3. 通过重定向 os.stdout/os.stderr 并读取管道内容来捕获程序输出；4. 使用 testify/assert 库简化断言逻辑，提高测试代码可读性和可维护性。这些方法共同确保命令行工具在各种场景下的行为符合预期。
测试 Golang 命令行工具，核心在于模拟用户输入（os.Stdin）和捕获程序输出（os.Stdout/os.Stderr）。这样才能验证工具在各种场景下的行为是否符合预期。
模拟 os.Stdin 和 os.Stdout 的最佳实践方法有很多，关键在于找到适合你项目复杂度和测试需求的方案。
使用 bytes.Buffer 模拟 os.Stdin
立即学习“go语言免费学习笔记（深入）”；
怎样使用 bytes.Buffer 模拟 os.Stdin？
bytes.Buffer 是一个内存中的字节缓冲区，非常适合模拟标准输入。你可以将需要输入的内容写入 bytes.Buffer，然后将其设置为 os.Stdin。下面是一个简单的例子：
package main
import (
"bytes"
"fmt"
"io"
"os"
"testing"
)
func main() {
var input string
fmt.Scanln(&input)
fmt.Println("You entered:", input)
}
func TestMainWithInput(t *testing.T) {
// 模拟输入
input := "test input"
oldStdin := os.Stdin // 保存原始的 os.Stdin
defer func() { os.Stdin = oldStdin }() // 测试结束后恢复
r, w, _ := os.Pipe()
os.Stdin = r
w.Write([]byte(input))
w.Close()
// 捕获输出
oldStdout := os.Stdout
defer func() { os.Stdout = oldStdout }()
rStdout, wStdout, _ := os.Pipe()
os.Stdout = wStdout
// 执行 main 函数 (或者你的命令行工具)
main()
// 读取捕获的输出
wStdout.Close()
var buf bytes.Buffer
io.Copy(&buf, rStdout)
output := buf.String()
// 验证输出
expectedOutput := "You entered: test input\n"
if output != expectedOutput {
t.Errorf("Expected output '%s', but got '%s'", expectedOutput, output)
}
}这段代码首先保存了原始的 os.Stdin，以便在测试结束后恢复。然后，创建了一个 bytes.Buffer，并将模拟的输入写入其中。接着，将 os.Stdin 设置为这个 bytes.Buffer。最后，调用 main 函数（或者你的命令行工具），并捕获其输出，验证输出是否符合预期。需要注意的是，这里使用了 os.Pipe 来更精确的控制输入输出流，避免潜在的并发问题。os.Pipe 创建了一对关联的文件描述符，一个用于读取，一个用于写入，比直接使用 bytes.Buffer 更健壮。
使用 os.Pipe 捕获 os.Stdout 和 os.Stderr
如何使用 os.Pipe 捕获命令行工具的输出？
os.Pipe 提供了一种在内存中创建管道的方法，可以用来捕获标准输出和标准错误。这对于测试命令行工具的输出非常有用。
package main
import (
"bytes"
"fmt"
"io"
"os"
"testing"
)
func myFunc() {
fmt.Println("This is a test output")
fmt.Fprintln(os.Stderr, "This is an error message")
}
func TestMyFunc(t *testing.T) {
// 捕获 stdout
oldStdout := os.Stdout
rStdout, wStdout, _ := os.Pipe()
os.Stdout = wStdout
// 捕获 stderr
oldStderr := os.Stderr
rStderr, wStderr, _ := os.Pipe()
os.Stderr = wStderr
// 执行函数
myFunc()
// 关闭写入端，读取输出
wStdout.Close()
wStderr.Close()
os.Stdout = oldStdout // 恢复 stdout
os.Stderr = oldStderr // 恢复 stderr
var stdoutBuf bytes.Buffer
io.Copy(&stdoutBuf, rStdout)
stdoutOutput := stdoutBuf.String()
var stderrBuf bytes.Buffer
io.Copy(&stderrBuf, rStderr)
stderrOutput := stderrBuf.String()
// 验证输出
expectedStdout := "This is a test output\n"
expectedStderr := "This is an error message\n"
if stdoutOutput != expectedStdout {
t.Errorf("Expected stdout '%s', but got '%s'", expectedStdout, stdoutOutput)
}
if stderrOutput != expectedStderr {
t.Errorf("Expected stderr '%s', but got '%s'", expectedStderr, stderrOutput)
}
}这段代码中，myFunc 函数同时向标准输出和标准错误输出内容。测试函数 TestMyFunc 使用 os.Pipe 分别捕获了标准输出和标准错误，并在测试结束后恢复了原始的 os.Stdout 和 os.Stderr。最后，它验证了捕获的输出是否与预期一致。
使用 testify/assert 进行断言
如何使用 testify/assert 简化断言过程？
testify/assert 是一个流行的 Golang 测试断言库，它提供了丰富的断言函数，可以简化测试代码，提高可读性。
package main
import (
"bytes"
"fmt"
"io"
"os"
"testing"
"github.com/stretchr/testify/assert"
)
func myFunc() {
fmt.Println("This is a test output")
fmt.Fprintln(os.Stderr, "This is an error message")
}
func TestMyFuncWithAssert(t *testing.T) {
// 捕获 stdout
oldStdout := os.Stdout
rStdout, wStdout, _ := os.Pipe()
os.Stdout = wStdout
// 捕获 stderr
oldStderr := os.Stderr
rStderr, wStderr, _ := os.Pipe()
os.Stderr = wStderr
// 执行函数
myFunc()
// 关闭写入端，读取输出
wStdout.Close()
wStderr.Close()
os.Stdout = oldStdout // 恢复 stdout
os.Stderr = oldStderr // 恢复 stderr
var stdoutBuf bytes.Buffer
io.Copy(&stdoutBuf, rStdout)
stdoutOutput := stdoutBuf.String()
var stderrBuf bytes.Buffer
io.Copy(&stderrBuf, rStderr)
stderrOutput := stderrBuf.String()
// 使用 assert 进行断言
expectedStdout := "This is a test output\n"
expectedStderr := "This is an error message\n"
assert.Equal(t, expectedStdout, stdoutOutput, "Stdout should match")
assert.Equal(t, expectedStderr, stderrOutput, "Stderr should match")
}在这个例子中，我们使用了 assert.Equal 函数来断言实际输出和预期输出是否相等。testify/assert 提供了许多其他有用的断言函数，例如 assert.Contains、assert.True、assert.False 等，可以根据需要选择使用。使用 testify/assert 可以使测试代码更加简洁易懂。
