mirror of
https://github.com/kemko/nomad.git
synced 2026-01-06 18:35:44 +03:00
Add a quick set of client/rpcproxy.ServerEndpoint equality tests
This commit is contained in:
38
client/rpcproxy/server_endpoint_internal_test.go
Normal file
38
client/rpcproxy/server_endpoint_internal_test.go
Normal file
@@ -0,0 +1,38 @@
|
||||
package rpcproxy
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
// func (k *EndpointKey) Equal(x *EndpointKey) {
|
||||
func TestServerEndpointKey_Equal(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
k1 *EndpointKey
|
||||
k2 *EndpointKey
|
||||
equal bool
|
||||
}{
|
||||
{
|
||||
name: "equal",
|
||||
k1: &EndpointKey{name: "k1"},
|
||||
k2: &EndpointKey{name: "k1"},
|
||||
equal: true,
|
||||
},
|
||||
{
|
||||
name: "not equal",
|
||||
k1: &EndpointKey{name: "k1"},
|
||||
k2: &EndpointKey{name: "k2"},
|
||||
equal: false,
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
if test.k1.Equal(test.k2) != test.equal {
|
||||
t.Errorf("fixture %s failed forward comparison", test.name)
|
||||
}
|
||||
|
||||
if test.k2.Equal(test.k1) != test.equal {
|
||||
t.Errorf("fixture %s failed reverse comparison", test.name)
|
||||
}
|
||||
}
|
||||
}
|
||||
79
client/rpcproxy/server_endpoint_test.go
Normal file
79
client/rpcproxy/server_endpoint_test.go
Normal file
@@ -0,0 +1,79 @@
|
||||
package rpcproxy_test
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
"testing"
|
||||
|
||||
"github.com/hashicorp/nomad/client/rpcproxy"
|
||||
)
|
||||
|
||||
// func (k *rpcproxy.EndpointKey) Equal(x *rpcproxy.EndpointKey) {
|
||||
func TestServerEndpointKey_Equal(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
s1 *rpcproxy.ServerEndpoint
|
||||
s2 *rpcproxy.ServerEndpoint
|
||||
equal bool
|
||||
}{
|
||||
{
|
||||
name: "equal",
|
||||
s1: &rpcproxy.ServerEndpoint{Name: "k1"},
|
||||
s2: &rpcproxy.ServerEndpoint{Name: "k1"},
|
||||
equal: true,
|
||||
},
|
||||
{
|
||||
name: "not equal",
|
||||
s1: &rpcproxy.ServerEndpoint{Name: "k1"},
|
||||
s2: &rpcproxy.ServerEndpoint{Name: "k2"},
|
||||
equal: false,
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
if test.s1.Key().Equal(test.s2.Key()) != test.equal {
|
||||
t.Errorf("fixture %s failed forward comparison", test.name)
|
||||
}
|
||||
|
||||
if test.s2.Key().Equal(test.s1.Key()) != test.equal {
|
||||
t.Errorf("fixture %s failed reverse comparison", test.name)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// func (k *rpcproxy.ServerEndpoint) String() {
|
||||
func TestServerEndpoint_String(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
s *rpcproxy.ServerEndpoint
|
||||
str string
|
||||
}{
|
||||
{
|
||||
name: "name",
|
||||
s: &rpcproxy.ServerEndpoint{Name: "s"},
|
||||
str: "s (:)",
|
||||
},
|
||||
{
|
||||
name: "name, host, port",
|
||||
s: &rpcproxy.ServerEndpoint{
|
||||
Name: "s",
|
||||
Host: "127.0.0.1",
|
||||
Port: "4647",
|
||||
},
|
||||
str: "s (tcp:127.0.0.1:4647)",
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
if test.s.Addr == nil && (test.s.Host != "" && test.s.Port != "") {
|
||||
fmt.Printf("Setting addr\n")
|
||||
addr, err := net.ResolveTCPAddr("tcp", net.JoinHostPort(test.s.Host, test.s.Port))
|
||||
if err == nil {
|
||||
test.s.Addr = addr
|
||||
}
|
||||
}
|
||||
if test.s.String() != test.str {
|
||||
t.Errorf("fixture %q failed: %q vs %q", test.name, test.s.String(), test.str)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user