60 lines
1.0 KiB
Go
60 lines
1.0 KiB
Go
|
package selector
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/micro/go-micro/registry"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestStrategies(t *testing.T) {
|
||
|
|
testData := []*registry.Service{
|
||
|
|
®istry.Service{
|
||
|
|
Name: "test1",
|
||
|
|
Version: "latest",
|
||
|
|
Nodes: []*registry.Node{
|
||
|
|
®istry.Node{
|
||
|
|
Id: "test1-1",
|
||
|
|
Address: "10.0.0.1",
|
||
|
|
Port: 1001,
|
||
|
|
},
|
||
|
|
®istry.Node{
|
||
|
|
Id: "test1-2",
|
||
|
|
Address: "10.0.0.2",
|
||
|
|
Port: 1002,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
®istry.Service{
|
||
|
|
Name: "test1",
|
||
|
|
Version: "default",
|
||
|
|
Nodes: []*registry.Node{
|
||
|
|
®istry.Node{
|
||
|
|
Id: "test1-3",
|
||
|
|
Address: "10.0.0.3",
|
||
|
|
Port: 1003,
|
||
|
|
},
|
||
|
|
®istry.Node{
|
||
|
|
Id: "test1-4",
|
||
|
|
Address: "10.0.0.4",
|
||
|
|
Port: 1004,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for name, strategy := range map[string]Strategy{"random": Random, "roundrobin": RoundRobin} {
|
||
|
|
next := strategy(testData)
|
||
|
|
counts := make(map[string]int)
|
||
|
|
|
||
|
|
for i := 0; i < 100; i++ {
|
||
|
|
node, err := next()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
counts[node.Id]++
|
||
|
|
}
|
||
|
|
|
||
|
|
t.Logf("%s: %+v\n", name, counts)
|
||
|
|
}
|
||
|
|
}
|