gitee.com/liuxuezhan/go-micro-v1.18.0@v1.0.0/client/selector/default_test.go (about)

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