github.com/filecoin-project/bacalhau@v0.3.23-0.20230228154132-45c989550ace/pkg/requester/ranking/utils_test.go (about)

     1  package ranking
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/filecoin-project/bacalhau/pkg/requester"
     7  	"github.com/libp2p/go-libp2p/core/peer"
     8  )
     9  
    10  func assertEquals(t *testing.T, ranks []requester.NodeRank, nodeID string, expectedRank int) {
    11  	for _, rank := range ranks {
    12  		if rank.NodeInfo.PeerInfo.ID == peer.ID(nodeID) {
    13  			if rank.Rank != expectedRank {
    14  				t.Errorf("expected rank %d for node %s, got %d", expectedRank, nodeID, rank.Rank)
    15  			}
    16  			return
    17  		}
    18  	}
    19  	t.Errorf("node %s not found", nodeID)
    20  }