Function TestWaitForTaskSuccess

internal/cli/cli_helpers_test.go:199–230

(t *testing.T)

}

func TestWaitForTaskSuccess(t *testing.T) {
	const upid = "UPID:pve:00001234:00000000:test:qmcreate:100:user@pam:"

	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")

		switch {
		case r.URL.Path == "/api2/json/access/ticket":
			_, _ = w.Write([]byte(authResponse))
		case r.URL.Path == "/api2/json/cluster/tasks":
			// Return the task as already complete with OK status.
			body := `{"data":[{"upid":"` + upid + `","status":"OK","endtime":1700000060,"starttime":1700000000,"node":"pve","type":"qmcreate"}]}`
			_, _ = w.Write([]byte(body))
		case strings.Contains(r.URL.Path, "/tasks/") && strings.HasSuffix(r.URL.Path, "/status"):
			_, _ = w.Write([]byte(taskStatusResponse("OK")))
		default:
			http.NotFound(w, r)
		}
	}))
	defer server.Close()

	client := newTestClient(t, server.URL)

	exitStatus, err := waitForTask(context.Background(), client, "pve", upid, "test-op")
	if err != nil {
		t.Fatalf("waitForTask returned unexpected error: %v", err)
	}

	if exitStatus != "OK" {
		t.Errorf("exitStatus = %q, want %q", exitStatus, "OK")
	}
}

func TestWaitForTaskFailure(t *testing.T) {
	const upid = "UPID:pve:00001234:00000000:test:qmcreate:100:user@pam:"