feat(config-qemu-memory-string): parse memory capacity both from string and integer.

proxmox/config_qemu_memory.go

@@ -1,7 +1,10 @@
 package proxmox
 
 import (
+	"encoding/json"
 	"errors"
+	"fmt"
+	"strconv"
 
 	"github.com/Telmate/proxmox-api-go/internal/parse"
 )
@@ -135,11 +138,51 @@ func (m QemuMemoryBalloonCapacity) Validate() error {
 type QemuMemoryCapacity uint32 // max 4178944
 
 const (
-	QemuMemoryCapacity_Error_Maximum string             = "memory capacity has a maximum of 4178944"
-	QemuMemoryCapacity_Error_Minimum string             = "memory capacity has a minimum of 1"
-	qemuMemoryCapacity_Max           QemuMemoryCapacity = 4178944
+	QemuMemoryCapacity_Error_Maximum     string             = "memory capacity has a maximum of 4178944"
+	QemuMemoryCapacity_Error_Minimum     string             = "memory capacity has a minimum of 1"
+	QemuMemoryCapacity_Error_InvalidType string             = "memory capacity has invalid type"
+	qemuMemoryCapacity_Max               QemuMemoryCapacity = 4178944
+	qemuMemoryCapacity_Base                                 = 10
 )
 
+func (capacity *QemuMemoryCapacity) populateString(value string) error {
+	resultValue, err := strconv.ParseUint(value, qemuMemoryCapacity_Base, 32)
+	if err != nil {
+		return err
+	}
+
+	*capacity = QemuMemoryCapacity(resultValue)
+	return nil
+}
+
+func (capacity *QemuMemoryCapacity) populateNumber(value float64) {
+	*capacity = QemuMemoryCapacity(value)
+}
+
+func (capacity *QemuMemoryCapacity) populate(rawValue interface{}) error {
+	switch value := rawValue.(type) {
+	case string:
+		return capacity.populateString(value)
+
+	case float64:
+		capacity.populateNumber(value)
+		return nil
+
+	default:
+		return fmt.Errorf("%s: %T", QemuMemoryCapacity_Error_InvalidType, rawValue)
+	}
+}
+
+func (capacity *QemuMemoryCapacity) UnmarshalJSON(data []byte) error {
+	var rawValue interface{}
+
+	if err := json.Unmarshal(data, &rawValue); err != nil {
+		return err
+	}
+
+	return capacity.populate(rawValue)
+}
+
 func (m QemuMemoryCapacity) Validate() error {
 	if m == 0 {
 		return errors.New(QemuMemoryCapacity_Error_Minimum)

proxmox/config_qemu_memory_test.go

@@ -1,7 +1,10 @@
 package proxmox
 
 import (
+	"encoding/json"
 	"errors"
+	"fmt"
+	"strconv"
 	"testing"
 
 	"github.com/Telmate/proxmox-api-go/internal/util"
@@ -219,3 +222,44 @@ func Test_QemuMemoryShares_Validate(t *testing.T) {
 		})
 	}
 }
+
+func Test_QemuMemoryCapacity_UnmarshalJSON(t *testing.T) {
+	tests := []struct {
+		name        string
+		input       []byte
+		outputValue QemuMemoryCapacity
+		outputError error
+	}{
+		{name: `Integer`,
+			input:       []byte(`32768`),
+			outputValue: QemuMemoryCapacity(32768),
+			outputError: nil,
+		},
+		{name: `String`,
+			input:       []byte(`"32768"`),
+			outputValue: QemuMemoryCapacity(32768),
+			outputError: nil,
+		},
+		{name: `InvalidSyntax`,
+			input:       []byte(`"abcd"`),
+			outputValue: QemuMemoryCapacity(0),
+			outputError: &strconv.NumError{Func: "ParseUint", Num: "abcd", Err: strconv.ErrSyntax},
+		},
+		{name: `InvalidType`,
+			input:       []byte(`true`),
+			outputValue: QemuMemoryCapacity(0),
+			outputError: fmt.Errorf("%s: %T", QemuMemoryCapacity_Error_InvalidType, true),
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			var capacity QemuMemoryCapacity
+
+			err := json.Unmarshal(test.input, &capacity)
+
+			require.Equal(t, test.outputError, err)
+			require.Equal(t, test.outputValue, capacity)
+		})
+	}
+}