mirror of
				https://source.quilibrium.com/quilibrium/ceremonyclient.git
				synced 2025-11-04 09:37:44 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			64 lines
		
	
	
		
			1.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			64 lines
		
	
	
		
			1.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package swarm_test
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"testing"
 | 
						|
	"time"
 | 
						|
 | 
						|
	"github.com/libp2p/go-libp2p/core/peer"
 | 
						|
	"github.com/libp2p/go-libp2p/core/peerstore"
 | 
						|
	. "github.com/libp2p/go-libp2p/p2p/net/swarm"
 | 
						|
 | 
						|
	ma "github.com/multiformats/go-multiaddr"
 | 
						|
	"github.com/stretchr/testify/require"
 | 
						|
)
 | 
						|
 | 
						|
func TestPeers(t *testing.T) {
 | 
						|
	ctx := context.Background()
 | 
						|
	swarms := makeSwarms(t, 2)
 | 
						|
	s1 := swarms[0]
 | 
						|
	s2 := swarms[1]
 | 
						|
 | 
						|
	connect := func(s *Swarm, dst peer.ID, addr ma.Multiaddr) {
 | 
						|
		// TODO: make a DialAddr func.
 | 
						|
		s.Peerstore().AddAddr(dst, addr, peerstore.PermanentAddrTTL)
 | 
						|
		// t.Logf("connections from %s", s.LocalPeer())
 | 
						|
		// for _, c := range s.ConnsToPeer(dst) {
 | 
						|
		// 	t.Logf("connection from %s to %s: %v", s.LocalPeer(), dst, c)
 | 
						|
		// }
 | 
						|
		// t.Logf("")
 | 
						|
		if _, err := s.DialPeer(ctx, dst); err != nil {
 | 
						|
			t.Fatal("error swarm dialing to peer", err)
 | 
						|
		}
 | 
						|
		// t.Log(s.swarm.Dump())
 | 
						|
	}
 | 
						|
 | 
						|
	connect(s1, s2.LocalPeer(), s2.ListenAddresses()[0])
 | 
						|
	require.Eventually(t, func() bool { return len(s2.Peers()) > 0 }, 3*time.Second, 50*time.Millisecond)
 | 
						|
	connect(s2, s1.LocalPeer(), s1.ListenAddresses()[0])
 | 
						|
 | 
						|
	for i := 0; i < 100; i++ {
 | 
						|
		connect(s1, s2.LocalPeer(), s2.ListenAddresses()[0])
 | 
						|
		connect(s2, s1.LocalPeer(), s1.ListenAddresses()[0])
 | 
						|
	}
 | 
						|
 | 
						|
	for _, s := range swarms {
 | 
						|
		log.Infof("%s swarm routing table: %s", s.LocalPeer(), s.Peers())
 | 
						|
	}
 | 
						|
 | 
						|
	test := func(s *Swarm) {
 | 
						|
		expect := 1
 | 
						|
		actual := len(s.Peers())
 | 
						|
		if actual != expect {
 | 
						|
			t.Errorf("%s has %d peers, not %d: %v", s.LocalPeer(), actual, expect, s.Peers())
 | 
						|
		}
 | 
						|
		actual = len(s.Conns())
 | 
						|
		if actual != expect {
 | 
						|
			t.Errorf("%s has %d conns, not %d: %v", s.LocalPeer(), actual, expect, s.Conns())
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	test(s1)
 | 
						|
	test(s2)
 | 
						|
}
 |