github.com/qri-io/qri@v0.10.1-0.20220104210721-c771715036cb/p2p/list_peers_test.go (about)

     1  package p2p
     2  
     3  import (
     4  	"context"
     5  	"testing"
     6  
     7  	p2ptest "github.com/qri-io/qri/p2p/test"
     8  	"github.com/qri-io/qri/profile"
     9  )
    10  
    11  func TestListPeers(t *testing.T) {
    12  	t.Skip("ramfox: flakey test until p2p test network setup is refactored")
    13  	ctx := context.Background()
    14  	factory := p2ptest.NewTestNodeFactory(NewTestableQriNode)
    15  	testPeers, err := p2ptest.NewTestNetwork(ctx, factory, 6)
    16  	if err != nil {
    17  		t.Fatalf("error creating network: %s", err.Error())
    18  	}
    19  	if err := p2ptest.ConnectNodes(ctx, testPeers); err != nil {
    20  		t.Fatalf("error connecting peers: %s", err.Error())
    21  	}
    22  
    23  	userID := profile.IDFromPeerID(testPeers[0].SimpleAddrInfo().ID)
    24  	peers, err := ListPeers(ctx, testPeers[0].(*QriNode), userID, 2, 3, false)
    25  	if err != nil {
    26  		t.Error(err.Error())
    27  	}
    28  	if len(peers) != 3 {
    29  		t.Errorf("expected 3 peers, got: %d", len(peers))
    30  	}
    31  }