move selector

selector/default_test.go

@@ -0,0 +1,32 @@
+package selector
+
+import (
+	"os"
+	"testing"
+
+	"github.com/micro/go-micro/v2/registry/memory"
+)
+
+func TestRegistrySelector(t *testing.T) {
+	counts := map[string]int{}
+
+	r := memory.NewRegistry(memory.Services(testData))
+	cache := NewSelector(Registry(r))
+
+	next, err := cache.Select("foo")
+	if err != nil {
+		t.Errorf("Unexpected error calling cache select: %v", err)
+	}
+
+	for i := 0; i < 100; i++ {
+		node, err := next()
+		if err != nil {
+			t.Errorf("Expected node err, got err: %v", err)
+		}
+		counts[node.Id]++
+	}
+
+	if len(os.Getenv("IN_TRAVIS_CI")) == 0 {
+		t.Logf("Selector Counts %v", counts)
+	}
+}