blob: 59edbcd7c47872201d57ed4897180832d39ed285 [file] [log] [blame]
// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package rpc
import (
"net"
"reflect"
"strings"
"testing"
"v.io/v23/naming"
)
func servers2names(servers []naming.MountedServer) []string {
e := naming.MountEntry{Servers: servers}
return e.Names()
}
func TestIncompatible(t *testing.T) {
servers := []naming.MountedServer{}
_, err := filterAndOrderServers(servers, []string{"tcp"})
if err == nil || err.Error() != "failed to find any compatible servers" {
t.Errorf("expected a different error: %v", err)
}
for _, a := range []string{"127.0.0.3", "127.0.0.4"} {
name := naming.JoinAddressName(naming.FormatEndpoint("tcp", a), "")
servers = append(servers, naming.MountedServer{Server: name})
}
_, err = filterAndOrderServers(servers, []string{"foobar"})
if err == nil || !strings.HasSuffix(err.Error(), "undesired protocol: tcp]") {
t.Errorf("expected a different error to: %v", err)
}
}
func TestOrderingByProtocol(t *testing.T) {
servers := []naming.MountedServer{}
_, ipnet, _ := net.ParseCIDR("127.0.0.0/8")
ipnets := []*net.IPNet{ipnet}
for _, a := range []string{"127.0.0.3", "127.0.0.4"} {
name := naming.JoinAddressName(naming.FormatEndpoint("tcp", a), "")
servers = append(servers, naming.MountedServer{Server: name})
}
for _, a := range []string{"127.0.0.1", "127.0.0.2"} {
name := naming.JoinAddressName(naming.FormatEndpoint("tcp4", a), "")
servers = append(servers, naming.MountedServer{Server: name})
}
for _, a := range []string{"127.0.0.10", "127.0.0.11"} {
name := naming.JoinAddressName(naming.FormatEndpoint("foobar", a), "")
servers = append(servers, naming.MountedServer{Server: name})
}
for _, a := range []string{"127.0.0.7", "127.0.0.8"} {
name := naming.JoinAddressName(naming.FormatEndpoint("tcp6", a), "")
servers = append(servers, naming.MountedServer{Server: name})
}
if _, err := filterAndOrderServers(servers, []string{"batman"}, ipnets...); err == nil {
t.Fatalf("expected an error")
}
// Add a server with naming.UnknownProtocol. This typically happens
// when the endpoint is in <host>:<port> format. Currently, the sorting
// is setup to always allow UnknownProtocol, but put it in the end.
// May want to revisit this choice, but for now the test captures what
// the current state of the code intends.
servers = append(servers, naming.MountedServer{Server: "127.0.0.12:14141"})
// Just foobar and tcp4
want := []string{
"/@6@foobar@127.0.0.10@@@@@@",
"/@6@foobar@127.0.0.11@@@@@@",
"/@6@tcp4@127.0.0.1@@@@@@",
"/@6@tcp4@127.0.0.2@@@@@@",
"/127.0.0.12:14141",
}
result, err := filterAndOrderServers(servers, []string{"foobar", "tcp4"}, ipnets...)
if err != nil {
t.Fatalf("unexpected error: %s", err)
}
if got := servers2names(result); !reflect.DeepEqual(got, want) {
t.Errorf("got: %v, want %v", got, want)
}
// Everything, since we didn't specify a protocol, but ordered by
// the internal metric - see defaultPreferredProtocolOrder.
// The order will be the default preferred order for protocols, the
// original ordering within each protocol, with protocols that
// are not in the default ordering list at the end.
want = []string{
"/@6@tcp4@127.0.0.1@@@@@@",
"/@6@tcp4@127.0.0.2@@@@@@",
"/@6@tcp@127.0.0.3@@@@@@",
"/@6@tcp@127.0.0.4@@@@@@",
"/@6@tcp6@127.0.0.7@@@@@@",
"/@6@tcp6@127.0.0.8@@@@@@",
"/@6@foobar@127.0.0.10@@@@@@",
"/@6@foobar@127.0.0.11@@@@@@",
"/127.0.0.12:14141",
}
if result, err = filterAndOrderServers(servers, nil, ipnets...); err != nil {
t.Fatalf("unexpected error: %s", err)
}
if got := servers2names(result); !reflect.DeepEqual(got, want) {
t.Errorf("got: %v, want %v", got, want)
}
if result, err = filterAndOrderServers(servers, []string{}, ipnets...); err != nil {
t.Fatalf("unexpected error: %s", err)
}
if got := servers2names(result); !reflect.DeepEqual(got, want) {
t.Errorf("got: %v, want %v", got, want)
}
// Just "tcp" implies tcp4 and tcp6 as well.
want = []string{
"/@6@tcp@127.0.0.3@@@@@@",
"/@6@tcp@127.0.0.4@@@@@@",
"/@6@tcp4@127.0.0.1@@@@@@",
"/@6@tcp4@127.0.0.2@@@@@@",
"/@6@tcp6@127.0.0.7@@@@@@",
"/@6@tcp6@127.0.0.8@@@@@@",
"/127.0.0.12:14141",
}
if result, err = filterAndOrderServers(servers, []string{"tcp"}, ipnets...); err != nil {
t.Fatalf("unexpected error: %s", err)
}
if got := servers2names(result); !reflect.DeepEqual(got, want) {
t.Errorf("got: %v, want %v", got, want)
}
// Ask for all protocols, with no ordering, except for locality
want = []string{
"/@6@tcp@127.0.0.3@@@@@@",
"/@6@tcp@127.0.0.1@@@@@@",
"/@6@tcp@74.125.69.139@@@@@@",
"/@6@tcp@192.168.1.10@@@@@@",
"/@6@tcp@74.125.142.83@@@@@@",
"/127.0.0.12:14141",
"/@6@foobar@127.0.0.10@@@@@@",
"/@6@foobar@127.0.0.11@@@@@@",
}
servers = []naming.MountedServer{}
// naming.UnknownProtocol
servers = append(servers, naming.MountedServer{Server: "127.0.0.12:14141"})
for _, a := range []string{"74.125.69.139", "127.0.0.3", "127.0.0.1", "192.168.1.10", "74.125.142.83"} {
name := naming.JoinAddressName(naming.FormatEndpoint("tcp", a), "")
servers = append(servers, naming.MountedServer{Server: name})
}
for _, a := range []string{"127.0.0.10", "127.0.0.11"} {
name := naming.JoinAddressName(naming.FormatEndpoint("foobar", a), "")
servers = append(servers, naming.MountedServer{Server: name})
}
if result, err = filterAndOrderServers(servers, []string{}, ipnets...); err != nil {
t.Fatalf("unexpected error: %s", err)
}
if got := servers2names(result); !reflect.DeepEqual(got, want) {
t.Errorf("got: %v, want %v", got, want)
}
}
func TestOrderingByLocality(t *testing.T) {
servers := []naming.MountedServer{}
_, ipnet, _ := net.ParseCIDR("127.0.0.0/8")
ipnets := []*net.IPNet{ipnet}
for _, a := range []string{"74.125.69.139", "127.0.0.3", "127.0.0.1", "192.168.1.10", "74.125.142.83"} {
name := naming.JoinAddressName(naming.FormatEndpoint("tcp", a), "")
servers = append(servers, naming.MountedServer{Server: name})
}
result, err := filterAndOrderServers(servers, []string{"tcp"}, ipnets...)
if err != nil {
t.Fatalf("unexpected error: %s", err)
}
want := []string{
"/@6@tcp@127.0.0.3@@@@@@",
"/@6@tcp@127.0.0.1@@@@@@",
"/@6@tcp@74.125.69.139@@@@@@",
"/@6@tcp@192.168.1.10@@@@@@",
"/@6@tcp@74.125.142.83@@@@@@",
}
if got := servers2names(result); !reflect.DeepEqual(got, want) {
t.Errorf("got: %v, want %v", got, want)
}
for _, a := range []string{"74.125.69.139", "127.0.0.3:123", "127.0.0.1", "192.168.1.10", "74.125.142.83"} {
name := naming.JoinAddressName(naming.FormatEndpoint("ws", a), "")
servers = append(servers, naming.MountedServer{Server: name})
}
if result, err = filterAndOrderServers(servers, []string{"ws", "tcp"}, ipnets...); err != nil {
t.Fatalf("unexpected error: %s", err)
}
want = []string{
"/@6@ws@127.0.0.3:123@@@@@@",
"/@6@ws@127.0.0.1@@@@@@",
"/@6@ws@74.125.69.139@@@@@@",
"/@6@ws@192.168.1.10@@@@@@",
"/@6@ws@74.125.142.83@@@@@@",
"/@6@tcp@127.0.0.3@@@@@@",
"/@6@tcp@127.0.0.1@@@@@@",
"/@6@tcp@74.125.69.139@@@@@@",
"/@6@tcp@192.168.1.10@@@@@@",
"/@6@tcp@74.125.142.83@@@@@@",
}
if got := servers2names(result); !reflect.DeepEqual(got, want) {
t.Errorf("got: %v, want %v", got, want)
}
}