我正在為 cli 應(yīng)用程序構(gòu)建 UI。我完成了這些功能，但我不知道如何測(cè)試它?；刭?gòu)：https ://github.com/erdaltsksn/cuifunc Success(message string) {    color.Success.Println("√", message)    os.Exit(0)}// Error prints a success message and exit status 1func Error(message string, err ...error) {    color.Danger.Println("X", message)    if len(err) > 0 {        for _, e := range err {            fmt.Println(" ", e.Error())        }    }    os.Exit(1)}我想為函數(shù)編寫單元測(cè)試。問(wèn)題是函數(shù)包含print和os.Exit()。我不知道如何為兩者編寫測(cè)試。本主題：如何在單元測(cè)試中測(cè)試函數(shù)的輸出 (stdout/stderr)有助于我測(cè)試打印函數(shù)。我需要添加os.Exit()我現(xiàn)在的解決方案：func captureOutput(f func()) string {    var buf bytes.Buffer    log.SetOutput(&buf)    f()    log.SetOutput(os.Stderr)    return buf.String()}func TestSuccess(t *testing.T) {    type args struct {        message string    }    tests := []struct {        name   string        args   args        output string    }{        {"Add test cases.", args{message: "my message"}, "ESC[1;32m"},    }    for _, tt := range tests {        t.Run(tt.name, func(t *testing.T) {            want := tt.output            got := captureOutput(func() {                cui.Success(tt.args.message)            })            got := err            if got.Error() != want {                t.Error("Got:", got, ",", "Want:", want)            }        })    }}