mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-04 12:47:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

log to memory in integration tests

Browse source
This commit is contained in:
Marten Seemann 2019-04-01 01:39:38 +09:00
parent 358fce241a
commit a0bf7c7ed0
1 changed files with 17 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

25
integrationtests/tools/testlog/testlog.go
View file

@ -1,9 +1,11 @@
package testlog package testlog
import ( import (
"bytes"
"flag" "flag"
"log" "log"
"os" "os"
"sync"
"github.com/lucas-clemente/quic-go/internal/utils" "github.com/lucas-clemente/quic-go/internal/utils"
@ -11,9 +13,12 @@ import (
. "github.com/onsi/gomega" . "github.com/onsi/gomega"
) )
const logBufSize = 100 * 1 << 20 // initial size of the log buffer: 100 MB
var ( var (
logFileName string // the log file set in the ginkgo flags logFileName string // the log file set in the ginkgo flags
logFile *os.File logBufOnce sync.Once
logBuf *bytes.Buffer
) )
// read the logfile command line flag // read the logfile command line flag
@ -25,18 +30,22 @@ func init() {
var _ = BeforeEach(func() { var _ = BeforeEach(func() {
log.SetFlags(log.Ldate | log.Ltime | log.Lmicroseconds) log.SetFlags(log.Ldate | log.Ltime | log.Lmicroseconds)
if len(logFileName) > 0 { if Debug() {
var err error logBufOnce.Do(func() {
logFile, err = os.Create(logFileName) logBuf = bytes.NewBuffer(make([]byte, 0, logBufSize))
Expect(err).ToNot(HaveOccurred()) })
log.SetOutput(logFile)
utils.DefaultLogger.SetLogLevel(utils.LogLevelDebug) utils.DefaultLogger.SetLogLevel(utils.LogLevelDebug)
log.SetOutput(logBuf)
} }
}) })
var _ = AfterEach(func() { var _ = AfterEach(func() {
if len(logFileName) > 0 { if Debug() {
_ = logFile.Close() logFile, err := os.Create(logFileName)
Expect(err).ToNot(HaveOccurred())
logFile.Write(logBuf.Bytes())
logFile.Close()
logBuf.Reset()
} }
}) })