| // 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"}, nil) |
| 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"}, nil) |
| 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{ |
| "/@5@foobar@127.0.0.10@@@@@", |
| "/@5@foobar@127.0.0.11@@@@@", |
| "/@5@tcp4@127.0.0.1@@@@@", |
| "/@5@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{ |
| "/@5@tcp4@127.0.0.1@@@@@", |
| "/@5@tcp4@127.0.0.2@@@@@", |
| "/@5@tcp@127.0.0.3@@@@@", |
| "/@5@tcp@127.0.0.4@@@@@", |
| "/@5@tcp6@127.0.0.7@@@@@", |
| "/@5@tcp6@127.0.0.8@@@@@", |
| "/@5@foobar@127.0.0.10@@@@@", |
| "/@5@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{ |
| "/@5@tcp@127.0.0.3@@@@@", |
| "/@5@tcp@127.0.0.4@@@@@", |
| "/@5@tcp4@127.0.0.1@@@@@", |
| "/@5@tcp4@127.0.0.2@@@@@", |
| "/@5@tcp6@127.0.0.7@@@@@", |
| "/@5@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{ |
| "/@5@tcp@127.0.0.3@@@@@", |
| "/@5@tcp@127.0.0.1@@@@@", |
| "/@5@tcp@74.125.69.139@@@@@", |
| "/@5@tcp@192.168.1.10@@@@@", |
| "/@5@tcp@74.125.142.83@@@@@", |
| "/127.0.0.12:14141", |
| "/@5@foobar@127.0.0.10@@@@@", |
| "/@5@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{ |
| "/@5@tcp@127.0.0.3@@@@@", |
| "/@5@tcp@127.0.0.1@@@@@", |
| "/@5@tcp@74.125.69.139@@@@@", |
| "/@5@tcp@192.168.1.10@@@@@", |
| "/@5@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{ |
| "/@5@ws@127.0.0.3:123@@@@@", |
| "/@5@ws@127.0.0.1@@@@@", |
| "/@5@ws@74.125.69.139@@@@@", |
| "/@5@ws@192.168.1.10@@@@@", |
| "/@5@ws@74.125.142.83@@@@@", |
| "/@5@tcp@127.0.0.3@@@@@", |
| "/@5@tcp@127.0.0.1@@@@@", |
| "/@5@tcp@74.125.69.139@@@@@", |
| "/@5@tcp@192.168.1.10@@@@@", |
| "/@5@tcp@74.125.142.83@@@@@", |
| } |
| if got := servers2names(result); !reflect.DeepEqual(got, want) { |
| t.Errorf("got: %v, want %v", got, want) |
| } |
| } |
