TestProbeRouteIsBusy: uses a non-cancellable state

Signed-off-by: Luis Gustavo S. Barreto <gustavo@ossystems.com.br>
UpdateHub · Jul 12, 2018 · 1522401 · 1522401
1 parent 254ffc7
commit 1522401
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/agent_backend_test.go b/server/agent_backend_test.go
@@ -254,14 +254,14 @@ func TestProbeRouteWithDefaultApiClient(t *testing.T) {
 func TestProbeRouteIsBusy(t *testing.T) {
 	out := map[string]interface{}{}
 	out["busy"] = true
-	out["current-state"] = "probe"
+	out["current-state"] = "rebooting"
 
 	expectedResponse, _ := json.MarshalIndent(out, "", "    ")
 
 	uh, ab, _, _ := setup(t)
 	defer teardown(t)
 
-	s := updatehub.NewProbeState(uh.DefaultApiClient)
+	s := updatehub.NewRebootingState(uh.DefaultApiClient, nil)
 	uh.SetState(s)
 
 	rwm := &responsewritermock.ResponseWriterMock{}