client: test heartbeats

client/client_test.go

@@ -169,3 +169,36 @@ func TestClient_Register(t *testing.T) {
 		t.Fatalf("err: %v", err)
 	})
 }
+
+func TestClient_Heartbeat(t *testing.T) {
+	s1, _ := testServer(t, func(c *nomad.Config) {
+		c.MinHeartbeatTTL = 50 * time.Millisecond
+	})
+	defer s1.Shutdown()
+	testutil.WaitForLeader(t, s1.RPC)
+
+	c1 := testClient(t, func(c *config.Config) {
+		c.RPCHandler = s1
+	})
+	defer c1.Shutdown()
+
+	req := structs.NodeSpecificRequest{
+		NodeID:       c1.Node().ID,
+		QueryOptions: structs.QueryOptions{Region: "region1"},
+	}
+	var out structs.SingleNodeResponse
+
+	// Register should succeed
+	testutil.WaitForResult(func() (bool, error) {
+		err := s1.RPC("Client.GetNode", &req, &out)
+		if err != nil {
+			return false, err
+		}
+		if out.Node == nil {
+			return false, fmt.Errorf("missing reg")
+		}
+		return out.Node.Status == structs.NodeStatusReady, nil
+	}, func(err error) {
+		t.Fatalf("err: %v", err)
+	})
+}