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
}