github.com/mmcquillan/packer@v1.1.1-0.20171009221028-c85cf0483a5d/builder/triton/driver_mock.go (about)

     1  package triton
     2  
     3  import (
     4  	"time"
     5  )
     6  
     7  type DriverMock struct {
     8  	CreateImageFromMachineId  string
     9  	CreateImageFromMachineErr error
    10  
    11  	CreateMachineId  string
    12  	CreateMachineErr error
    13  
    14  	DeleteImageId  string
    15  	DeleteImageErr error
    16  
    17  	DeleteMachineId  string
    18  	DeleteMachineErr error
    19  
    20  	GetMachineErr error
    21  
    22  	StopMachineId  string
    23  	StopMachineErr error
    24  
    25  	WaitForImageCreationErr error
    26  
    27  	WaitForMachineDeletionErr error
    28  
    29  	WaitForMachineStateErr error
    30  }
    31  
    32  func (d *DriverMock) CreateImageFromMachine(machineId string, config Config) (string, error) {
    33  	if d.CreateImageFromMachineErr != nil {
    34  		return "", d.CreateImageFromMachineErr
    35  	}
    36  
    37  	d.CreateImageFromMachineId = config.ImageName
    38  
    39  	return d.CreateImageFromMachineId, nil
    40  }
    41  
    42  func (d *DriverMock) CreateMachine(config Config) (string, error) {
    43  	if d.CreateMachineErr != nil {
    44  		return "", d.CreateMachineErr
    45  	}
    46  
    47  	d.CreateMachineId = config.MachineName
    48  
    49  	return d.CreateMachineId, nil
    50  }
    51  
    52  func (d *DriverMock) DeleteImage(imageId string) error {
    53  	if d.DeleteImageErr != nil {
    54  		return d.DeleteImageErr
    55  	}
    56  
    57  	d.DeleteImageId = imageId
    58  
    59  	return nil
    60  }
    61  
    62  func (d *DriverMock) DeleteMachine(machineId string) error {
    63  	if d.DeleteMachineErr != nil {
    64  		return d.DeleteMachineErr
    65  	}
    66  
    67  	d.DeleteMachineId = machineId
    68  
    69  	return nil
    70  }
    71  
    72  func (d *DriverMock) GetMachineIP(machineId string) (string, error) {
    73  	if d.GetMachineErr != nil {
    74  		return "", d.GetMachineErr
    75  	}
    76  
    77  	return "ip", nil
    78  }
    79  
    80  func (d *DriverMock) StopMachine(machineId string) error {
    81  	d.StopMachineId = machineId
    82  
    83  	return d.StopMachineErr
    84  }
    85  
    86  func (d *DriverMock) WaitForImageCreation(machineId string, timeout time.Duration) error {
    87  	return d.WaitForImageCreationErr
    88  }
    89  
    90  func (d *DriverMock) WaitForMachineDeletion(machineId string, timeout time.Duration) error {
    91  	return d.WaitForMachineDeletionErr
    92  }
    93  
    94  func (d *DriverMock) WaitForMachineState(machineId string, state string, timeout time.Duration) error {
    95  	return d.WaitForMachineStateErr
    96  }