Function newSingleNodeWithVMServer

pkg/api/grouped_cluster_test.go:242–287

newSingleNodeWithVMServer returns a test server that serves one node and one QEMU VM with the given vmID and name under /cluster/resources.

(ticket, nodeName string, vmID int, vmName, vmStatus string)

// newSingleNodeWithVMServer returns a test server that serves one node and one
// QEMU VM with the given vmID and name under /cluster/resources.
func newSingleNodeWithVMServer(ticket, nodeName string, vmID int, vmName, vmStatus string) *httptest.Server {
	return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		switch r.URL.Path {
		case testAPITicketPath:
			_ = json.NewEncoder(w).Encode(map[string]interface{}{
				"data": map[string]interface{}{
					"ticket": ticket, "CSRFPreventionToken": "csrf",
				},
			})
		case testAPIClusterResourcesPath:
			_ = json.NewEncoder(w).Encode(map[string]interface{}{
				"data": []interface{}{
					map[string]interface{}{
						"type": "node", "node": nodeName,
						"cpu": 0.1, "maxcpu": 4,
						"mem": 1024.0, "maxmem": 2048.0,
						"disk": 1024.0, "maxdisk": 2048.0,
						"uptime": 100,
					},
					map[string]interface{}{
						"type": "qemu", "node": nodeName,
						"vmid": vmID, "name": vmName, "status": vmStatus,
					},
				},
			})
		case testAPIClusterStatusPath:
			_ = json.NewEncoder(w).Encode(map[string]interface{}{
				"data": []interface{}{
					map[string]interface{}{
						"id": "node/" + nodeName, "name": nodeName,
						"type": "node", "online": 1,
					},
				},
			})
		case testAPINodesPath:
			_ = json.NewEncoder(w).Encode(map[string]interface{}{
				"data": []interface{}{
					map[string]interface{}{"node": nodeName, "status": "online"},
				},
			})
		default:
			w.WriteHeader(http.StatusNotFound)
		}
	}))
}

func TestFindVMByIDInGroup_Unique(t *testing.T) {
	s1 := newSingleNodeWithVMServer("t1", "pve1", 100, "web", "running")