cluster_test.go

package proxmox

import (
	"fmt"
	"net/http"
	"testing"

	"github.com/stretchr/testify/require"
)

func testStr(s string) *string {
	return &s
}

func testInt(i int) *int {
	return &i
}

func testFloat64(f float64) *float64 {
	return &f
}

func testIntOrString(is IntOrString) *IntOrString {
	return &is
}

func TestGetClusterResources(t *testing.T) {
	mux, server, client := setup(t)
	defer teardown(server)

	mux.HandleFunc("/api2/json/cluster/resources", func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		w.WriteHeader(http.StatusOK)
		_, err := fmt.Fprint(w, fixture("clusters/get_cluster_resources.json"))
		if err != nil {
			return
		}
	})

	want := GetClusterResourcesResponse{
		Data: []GetClusterResourcesData{
			{
				CPU:       testFloat64(.00215395696684676),
				Disk:      testInt(0),
				DiskRead:  testInt(1899047936),
				DiskWrite: testInt(2697581568),
				ID:        "qemu/101",
				MaxCPU:    testInt(4),
				MaxDisk:   testInt(549755813888),
				MaxMem:    testInt(17179869184),
				Mem:       testInt(3865654169),
				Name:      testStr("my-vm"),
				NetIn:     testInt(554461212),
				NetOut:    testInt(13830445),
				Node:      "node1",
				Status:    "running",
				Template:  testInt(0),
				Type:      "qemu",
				Uptime:    testInt(234806),
				VMID:      testIntOrString("101"),
			},
			{
				CgroupMode: testInt(2),
				CPU:        testFloat64(0.0496424063946151),
				Disk:       testInt(5996113920),
				ID:         "node/node1",
				Level:      testStr(""),
				MaxCPU:     testInt(16),
				MaxDisk:    testInt(100861726720),
				MaxMem:     testInt(134850514944),
				Mem:        testInt(64139268096),
				Node:       "node1",
				Status:     "online",
				Type:       "node",
				Uptime:     testInt(234832),
			},
			{
				Content:    testStr("iso,backup,vztmpl"),
				Disk:       testInt(5996113920),
				ID:         "storage/node1/local",
				MaxDisk:    testInt(100861726720),
				Node:       "node1",
				PluginType: testStr("dir"),
				Shared:     testInt(0),
				Status:     "available",
				Storage:    testStr("local"),
				Type:       "storage",
			},
			{
				ID:     "sdn/node1/localnetwork",
				Node:   "node1",
				SDN:    testStr("localnetwork"),
				Status: "ok",
				Type:   "sdn",
			},
		},
	}

	r, resp, err := client.Cluster.GetClusterResources()
	require.NoError(t, err)
	require.NotNil(t, resp)
	require.Equal(t, want, *r)
}