108 lines
2.1 KiB
Go
108 lines
2.1 KiB
Go
package triton
|
|
|
|
import (
|
|
"time"
|
|
)
|
|
|
|
type DriverMock struct {
|
|
CreateImageFromMachineId string
|
|
CreateImageFromMachineErr error
|
|
|
|
CreateMachineId string
|
|
CreateMachineErr error
|
|
|
|
DeleteImageId string
|
|
DeleteImageErr error
|
|
|
|
DeleteMachineId string
|
|
DeleteMachineErr error
|
|
|
|
GetImageId string
|
|
GetImageErr error
|
|
|
|
GetMachineErr error
|
|
|
|
StopMachineId string
|
|
StopMachineErr error
|
|
|
|
WaitForImageCreationErr error
|
|
|
|
WaitForMachineDeletionErr error
|
|
|
|
WaitForMachineStateErr error
|
|
}
|
|
|
|
func (d *DriverMock) GetImage(config Config) (string, error) {
|
|
if d.GetImageErr != nil {
|
|
return "", d.GetImageErr
|
|
}
|
|
|
|
return config.MachineImage, nil
|
|
}
|
|
|
|
func (d *DriverMock) CreateImageFromMachine(machineId string, config Config) (string, error) {
|
|
if d.CreateImageFromMachineErr != nil {
|
|
return "", d.CreateImageFromMachineErr
|
|
}
|
|
|
|
d.CreateImageFromMachineId = config.ImageName
|
|
|
|
return d.CreateImageFromMachineId, nil
|
|
}
|
|
|
|
func (d *DriverMock) CreateMachine(config Config) (string, error) {
|
|
if d.CreateMachineErr != nil {
|
|
return "", d.CreateMachineErr
|
|
}
|
|
|
|
d.CreateMachineId = config.MachineName
|
|
|
|
return d.CreateMachineId, nil
|
|
}
|
|
|
|
func (d *DriverMock) DeleteImage(imageId string) error {
|
|
if d.DeleteImageErr != nil {
|
|
return d.DeleteImageErr
|
|
}
|
|
|
|
d.DeleteImageId = imageId
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *DriverMock) DeleteMachine(machineId string) error {
|
|
if d.DeleteMachineErr != nil {
|
|
return d.DeleteMachineErr
|
|
}
|
|
|
|
d.DeleteMachineId = machineId
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *DriverMock) GetMachineIP(machineId string) (string, error) {
|
|
if d.GetMachineErr != nil {
|
|
return "", d.GetMachineErr
|
|
}
|
|
|
|
return "ip", nil
|
|
}
|
|
|
|
func (d *DriverMock) StopMachine(machineId string) error {
|
|
d.StopMachineId = machineId
|
|
|
|
return d.StopMachineErr
|
|
}
|
|
|
|
func (d *DriverMock) WaitForImageCreation(machineId string, timeout time.Duration) error {
|
|
return d.WaitForImageCreationErr
|
|
}
|
|
|
|
func (d *DriverMock) WaitForMachineDeletion(machineId string, timeout time.Duration) error {
|
|
return d.WaitForMachineDeletionErr
|
|
}
|
|
|
|
func (d *DriverMock) WaitForMachineState(machineId string, state string, timeout time.Duration) error {
|
|
return d.WaitForMachineStateErr
|
|
}
|