open-nomad/client/fingerprint/storage_test.go

package fingerprint

import (
	"strconv"
	"testing"

	"github.com/hashicorp/nomad/nomad/structs"
)

func TestStorageFingerprint(t *testing.T) {
	fp := NewStorageFingerprint(testLogger())
	node := &structs.Node{
		Attributes: make(map[string]string),
	}

	assertFingerprintOK(t, fp, node)

	assertNodeAttributeContains(t, node, "unique.storage.volume")
	assertNodeAttributeContains(t, node, "unique.storage.bytestotal")
	assertNodeAttributeContains(t, node, "unique.storage.bytesfree")

	total, err := strconv.ParseInt(node.Attributes["unique.storage.bytestotal"], 10, 64)
	if err != nil {
		t.Fatalf("Failed to parse unique.storage.bytestotal: %s", err)
	}
	free, err := strconv.ParseInt(node.Attributes["unique.storage.bytesfree"], 10, 64)
	if err != nil {
		t.Fatalf("Failed to parse unique.storage.bytesfree: %s", err)
	}

	if free > total {
		t.Fatalf("unique.storage.bytesfree %d is larger than unique.storage.bytestotal %d", free, total)
	}

	if node.Resources == nil {
		t.Fatalf("Node Resources was nil")
	}
	if node.Resources.DiskMB == 0 {
		t.Errorf("Expected node.Resources.DiskMB to be non-zero")
	}
}