CPU limit utilities with tests

Author: scott-the-programmer

minikube/lib/memory.go renamed to minikube/lib/host.go

@@ -4,13 +4,17 @@ import (
 	"k8s.io/minikube/pkg/minikube/machine"
 )
 
-
 var NoLimit = "no-limit"
 var Max = "max"
 
 // MemoryInfo holds system and container memory information
 type MemoryInfo struct {
-	SystemMemory    int
+	SystemMemory int
+}
+
+// CPUInfo holds system and container CPU information
+type CPUInfo struct {
+	CPUs int
 }
 
 // GetMemoryLimits returns the amount of memory allocated to the system and container
@@ -24,9 +28,18 @@ func GetMemoryLimit() (*MemoryInfo, error) {
 
 	// Subtract 1gb for overhead
 	memInfo := &MemoryInfo{
-		SystemMemory:    int(info.Memory) - 1024,
+		SystemMemory: int(info.Memory) - 1024,
 	}
 
 	return memInfo, nil
 }
 
+func GetCPULimit() (int, error) {
+	info, _, _, cpuErr := machine.LocalHostInfo()
+
+	if cpuErr != nil {
+		return 0, cpuErr
+	}
+
+	return int(info.CPUs), nil
+}

minikube/state_utils/memory.go renamed to minikube/state_utils/host.go

@@ -1,6 +1,8 @@
 package state_utils
 
 import (
+	"strconv"
+
 	"github.com/scott-the-programmer/terraform-provider-minikube/minikube/lib"
 	pkgutil "k8s.io/minikube/pkg/util"
 )
@@ -36,3 +38,23 @@ func GetMemory(memoryStr string) (int, error) {
 
 	return memoryMb, err
 }
+
+func GetCPU(cpuStr string) (int, error) {
+	var cpu int
+	var err error
+	if cpuStr == lib.Max {
+		cpu, err = lib.GetCPULimit()
+		if err != nil {
+			return 0, err
+		}
+	} else if cpuStr == lib.NoLimit {
+		cpu = 0
+	} else {
+		cpu, err = strconv.Atoi(cpuStr)
+		if err != nil {
+			return 0, err
+		}
+	}
+
+	return cpu, nil
+}

minikube/state_utils/memory_test.go renamed to minikube/state_utils/host_test.go

@@ -65,3 +65,57 @@ func TestGetMemory(t *testing.T) {
 		})
 	}
 }
+
+func TestGetCPU(t *testing.T) {
+	tests := []struct {
+		name        string
+		input       string
+		expected    int
+		expectError bool
+	}{
+		{
+			name:        "valid CPU count",
+			input:       "2",
+			expected:    2,
+			expectError: false,
+		},
+		{
+			name:        "no limit case",
+			input:       lib.NoLimit,
+			expected:    0,
+			expectError: false,
+		},
+		{
+			name:        "invalid CPU count",
+			input:       "invalid",
+			expected:    0,
+			expectError: true,
+		},
+		{
+			name:        "negative CPU count",
+			input:       "-1",
+			expected:    0,
+			expectError: true,
+		},
+		{
+			name:        "empty string",
+			input:       "",
+			expected:    0,
+			expectError: true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+
+			result, err := GetCPU(tt.input)
+
+			if tt.expectError {
+				assert.Error(t, err)
+			} else {
+				assert.NoError(t, err)
+				assert.Equal(t, tt.expected, result)
+			}
+		})
+	}
+}