Function TestBuildModifyPresets

cmd/modify_test.go:427–462

── buildModifyPresets ──────────────────────────────────────────────────────

(t *testing.T)

// ── buildModifyPresets ──────────────────────────────────────────────────────

func TestBuildModifyPresets(t *testing.T) {
	t.Run("no flags set produces empty presets", func(t *testing.T) {
		cmd := newModifyCmd()
		p := buildModifyPresets(cmd)
		assert.True(t, p.IsEmpty())
	})

	t.Run("all flags set populates all fields", func(t *testing.T) {
		cmd := newModifyCmd()
		_ = cmd.Flags().Set("gpu", "h100")
		_ = cmd.Flags().Set("num-gpus", "2")
		_ = cmd.Flags().Set("vcpus", "8")
		_ = cmd.Flags().Set("disk", "200")

		p := buildModifyPresets(cmd)
		assert.False(t, p.IsEmpty())
		require.NotNil(t, p.GPUType)
		assert.Equal(t, "h100", *p.GPUType)
		require.NotNil(t, p.NumGPUs)
		assert.Equal(t, 2, *p.NumGPUs)
		require.NotNil(t, p.VCPUs)
		assert.Equal(t, 8, *p.VCPUs)
		require.NotNil(t, p.DiskSizeGB)
		assert.Equal(t, 200, *p.DiskSizeGB)
	})

	t.Run("partial flags set", func(t *testing.T) {
		cmd := newModifyCmd()
		_ = cmd.Flags().Set("gpu", "a100")

		p := buildModifyPresets(cmd)
		assert.False(t, p.IsEmpty())
		require.NotNil(t, p.GPUType)
		assert.Equal(t, "a100", *p.GPUType)
	})
}