Merge pull request #17 from libp2p/fix/issue-16

Pacify the race detector
libp2p · Mar 5, 2019 · 6d26667 · 6d26667
2 parents fb9be13 + 5ab68af
commit 6d26667
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/p2p/host/autonat/autonat.go b/p2p/host/autonat/autonat.go
@@ -85,6 +85,8 @@ func NewAutoNAT(ctx context.Context, h host.Host, getAddrs GetAddrs) AutoNAT {
 }
 
 func (as *AmbientAutoNAT) Status() NATStatus {
+	as.mx.Lock()
+	defer as.mx.Unlock()
 	return as.status
 }
 

diff --git a/p2p/host/autonat/autonat_test.go b/p2p/host/autonat/autonat_test.go
@@ -73,8 +73,9 @@ func newDialResponseError(status pb.Message_ResponseStatus, text string) *pb.Mes
 func makeAutoNAT(ctx context.Context, t *testing.T, ash host.Host) (host.Host, AutoNAT) {
 	h := bhost.NewBlankHost(swarmt.GenSwarm(t, ctx))
 	a := NewAutoNAT(ctx, h, nil)
+	a.(*AmbientAutoNAT).mx.Lock()
 	a.(*AmbientAutoNAT).peers[ash.ID()] = ash.Addrs()
-
+	a.(*AmbientAutoNAT).mx.Unlock()
 	return h, a
 }