You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
1.6 KiB
71 lines
1.6 KiB
package random
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/go-kratos/kratos/v2/registry"
|
|
"github.com/go-kratos/kratos/v2/selector"
|
|
"github.com/go-kratos/kratos/v2/selector/filter"
|
|
)
|
|
|
|
func TestWrr(t *testing.T) {
|
|
random := New()
|
|
var nodes []selector.Node
|
|
nodes = append(nodes, selector.NewNode(
|
|
"http",
|
|
"127.0.0.1:8080",
|
|
®istry.ServiceInstance{
|
|
ID: "127.0.0.1:8080",
|
|
Version: "v2.0.0",
|
|
Metadata: map[string]string{"weight": "10"},
|
|
}))
|
|
nodes = append(nodes, selector.NewNode(
|
|
"http",
|
|
"127.0.0.1:9090",
|
|
®istry.ServiceInstance{
|
|
ID: "127.0.0.1:9090",
|
|
Version: "v2.0.0",
|
|
Metadata: map[string]string{"weight": "20"},
|
|
}))
|
|
random.Apply(nodes)
|
|
var count1, count2 int
|
|
for i := 0; i < 200; i++ {
|
|
n, done, err := random.Select(context.Background(), selector.WithNodeFilter(filter.Version("v2.0.0")))
|
|
if err != nil {
|
|
t.Errorf("expect no error, got %v", err)
|
|
}
|
|
if done == nil {
|
|
t.Errorf("expect not nil, got:%v", done)
|
|
}
|
|
if n == nil {
|
|
t.Errorf("expect not nil, got:%v", n)
|
|
}
|
|
done(context.Background(), selector.DoneInfo{})
|
|
if n.Address() == "127.0.0.1:8080" {
|
|
count1++
|
|
} else if n.Address() == "127.0.0.1:9090" {
|
|
count2++
|
|
}
|
|
}
|
|
if count1 <= 80 {
|
|
t.Errorf("count1(%v) <= 80", count1)
|
|
}
|
|
if count1 >= 120 {
|
|
t.Errorf("count1(%v) >= 120", count1)
|
|
}
|
|
if count2 <= 80 {
|
|
t.Errorf("count2(%v) <= 80", count2)
|
|
}
|
|
if count2 >= 120 {
|
|
t.Errorf("count2(%v) >= 120", count2)
|
|
}
|
|
}
|
|
|
|
func TestEmpty(t *testing.T) {
|
|
b := &Balancer{}
|
|
_, _, err := b.Pick(context.Background(), []selector.WeightedNode{})
|
|
if err == nil {
|
|
t.Errorf("expect nil, got %v", err)
|
|
}
|
|
}
|
|
|