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.
43 lines
673 B
43 lines
673 B
3 years ago
|
package balancer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRoundRobin_NextIndex(t *testing.T) {
|
||
|
|
a := assert.New(t)
|
||
|
|
r := &RoundRobin{}
|
||
|
|
total := 5
|
||
|
|
for i := 1; i < total; i++ {
|
||
|
|
a.Equal(i, r.NextIndex(total))
|
||
|
|
}
|
||
|
|
for i := 0; i < total; i++ {
|
||
|
|
a.Equal(i, r.NextIndex(total))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRoundRobin_NextPeer(t *testing.T) {
|
||
|
|
a := assert.New(t)
|
||
|
|
r := &RoundRobin{}
|
||
|
|
|
||
|
|
// not slice
|
||
|
|
{
|
||
|
|
err, _ := r.NextPeer("s")
|
||
|
|
a.Equal(ErrInputNotSlice, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// no nodes
|
||
|
|
{
|
||
|
|
err, _ := r.NextPeer([]string{})
|
||
|
|
a.Equal(ErrNoAvaliableNode, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// pass
|
||
|
|
{
|
||
|
|
err, res := r.NextPeer([]string{"a"})
|
||
|
|
a.NoError(err)
|
||
|
|
a.Equal("a", res.(string))
|
||
|
|
}
|
||
|
|
}
|