github.com/annwntech/go-micro/v2@v2.9.5/client/selector/default_test.go (about)

     1  package selector
     2  
     3  import (
     4  	"os"
     5  	"testing"
     6  
     7  	"github.com/annwntech/go-micro/v2/registry/memory"
     8  )
     9  
    10  func TestRegistrySelector(t *testing.T) {
    11  	counts := map[string]int{}
    12  
    13  	r := memory.NewRegistry(memory.Services(testData))
    14  	cache := NewSelector(Registry(r))
    15  
    16  	next, err := cache.Select("foo")
    17  	if err != nil {
    18  		t.Errorf("Unexpected error calling cache select: %v", err)
    19  	}
    20  
    21  	for i := 0; i < 100; i++ {
    22  		node, err := next()
    23  		if err != nil {
    24  			t.Errorf("Expected node err, got err: %v", err)
    25  		}
    26  		counts[node.Id]++
    27  	}
    28  
    29  	if len(os.Getenv("IN_TRAVIS_CI")) == 0 {
    30  		t.Logf("Selector Counts %v", counts)
    31  	}
    32  }