micro/selector/roundrobin/roundrobin_test.go

package roundrobin

import (
	"testing"

	"go.unistack.org/micro/v3/selector"
)

func TestRoundRobin(t *testing.T) {
	selector.Tests(t, NewSelector())

	r1 := "127.0.0.1:8000"
	r2 := "127.0.0.1:8001"
	r3 := "127.0.0.1:8002"

	sel := NewSelector()

	// By passing r1 and r2 first, it forces a set sequence of (r1 => r2 => r3 => r1)

	next, err := sel.Select([]string{r1})
	if err != nil {
		t.Fatal(err)
	}
	r := next()

	if r1 != r {
		t.Fatal("Expected route to be r == r1")
	}

	next, err = sel.Select([]string{r2})
	if err != nil {
		t.Fatal(err)
	}
	r = next()
	if r2 != r {
		t.Fatal("Expected route to be r2")
	}

	routes := []string{r1, r2, r3}
	next, err = sel.Select(routes)
	if err != nil {
		t.Fatal(err)
	}
	n1, n2, n3, n4 := next(), next(), next(), next()

	// start element is random but then it should loop through in order
	start := -1
	for i := 0; i < 3; i++ {
		if n1 == routes[i] {
			start = i
			break
		}
	}
	if start == -1 {
		t.Fatalf("start == -1 %v %v", start, -1)
	}
	if routes[start] != n1 {
		t.Fatal("Unexpected route")
	}
	if routes[(start+1)%3] != n2 {
		t.Fatal("Unexpected route")
	}
	if routes[(start+2)%3] != n3 {
		t.Fatal("Unexpected route")
	}
	if routes[(start+3)%3] != n4 {
		t.Fatal("Unexpected route")
	}
}