89 lines
1.9 KiB
Go
89 lines
1.9 KiB
Go
package vagrant
|
|
|
|
// Create a mock driver so that we can test Vagrant builder steps
|
|
type MockVagrantDriver struct {
|
|
InitCalled bool
|
|
AddCalled bool
|
|
UpCalled bool
|
|
HaltCalled bool
|
|
SuspendCalled bool
|
|
SSHConfigCalled bool
|
|
DestroyCalled bool
|
|
PackageCalled bool
|
|
VerifyCalled bool
|
|
VersionCalled bool
|
|
|
|
ReturnError error
|
|
ReturnSSHConfig *VagrantSSHConfig
|
|
GlobalID string
|
|
}
|
|
|
|
func (d *MockVagrantDriver) Init([]string) error {
|
|
d.InitCalled = true
|
|
return d.ReturnError
|
|
}
|
|
|
|
func (d *MockVagrantDriver) Add([]string) error {
|
|
d.AddCalled = true
|
|
return d.ReturnError
|
|
}
|
|
|
|
func (d *MockVagrantDriver) Up([]string) (string, string, error) {
|
|
d.UpCalled = true
|
|
return "", "", nil
|
|
}
|
|
|
|
func (d *MockVagrantDriver) Halt(string) error {
|
|
d.HaltCalled = true
|
|
return d.ReturnError
|
|
}
|
|
|
|
func (d *MockVagrantDriver) Suspend(string) error {
|
|
d.SuspendCalled = true
|
|
return d.ReturnError
|
|
}
|
|
|
|
func (d *MockVagrantDriver) SSHConfig(gid string) (*VagrantSSHConfig, error) {
|
|
d.SSHConfigCalled = true
|
|
// track the input value
|
|
d.GlobalID = gid
|
|
|
|
if d.ReturnSSHConfig != nil {
|
|
return d.ReturnSSHConfig, nil
|
|
}
|
|
|
|
sshConfig := VagrantSSHConfig{
|
|
Hostname: "127.0.0.1",
|
|
User: "vagrant",
|
|
Port: "2222",
|
|
UserKnownHostsFile: "/dev/null",
|
|
StrictHostKeyChecking: false,
|
|
PasswordAuthentication: false,
|
|
IdentityFile: "\"/path with spaces/insecure_private_key\"",
|
|
IdentitiesOnly: true,
|
|
LogLevel: "FATAL"}
|
|
return &sshConfig, d.ReturnError
|
|
}
|
|
|
|
func (d *MockVagrantDriver) Destroy(string) error {
|
|
d.DestroyCalled = true
|
|
return d.ReturnError
|
|
}
|
|
|
|
func (d *MockVagrantDriver) Package([]string) error {
|
|
d.PackageCalled = true
|
|
return d.ReturnError
|
|
}
|
|
|
|
func (d *MockVagrantDriver) Verify() error {
|
|
d.VerifyCalled = true
|
|
return d.ReturnError
|
|
}
|
|
|
|
func (d *MockVagrantDriver) Version() (string, error) {
|
|
d.VersionCalled = true
|
|
return "", d.ReturnError
|
|
}
|
|
|
|
// End of mock definition
|