synchronize testhelpers.LogHandler

OffchainLabs · Sep 16, 2022 · aa1c7fb · aa1c7fb
1 parent 2483230
commit aa1c7fb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/util/testhelpers/testhelpers.go b/util/testhelpers/testhelpers.go
@@ -7,6 +7,7 @@ import (
 	"math/rand"
 	"os"
 	"regexp"
+	"sync"
 	"testing"
 
 	"github.com/ethereum/go-ethereum/common"
@@ -42,6 +43,7 @@ func RandomAddress() common.Address {
 }
 
 type LogHandler struct {
+	mutex         sync.Mutex
 	t             *testing.T
 	records       []log.Record
 	streamHandler log.Handler
@@ -51,13 +53,17 @@ func (h *LogHandler) Log(record *log.Record) error {
 	if err := h.streamHandler.Log(record); err != nil {
 		return err
 	}
+	h.mutex.Lock()
+	defer h.mutex.Unlock()
 	h.records = append(h.records, *record)
 	return nil
 }
 
 func (h *LogHandler) WasLogged(pattern string) bool {
 	re, err := regexp.Compile(pattern)
 	RequireImpl(h.t, err)
+	h.mutex.Lock()
+	defer h.mutex.Unlock()
 	for _, record := range h.records {
 		if re.MatchString(record.Msg) {
 			return true