github.com/mattyr/nomad@v0.3.3-0.20160919021406-3485a065154a/client/rpcproxy/server_endpoint_test.go

package rpcproxy

import (
	"fmt"
	"net"
	"testing"
)

// func (k *EndpointKey) Equal(x *EndpointKey) {
func TestServerEndpointKey_Equal(t *testing.T) {
	tests := []struct {
		name  string
		s1    *ServerEndpoint
		s2    *ServerEndpoint
		equal bool
	}{
		{
			name:  "equal",
			s1:    &ServerEndpoint{Name: "k1"},
			s2:    &ServerEndpoint{Name: "k1"},
			equal: true,
		},
		{
			name:  "not equal",
			s1:    &ServerEndpoint{Name: "k1"},
			s2:    &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 *ServerEndpoint) String() {
func TestServerEndpoint_String(t *testing.T) {
	tests := []struct {
		name string
		s    *ServerEndpoint
		str  string
	}{
		{
			name: "name",
			s:    &ServerEndpoint{Name: "s"},
			str:  "s (:)",
		},
		{
			name: "name, host, port",
			s: &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)
		}
	}
}