swarm/storage: fix mockNetFetcher data races (ethereum#18462)

fixes: ethersphere/swarm#1117
FinTechToken · Jan 16, 2019 · f728837 · f728837
1 parent 96c7c18
commit f728837
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/swarm/storage/netstore_test.go b/swarm/storage/netstore_test.go
@@ -43,6 +43,7 @@ type mockNetFetcher struct {
 	quit            <-chan struct{}
 	ctx             context.Context
 	hopCounts       []uint8
+	mu              sync.Mutex
 }
 
 func (m *mockNetFetcher) Offer(ctx context.Context, source *enode.ID) {
@@ -51,6 +52,9 @@ func (m *mockNetFetcher) Offer(ctx context.Context, source *enode.ID) {
 }
 
 func (m *mockNetFetcher) Request(ctx context.Context, hopCount uint8) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
 	m.requestCalled = true
 	var peers []Address
 	m.peers.Range(func(key interface{}, _ interface{}) bool {