Gossipsub v2.0: Test sending IANNOUNCE out

libp2p · Dec 17, 2024 · 999e46f · 999e46f
1 parent 817a9a2
commit 999e46f
Show file tree

Hide file tree

Showing 2 changed files with 474 additions and 3 deletions.
diff --git a/gossipsub_spam_test.go b/gossipsub_spam_test.go
@@ -32,7 +32,9 @@ func TestGossipsubAttackSpamIWANT(t *testing.T) {
 	attacker := hosts[1]
 
 	// Set up gossipsub on the legit host
-	ps, err := NewGossipSub(ctx, legit)
+	params := DefaultGossipSubParams()
+	params.Dannounce = 0
+	ps, err := NewGossipSub(ctx, legit, WithGossipSubParams(params))
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -149,7 +151,10 @@ func TestGossipsubAttackSpamIHAVE(t *testing.T) {
 	attacker := hosts[1]
 
 	// Set up gossipsub on the legit host
+	params := DefaultGossipSubParams()
+	params.Dannounce = 0
 	ps, err := NewGossipSub(ctx, legit,
+		WithGossipSubParams(params),
 		WithPeerScore(
 			&PeerScoreParams{
 				AppSpecificScore:       func(peer.ID) float64 { return 0 },
@@ -300,7 +305,9 @@ func TestGossipsubAttackGRAFTNonExistentTopic(t *testing.T) {
 	attacker := hosts[1]
 
 	// Set up gossipsub on the legit host
-	ps, err := NewGossipSub(ctx, legit)
+	params := DefaultGossipSubParams()
+	params.Dannounce = 0
+	ps, err := NewGossipSub(ctx, legit, WithGossipSubParams(params))
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -384,7 +391,10 @@ func TestGossipsubAttackGRAFTDuringBackoff(t *testing.T) {
 	attacker := hosts[1]
 
 	// Set up gossipsub on the legit host
+	params := DefaultGossipSubParams()
+	params.Dannounce = 0
 	ps, err := NewGossipSub(ctx, legit,
+		WithGossipSubParams(params),
 		WithPeerScore(
 			&PeerScoreParams{
 				AppSpecificScore:       func(peer.ID) float64 { return 0 },
@@ -665,7 +675,10 @@ func TestGossipsubAttackInvalidMessageSpam(t *testing.T) {
 
 	// Set up gossipsub on the legit host
 	tracer := &gsAttackInvalidMsgTracer{}
+	gossipsubParams := DefaultGossipSubParams()
+	gossipsubParams.Dannounce = 0
 	ps, err := NewGossipSub(ctx, legit,
+		WithGossipSubParams(gossipsubParams),
 		WithEventTracer(tracer),
 		WithPeerScore(params, thresholds),
 	)
@@ -894,7 +907,7 @@ func TestGossipsubAttackSpamIDONTWANT(t *testing.T) {
 type mockGSOnRead func(writeMsg func(*pb.RPC), irpc *pb.RPC)
 
 func newMockGS(ctx context.Context, t *testing.T, attacker host.Host, onReadMsg mockGSOnRead) {
-	newMockGSWithVersion(ctx, t, attacker, protocol.ID("/meshsub/1.2.0"), onReadMsg)
+	newMockGSWithVersion(ctx, t, attacker, protocol.ID("/meshsub/2.0.0"), onReadMsg)
 }
 
 func newMockGSWithVersion(ctx context.Context, t *testing.T, attacker host.Host, gossipSubID protocol.ID, onReadMsg mockGSOnRead) {