github.phpd.cn/hashicorp/packer@v1.3.2/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  	GetImageId  string
    21  	GetImageErr error
    22  
    23  	GetMachineErr error
    24  
    25  	StopMachineId  string
    26  	StopMachineErr error
    27  
    28  	WaitForImageCreationErr error
    29  
    30  	WaitForMachineDeletionErr error
    31  
    32  	WaitForMachineStateErr error
    33  }
    34  
    35  func (d *DriverMock) GetImage(config Config) (string, error) {
    36  	if d.GetImageErr != nil {
    37  		return "", d.GetImageErr
    38  	}
    39  
    40  	return config.MachineImage, nil
    41  }
    42  
    43  func (d *DriverMock) CreateImageFromMachine(machineId string, config Config) (string, error) {
    44  	if d.CreateImageFromMachineErr != nil {
    45  		return "", d.CreateImageFromMachineErr
    46  	}
    47  
    48  	d.CreateImageFromMachineId = config.ImageName
    49  
    50  	return d.CreateImageFromMachineId, nil
    51  }
    52  
    53  func (d *DriverMock) CreateMachine(config Config) (string, error) {
    54  	if d.CreateMachineErr != nil {
    55  		return "", d.CreateMachineErr
    56  	}
    57  
    58  	d.CreateMachineId = config.MachineName
    59  
    60  	return d.CreateMachineId, nil
    61  }
    62  
    63  func (d *DriverMock) DeleteImage(imageId string) error {
    64  	if d.DeleteImageErr != nil {
    65  		return d.DeleteImageErr
    66  	}
    67  
    68  	d.DeleteImageId = imageId
    69  
    70  	return nil
    71  }
    72  
    73  func (d *DriverMock) DeleteMachine(machineId string) error {
    74  	if d.DeleteMachineErr != nil {
    75  		return d.DeleteMachineErr
    76  	}
    77  
    78  	d.DeleteMachineId = machineId
    79  
    80  	return nil
    81  }
    82  
    83  func (d *DriverMock) GetMachineIP(machineId string) (string, error) {
    84  	if d.GetMachineErr != nil {
    85  		return "", d.GetMachineErr
    86  	}
    87  
    88  	return "ip", nil
    89  }
    90  
    91  func (d *DriverMock) StopMachine(machineId string) error {
    92  	d.StopMachineId = machineId
    93  
    94  	return d.StopMachineErr
    95  }
    96  
    97  func (d *DriverMock) WaitForImageCreation(machineId string, timeout time.Duration) error {
    98  	return d.WaitForImageCreationErr
    99  }
   100  
   101  func (d *DriverMock) WaitForMachineDeletion(machineId string, timeout time.Duration) error {
   102  	return d.WaitForMachineDeletionErr
   103  }
   104  
   105  func (d *DriverMock) WaitForMachineState(machineId string, state string, timeout time.Duration) error {
   106  	return d.WaitForMachineStateErr
   107  }