Function TestTaskManager_Cancel

internal/taskmanager/manager_test.go:100–153

(t *testing.T)

}

func TestTaskManager_Cancel(t *testing.T) {
	// Setup Mock Server
	state := mockpve.NewMockState()
	r := mux.NewRouter()
	r.HandleFunc("/api2/json/nodes/{node}/tasks/{upid}/status", mockpve.HandleTaskStatus(state)).Methods("GET")
	r.HandleFunc("/api2/json/nodes/{node}/tasks/{upid}", mockpve.HandleStopTask(state)).Methods("DELETE")
	r.HandleFunc("/api2/json/access/ticket", func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		_, _ = fmt.Fprint(w, `{"data": {"ticket": "dummy", "CSRFPreventionToken": "dummy"}}`)
	})
	server := httptest.NewServer(r)
	defer server.Close()

	config := &mockConfig{addr: server.URL}
	opts := api.ClientOption(func(c *api.ClientOptions) {
		c.Logger = &mockLogger{}
		c.Cache = &interfaces.NoOpCache{}
	})
	client, _ := api.NewClient(config, opts)

	resolver := func(nodeName string) (*api.Client, error) {
		return client, nil
	}
	tm := NewTaskManager(resolver, nil)
	defer tm.Stop()

	upid := state.CreateTask("pve", "qmstart", "101", "root@pam")
	task := &Task{
		TargetVMID: 101,
		TargetNode: "pve",
		Type:       "Start",
		Operation: func() (string, error) {
			return upid, nil
		},
	}
	tm.Enqueue(task)

	// Wait for running
	assert.Eventually(t, func() bool {
		running := tm.GetActiveTaskForVM("pve", 101)
		return running != nil && running.Status == StatusRunning
	}, 1*time.Second, 10*time.Millisecond)

	// Cancel
	err := tm.CancelTask(task.ID)
	assert.NoError(t, err)

	// Wait for status to become failed/completed (since StopTask in mock just errors it out or stops it)
	// Mock HandleStopTask calls CompleteTask(..., "ERROR")

	assert.Eventually(t, func() bool {
		return tm.GetActiveTaskForVM("pve", 101) == nil
	}, 6*time.Second, 100*time.Millisecond)
}

func TestTaskManager_AllowsSameVMIDOnDifferentNodes(t *testing.T) {
	state := mockpve.NewMockState()