| // 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 naming |
| |
| import ( |
| "net" |
| "reflect" |
| "testing" |
| |
| "v.io/v23/naming" |
| "v.io/v23/rpc/version" |
| ) |
| |
| func TestEndpoint(t *testing.T) { |
| defver := defaultVersion |
| defer func() { |
| defaultVersion = defver |
| }() |
| v1 := &Endpoint{ |
| Protocol: naming.UnknownProtocol, |
| Address: "batman.com:1234", |
| RID: naming.FixedRoutingID(0xdabbad00), |
| IsMountTable: true, |
| } |
| v2 := &Endpoint{ |
| Protocol: naming.UnknownProtocol, |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0xdabbad00), |
| MinRPCVersion: 1, |
| MaxRPCVersion: 10, |
| IsMountTable: true, |
| } |
| v2hp := &Endpoint{ |
| Protocol: naming.UnknownProtocol, |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0x0), |
| MinRPCVersion: version.DeprecatedRPCVersion, |
| MaxRPCVersion: version.DeprecatedRPCVersion, |
| IsMountTable: true, |
| } |
| v3s := &Endpoint{ |
| Protocol: naming.UnknownProtocol, |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0x0), |
| MinRPCVersion: 2, |
| MaxRPCVersion: 3, |
| IsMountTable: false, |
| } |
| v3m := &Endpoint{ |
| Protocol: naming.UnknownProtocol, |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0xdabbad00), |
| MinRPCVersion: 2, |
| MaxRPCVersion: 3, |
| IsMountTable: true, |
| } |
| v3tcp := &Endpoint{ |
| Protocol: "tcp", |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0x0), |
| MinRPCVersion: 2, |
| MaxRPCVersion: 3, |
| IsMountTable: false, |
| } |
| v3ws6 := &Endpoint{ |
| Protocol: "ws6", |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0x0), |
| MinRPCVersion: 2, |
| MaxRPCVersion: 3, |
| IsMountTable: false, |
| } |
| v4 := &Endpoint{ |
| Protocol: "tcp", |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0xba77), |
| MinRPCVersion: 4, |
| MaxRPCVersion: 5, |
| IsMountTable: true, |
| Blessings: []string{"dev.v.io/foo@bar.com", "dev.v.io/bar@bar.com/delegate"}, |
| } |
| v4b := &Endpoint{ |
| Protocol: "tcp", |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0xba77), |
| MinRPCVersion: 4, |
| MaxRPCVersion: 5, |
| IsMountTable: true, |
| // Blessings that look similar to other parts of the endpoint. |
| Blessings: []string{"@@", "@s", "@m"}, |
| } |
| v5 := &Endpoint{ |
| Protocol: "tcp", |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0xba77), |
| IsMountTable: true, |
| Blessings: []string{"dev.v.io/foo@bar.com", "dev.v.io/bar@bar.com/delegate"}, |
| MinRPCVersion: version.DeprecatedRPCVersion, |
| MaxRPCVersion: version.DeprecatedRPCVersion, |
| } |
| v5b := &Endpoint{ |
| Protocol: "tcp", |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0xba77), |
| IsMountTable: true, |
| // Blessings that look similar to other parts of the endpoint. |
| Blessings: []string{"@@", "@s", "@m"}, |
| MinRPCVersion: version.DeprecatedRPCVersion, |
| MaxRPCVersion: version.DeprecatedRPCVersion, |
| } |
| v5c := &Endpoint{ |
| Protocol: "tcp", |
| Address: "batman.com:2345", |
| RID: naming.FixedRoutingID(0xba77), |
| IsMountTable: false, |
| Blessings: []string{"dev.v.io/foo@bar.com", "dev.v.io/bar@bar.com/delegate"}, |
| MinRPCVersion: version.DeprecatedRPCVersion, |
| MaxRPCVersion: version.DeprecatedRPCVersion, |
| } |
| |
| testcasesA := []struct { |
| endpoint naming.Endpoint |
| address string |
| }{ |
| {v1, "batman.com:1234"}, |
| {v2, "batman.com:2345"}, |
| {v2hp, "batman.com:2345"}, |
| } |
| for _, test := range testcasesA { |
| addr := test.endpoint.Addr() |
| if addr.String() != test.address { |
| t.Errorf("unexpected address %q, not %q", addr.String(), test.address) |
| } |
| } |
| |
| // Test v3, v4, v5 endpoints. |
| testcasesC := []struct { |
| Endpoint naming.Endpoint |
| String string |
| Version int |
| }{ |
| {v3s, "@3@@batman.com:2345@00000000000000000000000000000000@2@3@s@@", 3}, |
| {v3m, "@3@@batman.com:2345@000000000000000000000000dabbad00@2@3@m@@", 3}, |
| {v3tcp, "@3@tcp@batman.com:2345@00000000000000000000000000000000@2@3@s@@", 3}, |
| {v3ws6, "@3@ws6@batman.com:2345@00000000000000000000000000000000@2@3@s@@", 3}, |
| {v3s, "@4@@batman.com:2345@00000000000000000000000000000000@2@3@s@@@", 4}, |
| {v4, "@4@tcp@batman.com:2345@0000000000000000000000000000ba77@4@5@m@dev.v.io/foo@bar.com,dev.v.io/bar@bar.com/delegate@@", 4}, |
| {v4b, "@4@tcp@batman.com:2345@0000000000000000000000000000ba77@4@5@m@@@,@s,@m@@", 4}, |
| {v5, "@5@tcp@batman.com:2345@0000000000000000000000000000ba77@m@dev.v.io/foo@bar.com,dev.v.io/bar@bar.com/delegate@@", 5}, |
| {v5b, "@5@tcp@batman.com:2345@0000000000000000000000000000ba77@m@@@,@s,@m@@", 5}, |
| {v5c, "@5@tcp@batman.com:2345@0000000000000000000000000000ba77@s@dev.v.io/foo@bar.com,dev.v.io/bar@bar.com/delegate@@", 5}, |
| } |
| |
| for _, test := range testcasesC { |
| if got, want := test.Endpoint.VersionedString(test.Version), test.String; got != want { |
| t.Errorf("Got %q want %q for endpoint (v%d): %#v", got, want, test.Version, test.Endpoint) |
| } |
| ep, err := NewEndpoint(test.String) |
| if err != nil { |
| t.Errorf("Endpoint(%q) failed with %v", test.String, err) |
| continue |
| } |
| if !reflect.DeepEqual(ep, test.Endpoint) { |
| t.Errorf("Got endpoint %#v, want %#v for string %q", ep, test.Endpoint, test.String) |
| } |
| } |
| |
| // Make sure we can continue to parse and create v2 endpoints. |
| defaultVersion = 2 |
| testcasesB := []struct { |
| Endpoint naming.Endpoint |
| String string |
| Input string |
| min, max version.RPCVersion |
| servesMT bool |
| }{ |
| {v1, "@2@@batman.com:1234@000000000000000000000000dabbad00@@@@", "", version.UnknownRPCVersion, version.UnknownRPCVersion, true}, |
| {v2, "@2@@batman.com:2345@000000000000000000000000dabbad00@1@10@@", "", 1, 10, true}, |
| {v2hp, "@2@@batman.com:2345@00000000000000000000000000000000@1@1@@", "batman.com:2345", 2, 3, true}, |
| } |
| |
| for _, test := range testcasesB { |
| if got, want := test.Endpoint.String(), test.String; got != want { |
| t.Errorf("Got %q want %q for endpoint %T = %#v", got, want, test.Endpoint, test.Endpoint) |
| } |
| str := test.Input |
| var ep naming.Endpoint |
| var err error |
| if str == "" { |
| str = test.String |
| ep, err = NewEndpoint(str) |
| } else { |
| ep, err = NewEndpoint(naming.FormatEndpoint(naming.UnknownProtocol, str, |
| version.RPCVersionRange{test.min, test.max}, |
| naming.ServesMountTable(test.servesMT))) |
| } |
| if err != nil { |
| t.Errorf("Endpoint(%q) failed with %v", str, err) |
| continue |
| } |
| if !reflect.DeepEqual(ep, test.Endpoint) { |
| t.Errorf("Got endpoint %T = %#v, want %T = %#v for string %q", ep, ep, test.Endpoint, test.Endpoint, str) |
| } |
| } |
| } |
| |
| type endpointTest struct { |
| input, output string |
| err error |
| } |
| |
| func TestEndpointDefaults(t *testing.T) { |
| testcases := []endpointTest{ |
| {"@1@tcp@batman@@@", "@5@tcp@batman@00000000000000000000000000000000@m@@@", nil}, |
| {"@2@tcp@robin@@@@@", "@5@tcp@robin@00000000000000000000000000000000@m@@@", nil}, |
| {"@1@@@@@", "@5@@:0@00000000000000000000000000000000@m@@@", nil}, |
| {"@2@@@@@@@", "@5@@:0@00000000000000000000000000000000@m@@@", nil}, |
| {"@1@tcp@batman:12@@@", "@5@tcp@batman:12@00000000000000000000000000000000@m@@@", nil}, |
| {"@2@tcp@foo:12@@9@@@", "@5@tcp@foo:12@00000000000000000000000000000000@m@@@", nil}, |
| {"@2@tcp@foo:12@@@4@@", "@5@tcp@foo:12@00000000000000000000000000000000@m@@@", nil}, |
| {"@2@tcp@foo:12@@2@4@@", "@5@tcp@foo:12@00000000000000000000000000000000@m@@@", nil}, |
| {"@3@@host:11@@@@m@@", "@5@@host:11@00000000000000000000000000000000@m@@@", nil}, |
| {"@3@@host:12@@@@@@", "@5@@host:12@00000000000000000000000000000000@m@@@", nil}, |
| } |
| runEndpointTests(t, testcases) |
| } |
| |
| func runEndpointTests(t *testing.T, testcases []endpointTest) { |
| for _, test := range testcases { |
| ep, err := NewEndpoint(test.input) |
| if err == nil && test.err == nil && ep.String() != test.output { |
| t.Errorf("NewEndpoint(%q): unexpected endpoint string %q != %q", |
| test.input, ep.String(), test.output) |
| continue |
| } |
| switch { |
| case test.err == err: // do nothing |
| case test.err == nil && err != nil: |
| t.Errorf("NewEndpoint(%q): unexpected error %q", test.output, err) |
| case test.err != nil && err == nil: |
| t.Errorf("NewEndpoint(%q): missing error %q", test.output, test.err) |
| case err.Error() != test.err.Error(): |
| t.Errorf("NewEndpoint(%q): unexpected error %q != %q", test.output, err, test.err) |
| } |
| } |
| } |
| |
| func TestHostPortEndpoint(t *testing.T) { |
| defver := defaultVersion |
| defer func() { |
| defaultVersion = defver |
| }() |
| defaultVersion = 4 |
| testcases := []endpointTest{ |
| {"localhost:10", "@4@@localhost:10@00000000000000000000000000000000@@@m@@@", nil}, |
| {"localhost:", "@4@@localhost:@00000000000000000000000000000000@@@m@@@", nil}, |
| {"localhost", "", errInvalidEndpointString}, |
| {"(dev.v.io/service/mounttabled)@ns.dev.v.io:8101", "@4@@ns.dev.v.io:8101@00000000000000000000000000000000@@@m@dev.v.io/service/mounttabled@@", nil}, |
| {"(dev.v.io/users/foo@bar.com)@ns.dev.v.io:8101", "@4@@ns.dev.v.io:8101@00000000000000000000000000000000@@@m@dev.v.io/users/foo@bar.com@@", nil}, |
| {"(@1@tcp)@ns.dev.v.io:8101", "@4@@ns.dev.v.io:8101@00000000000000000000000000000000@@@m@@1@tcp@@", nil}, |
| } |
| runEndpointTests(t, testcases) |
| } |
| |
| func TestParseHostPort(t *testing.T) { |
| dns := &Endpoint{ |
| Protocol: "tcp", |
| Address: "batman.com:4444", |
| MinRPCVersion: version.DeprecatedRPCVersion, |
| MaxRPCVersion: version.DeprecatedRPCVersion, |
| IsMountTable: true, |
| } |
| ipv4 := &Endpoint{ |
| Protocol: "tcp", |
| Address: "192.168.1.1:4444", |
| MinRPCVersion: version.DeprecatedRPCVersion, |
| MaxRPCVersion: version.DeprecatedRPCVersion, |
| IsMountTable: true, |
| } |
| ipv6 := &Endpoint{ |
| Protocol: "tcp", |
| Address: "[01:02::]:4444", |
| MinRPCVersion: version.DeprecatedRPCVersion, |
| MaxRPCVersion: version.DeprecatedRPCVersion, |
| IsMountTable: true, |
| } |
| testcases := []struct { |
| Endpoint naming.Endpoint |
| Host, Port string |
| }{ |
| {dns, "batman.com", "4444"}, |
| {ipv4, "192.168.1.1", "4444"}, |
| {ipv6, "01:02::", "4444"}, |
| } |
| |
| for _, test := range testcases { |
| addr := net.JoinHostPort(test.Host, test.Port) |
| epString := naming.FormatEndpoint("tcp", addr) |
| if ep, err := NewEndpoint(epString); err != nil { |
| t.Errorf("NewEndpoint(%q) failed with %v", addr, err) |
| } else { |
| if !reflect.DeepEqual(test.Endpoint, ep) { |
| t.Errorf("Got endpoint %T = %#v, want %T = %#v for string %q", ep, ep, test.Endpoint, test.Endpoint, addr) |
| } |
| } |
| } |
| } |
