From a0bf7c7ed00b7d66ef8717aa3e359bc2740f43e8 Mon Sep 17 00:00:00 2001
From: Marten Seemann <martenseemann@gmail.com>
Date: Mon, 1 Apr 2019 01:39:38 +0900
Subject: [PATCH] log to memory in integration tests

---
 integrationtests/tools/testlog/testlog.go | 25 +++++++++++++++--------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/integrationtests/tools/testlog/testlog.go b/integrationtests/tools/testlog/testlog.go
index d3206ba1..0d86597b 100644
--- a/integrationtests/tools/testlog/testlog.go
+++ b/integrationtests/tools/testlog/testlog.go
@@ -1,9 +1,11 @@
 package testlog
 
 import (
+	"bytes"
 	"flag"
 	"log"
 	"os"
+	"sync"
 
 	"github.com/lucas-clemente/quic-go/internal/utils"
 
@@ -11,9 +13,12 @@ import (
 	. "github.com/onsi/gomega"
 )
 
+const logBufSize = 100 * 1 << 20 // initial size of the log buffer: 100 MB
+
 var (
 	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
@@ -25,18 +30,22 @@ func init() {
 var _ = BeforeEach(func() {
 	log.SetFlags(log.Ldate | log.Ltime | log.Lmicroseconds)
 
-	if len(logFileName) > 0 {
-		var err error
-		logFile, err = os.Create(logFileName)
-		Expect(err).ToNot(HaveOccurred())
-		log.SetOutput(logFile)
+	if Debug() {
+		logBufOnce.Do(func() {
+			logBuf = bytes.NewBuffer(make([]byte, 0, logBufSize))
+		})
 		utils.DefaultLogger.SetLogLevel(utils.LogLevelDebug)
+		log.SetOutput(logBuf)
 	}
 })
 
 var _ = AfterEach(func() {
-	if len(logFileName) > 0 {
-		_ = logFile.Close()
+	if Debug() {
+		logFile, err := os.Create(logFileName)
+		Expect(err).ToNot(HaveOccurred())
+		logFile.Write(logBuf.Bytes())
+		logFile.Close()
+		logBuf.Reset()
 	}
 })