report: Fix tests

INFURA · Dec 18, 2019 · f54db6d · f54db6d
1 parent 428bb2a
commit f54db6d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/report.go b/report.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"io"
+	"sync"
 	"time"
 )
 
@@ -14,6 +15,7 @@ type report struct {
 	// MismatchedResponse is called when a response set does not match across clients
 	MismatchedResponse func([]Response)
 
+	once             sync.Once
 	pendingResponses map[requestID][]Response
 
 	requests   int // Number of requests
@@ -103,10 +105,14 @@ func (r *report) handle(resp Response) error {
 	return nil
 }
 
-func (r *report) Serve(ctx context.Context, respCh <-chan Response) error {
-	if r.pendingResponses == nil {
+func (r *report) init() {
+	r.once.Do(func() {
 		r.pendingResponses = make(map[requestID][]Response)
-	}
+	})
+}
+
+func (r *report) Serve(ctx context.Context, respCh <-chan Response) error {
+	r.init()
 
 	r.started = time.Now()
 	for {

diff --git a/report_test.go b/report_test.go
@@ -14,10 +14,8 @@ func TestReport(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	r, err := Report(clients)
-	if err != nil {
-		t.Fatal(err)
-	}
+	r := report{Clients: clients}
+	r.init()
 
 	r.handle(Response{
 		client: clients[0],