client: testing RPC

client/client_test.go

@@ -1,6 +1,59 @@
 package client
 
-import "testing"
+import (
+	"fmt"
+	"net"
+	"sync/atomic"
+	"testing"
+	"time"
+
+	"github.com/hashicorp/nomad/nomad"
+	"github.com/hashicorp/nomad/testutil"
+)
+
+var nextPort uint32 = 16000
+
+func getPort() int {
+	return int(atomic.AddUint32(&nextPort, 1))
+}
+
+func testServer(t *testing.T, cb func(*nomad.Config)) (*nomad.Server, string) {
+	// Setup the default settings
+	config := nomad.DefaultConfig()
+	config.Build = "unittest"
+	config.DevMode = true
+	config.RPCAddr = &net.TCPAddr{
+		IP:   []byte{127, 0, 0, 1},
+		Port: getPort(),
+	}
+	config.NodeName = fmt.Sprintf("Node %d", config.RPCAddr.Port)
+
+	// Tighten the Serf timing
+	config.SerfConfig.MemberlistConfig.BindAddr = "127.0.0.1"
+	config.SerfConfig.MemberlistConfig.BindPort = getPort()
+	config.SerfConfig.MemberlistConfig.SuspicionMult = 2
+	config.SerfConfig.MemberlistConfig.RetransmitMult = 2
+	config.SerfConfig.MemberlistConfig.ProbeTimeout = 50 * time.Millisecond
+	config.SerfConfig.MemberlistConfig.ProbeInterval = 100 * time.Millisecond
+	config.SerfConfig.MemberlistConfig.GossipInterval = 100 * time.Millisecond
+
+	// Tighten the Raft timing
+	config.RaftConfig.LeaderLeaseTimeout = 20 * time.Millisecond
+	config.RaftConfig.HeartbeatTimeout = 40 * time.Millisecond
+	config.RaftConfig.ElectionTimeout = 40 * time.Millisecond
+
+	// Invoke the callback if any
+	if cb != nil {
+		cb(config)
+	}
+
+	// Create server
+	server, err := nomad.NewServer(config)
+	if err != nil {
+		t.Fatalf("err: %v", err)
+	}
+	return server, config.RPCAddr.String()
+}
 
 func testClient(t *testing.T, cb func(c *Config)) *Client {
 	conf := DefaultConfig()
@@ -21,3 +74,22 @@ func TestClient_StartStop(t *testing.T) {
 		t.Fatalf("err: %v", err)
 	}
 }
+
+func TestClient_RPC(t *testing.T) {
+	s1, addr := testServer(t, nil)
+	defer s1.Shutdown()
+
+	c1 := testClient(t, func(c *Config) {
+		c.Servers = []string{addr}
+	})
+	defer c1.Shutdown()
+
+	// RPC should succeed
+	testutil.WaitForResult(func() (bool, error) {
+		var out struct{}
+		err := c1.RPC("Status.Ping", struct{}{}, &out)
+		return err == nil, err
+	}, func(err error) {
+		t.Fatalf("err: %v", err)
+	})
+}