web-dev-qa-db-ja.com

go testでログ/出力を確認する方法は？

場合によってはエラーをログに記録するこの関数があります。

func readByte(/*...*/){
    // ...
    if err != nil {
        fmt.Println("ERROR")
        log.Print("Couldn't read first byte")
        return
    }
    // ...
}

ここで、テストファイルで、この関数の出力エラーを確認します。

    c.Assert(OUTPUT, check.Matches, "teste")

どうすればログにアクセスできますか？バッファを入れようとしましたが、うまくいきませんでした。 readByte関数コードを変更せずにこのログをキャッチする正しい方法は何ですか？

testinggo
14
Vivi

例えば、

readbyte_test.go：

package main

import (
    "bytes"
    "fmt"
    "io"
    "log"
    "os"
    "testing"
)

func readByte( /*...*/ ) {
    // ...
    err := io.EOF // force an error
    if err != nil {
        fmt.Println("ERROR")
        log.Print("Couldn't read first byte")
        return
    }
    // ...
}

func TestReadByte(t *testing.T) {
    var buf bytes.Buffer
    log.SetOutput(&buf)
    defer func() {
        log.SetOutput(os.Stderr)
    }()
    readByte()
    t.Log(buf.String())
}

出力：

$ go test -v readbyte_test.go 
=== RUN   TestReadByte
ERROR
--- PASS: TestReadByte (0.00s)
    readbyte_test.go:30: 2017/05/22 16:41:00 Couldn't read first byte
PASS
ok      command-line-arguments  0.004s
$
16
peterSO