micro/selector/random_selector_test.go

33 lines
534 B
Go
Raw Normal View History

2015-12-09 22:23:16 +03:00
package selector
2015-12-09 03:02:45 +03:00
import (
"testing"
2015-12-09 22:23:16 +03:00
"github.com/micro/go-micro/registry/mock"
2015-12-09 03:02:45 +03:00
)
func TestRandomSelector(t *testing.T) {
counts := map[string]int{}
2015-12-09 22:23:16 +03:00
bl := &randomSelector{
so: Options{
Registry: mock.NewRegistry(),
2015-12-09 03:02:45 +03:00
},
}
2015-12-09 22:23:16 +03:00
next, err := bl.Select("foo")
2015-12-09 03:02:45 +03:00
if err != nil {
2015-12-09 05:31:17 +03:00
t.Errorf("Unexpected error calling random select: %v", err)
2015-12-09 03:02:45 +03:00
}
for i := 0; i < 100; i++ {
node, err := next()
if err != nil {
t.Errorf("Expected node err, got err: %v", err)
}
counts[node.Id]++
}
t.Logf("Random Counts %v", counts)
}