small fix, don't worry about it

This commit is contained in:
Tomas Dvorak
2026-04-10 12:02:36 +02:00
parent 08bd0c6e5c
commit 08cb5754f3
638 changed files with 57332 additions and 34706 deletions
@@ -0,0 +1,38 @@
package networking
import "testing"
func TestLoadBalancerRoundRobinSelection(t *testing.T) {
lb := NewLoadBalancer(StrategyRoundRobin)
instances := []*ServiceInstance{
{ID: "a"},
{ID: "b"},
}
first := lb.SelectInstance(instances, "")
second := lb.SelectInstance(instances, "")
third := lb.SelectInstance(instances, "")
if first == nil || second == nil || third == nil {
t.Fatalf("expected non-nil selections")
}
if first.ID != "a" || second.ID != "b" || third.ID != "a" {
t.Fatalf("unexpected round-robin order: %s, %s, %s", first.ID, second.ID, third.ID)
}
}
func TestLoadBalancerLeastConnectionsSelection(t *testing.T) {
lb := NewLoadBalancer(StrategyLeastConnections)
instances := []*ServiceInstance{
{ID: "busy", Metadata: map[string]string{"active_connections": "42"}},
{ID: "idle", Metadata: map[string]string{"active_connections": "1"}},
}
selected := lb.SelectInstance(instances, "")
if selected == nil {
t.Fatalf("expected non-nil instance")
}
if selected.ID != "idle" {
t.Fatalf("expected idle instance, got %s", selected.ID)
}
}