From 57c688ecdc6b2f2b3c5e8188bdfda24a0bd8647b Mon Sep 17 00:00:00 2001
From: sukun <sukunrt@gmail.com>
Date: Thu, 2 May 2024 18:09:20 +0530
Subject: [PATCH] connectedness event deadlock test

---
 p2p/net/swarm/swarm.go            |  2 ++
 p2p/net/swarm/swarm_event_test.go | 45 +++++++++++++++++++++++++++++++
 2 files changed, 47 insertions(+)

diff --git a/p2p/net/swarm/swarm.go b/p2p/net/swarm/swarm.go
index e3dd183c60..5a77941a63 100644
--- a/p2p/net/swarm/swarm.go
+++ b/p2p/net/swarm/swarm.go
@@ -807,10 +807,12 @@ func (s *Swarm) removeConn(c *Conn) {
 	s.conns.Unlock()
 
 	if oldState != newState {
+		fmt.Println("going to emit event", newState)
 		s.emitter.Emit(event.EvtPeerConnectednessChanged{
 			Peer:          p,
 			Connectedness: newState,
 		})
+		fmt.Println("emitted event", newState)
 	}
 }
 
diff --git a/p2p/net/swarm/swarm_event_test.go b/p2p/net/swarm/swarm_event_test.go
index 8d2b2d79ce..fe9067fda9 100644
--- a/p2p/net/swarm/swarm_event_test.go
+++ b/p2p/net/swarm/swarm_event_test.go
@@ -2,6 +2,7 @@ package swarm_test
 
 import (
 	"context"
+	"fmt"
 	"testing"
 	"time"
 
@@ -12,6 +13,7 @@ import (
 	swarmt "github.com/libp2p/go-libp2p/p2p/net/swarm/testing"
 
 	ma "github.com/multiformats/go-multiaddr"
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
 
@@ -201,3 +203,46 @@ func TestConnectednessEvents(t *testing.T) {
 		t.Fatal("expected all disconnected events after swarm close to be completed")
 	}
 }
+
+func TestConnectednessEventDeadlock(t *testing.T) {
+	s1, sub1 := newSwarmWithSubscription(t)
+	const N = 100
+	peers := make([]*Swarm, N)
+	for i := 0; i < N; i++ {
+		peers[i] = swarmt.GenSwarm(t)
+	}
+
+	// First check all connected events
+	done := make(chan struct{})
+	go func() {
+		defer close(done)
+		count := 0
+		for count < N {
+			e := <-sub1.Out()
+			// sleep to simulate a slow consumer
+			evt, ok := e.(event.EvtPeerConnectednessChanged)
+			if !ok {
+				t.Error("invalid event received", e)
+				return
+			}
+			if evt.Connectedness != network.Connected {
+				continue
+			}
+			count++
+			fmt.Println(count)
+			s1.ClosePeer(evt.Peer)
+		}
+	}()
+	for i := 0; i < N; i++ {
+		s1.Peerstore().AddAddrs(peers[i].LocalPeer(), []ma.Multiaddr{peers[i].ListenAddresses()[0]}, time.Hour)
+		go func(i int) {
+			_, err := s1.DialPeer(context.Background(), peers[i].LocalPeer())
+			assert.NoError(t, err)
+		}(i)
+	}
+	select {
+	case <-done:
+	case <-time.After(100 * time.Second):
+		t.Fatal("expected all connectedness events to be completed")
+	}
+}