Function TestCreateConfigDiskSizeBoundaries

cmd/create_test.go:451–502

TestCreateConfigDiskSizeBoundaries verifies that disk size validation correctly enforces the storage range from the GPU spec. The a6000 prototyping spec has StorageGB: {Min: 100, Max: 500}.

(t *testing.T)

// correctly enforces the storage range from the GPU spec.
// The a6000 prototyping spec has StorageGB: {Min: 100, Max: 500}.
func TestCreateConfigDiskSizeBoundaries(t *testing.T) {
	tests := []struct {
		name        string
		diskSizeGB  int
		expectError bool
	}{
		{
			name:        "minimum valid disk size",
			diskSizeGB:  100,
			expectError: false,
		},
		{
			name:        "maximum valid disk size",
			diskSizeGB:  500,
			expectError: false,
		},
		{
			name:        "disk size too small",
			diskSizeGB:  99,
			expectError: true,
		},
		{
			name:        "disk size too large",
			diskSizeGB:  501,
			expectError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			config := &tui.CreateConfig{
				GPUType:    "a6000",
				VCPUs:      6,
				Template:   "ubuntu-22.04",
				DiskSizeGB: tt.diskSizeGB,
			}

			templates := []api.TemplateEntry{
				tmplEntry("ubuntu-22.04", "Ubuntu 22.04"),
			}

			err := validateCreateConfig(config, templates, []api.Snapshot{}, true, testSpecStore())

			if tt.expectError {
				assert.Error(t, err)
				assert.Contains(t, err.Error(), "disk size must be between 100 and 500 GB")
			} else {
				assert.NoError(t, err)
			}
		})
	}
}