gossipsub

comm.go

@@ -104,3 +104,21 @@ func rpcWithSubs(subs ...*pb.RPC_SubOpts) *RPC {
 func rpcWithMessages(msgs ...*pb.Message) *RPC {
 	return &RPC{RPC: pb.RPC{Publish: msgs}}
 }
+
+func rpcWithControl(msgs []*pb.Message,
+	ihave []*pb.ControlIHave,
+	iwant []*pb.ControlIWant,
+	graft []*pb.ControlGraft,
+	prune []*pb.ControlPrune) *RPC {
+	return &RPC{
+		RPC: pb.RPC{
+			Publish: msgs,
+			Control: &pb.ControlMessage{
+				Ihave: ihave,
+				Iwant: iwant,
+				Graft: graft,
+				Prune: prune,
+			},
+		},
+	}
+}

floodsub.go

@@ -78,3 +78,7 @@ func (fs *FloodSubRouter) Publish(from peer.ID, msg *pb.Message) {
 		}
 	}
 }
+
+func (fs *FloodSubRouter) Join(topic string) {}
+
+func (fs *FloodSubRouter) Leave(topic string) {}

floodsub_test.go

@@ -55,8 +55,16 @@ func connect(t *testing.T, a, b host.Host) {
 }
 
 func sparseConnect(t *testing.T, hosts []host.Host) {
+	connectSome(t, hosts, 3)
+}
+
+func denseConnect(t *testing.T, hosts []host.Host) {
+	connectSome(t, hosts, 10)
+}
+
+func connectSome(t *testing.T, hosts []host.Host, d int) {
 	for i, a := range hosts {
-		for j := 0; j < 3; j++ {
+		for j := 0; j < d; j++ {
 			n := rand.Intn(len(hosts))
 			if n == i {
 				j--