From 82ee68a42da2736e68b299561d5892d005a7e79f Mon Sep 17 00:00:00 2001 From: Alex Dadgar Date: Mon, 25 Jul 2016 12:46:18 -0700 Subject: [PATCH] add down to up test --- nomad/node_endpoint_test.go | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) diff --git a/nomad/node_endpoint_test.go b/nomad/node_endpoint_test.go index f9471ddd4..e8ad81a6a 100644 --- a/nomad/node_endpoint_test.go +++ b/nomad/node_endpoint_test.go @@ -212,6 +212,37 @@ func TestClientEndpoint_Register_GetEvals(t *testing.T) { if out.ModifyIndex != resp.Index { t.Fatalf("index mis-match") } + + // Transistion it to down and then ready + node.Status = structs.NodeStatusDown + reg = &structs.NodeRegisterRequest{ + Node: node, + WriteRequest: structs.WriteRequest{Region: "global"}, + } + + // Fetch the response + if err := msgpackrpc.CallWithCodec(codec, "Node.Register", reg, &resp); err != nil { + t.Fatalf("err: %v", err) + } + + if len(resp.EvalIDs) != 1 { + t.Fatalf("expected one eval; got %#v", resp.EvalIDs) + } + + node.Status = structs.NodeStatusReady + reg = &structs.NodeRegisterRequest{ + Node: node, + WriteRequest: structs.WriteRequest{Region: "global"}, + } + + // Fetch the response + if err := msgpackrpc.CallWithCodec(codec, "Node.Register", reg, &resp); err != nil { + t.Fatalf("err: %v", err) + } + + if len(resp.EvalIDs) != 1 { + t.Fatalf("expected one eval; got %#v", resp.EvalIDs) + } } func TestClientEndpoint_UpdateStatus_GetEvals(t *testing.T) {