summaryrefslogtreecommitdiff
path: root/pkg/machine/e2e/config_set_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'pkg/machine/e2e/config_set_test.go')
-rw-r--r--pkg/machine/e2e/config_set_test.go43
1 files changed, 43 insertions, 0 deletions
diff --git a/pkg/machine/e2e/config_set_test.go b/pkg/machine/e2e/config_set_test.go
new file mode 100644
index 000000000..3c773b970
--- /dev/null
+++ b/pkg/machine/e2e/config_set_test.go
@@ -0,0 +1,43 @@
+package e2e_test
+
+import (
+ "strconv"
+)
+
+type setMachine struct {
+ cpus *uint
+ diskSize *uint
+ memory *uint
+
+ cmd []string
+}
+
+func (i *setMachine) buildCmd(m *machineTestBuilder) []string {
+ cmd := []string{"machine", "set"}
+ if i.cpus != nil {
+ cmd = append(cmd, "--cpus", strconv.Itoa(int(*i.cpus)))
+ }
+ if i.diskSize != nil {
+ cmd = append(cmd, "--disk-size", strconv.Itoa(int(*i.diskSize)))
+ }
+ if i.memory != nil {
+ cmd = append(cmd, "--memory", strconv.Itoa(int(*i.memory)))
+ }
+ cmd = append(cmd, m.name)
+ i.cmd = cmd
+ return cmd
+}
+
+func (i *setMachine) withCPUs(num uint) *setMachine {
+ i.cpus = &num
+ return i
+}
+func (i *setMachine) withDiskSize(size uint) *setMachine {
+ i.diskSize = &size
+ return i
+}
+
+func (i *setMachine) withMemory(num uint) *setMachine {
+ i.memory = &num
+ return i
+}