44 lines
801 B
Go
44 lines
801 B
Go
package e2e
|
|
|
|
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
|
|
}
|