109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package proxmox
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/Telmate/proxmox-api-go/proxmox"
|
|
"github.com/hashicorp/packer/packer-plugin-sdk/multistep"
|
|
packersdk "github.com/hashicorp/packer/packer-plugin-sdk/packer"
|
|
)
|
|
|
|
type startedVMCleanerMock struct {
|
|
stopVm func() (string, error)
|
|
deleteVm func() (string, error)
|
|
}
|
|
|
|
func (m startedVMCleanerMock) StopVm(*proxmox.VmRef) (string, error) {
|
|
return m.stopVm()
|
|
}
|
|
func (m startedVMCleanerMock) DeleteVm(*proxmox.VmRef) (string, error) {
|
|
return m.deleteVm()
|
|
}
|
|
|
|
var _ startedVMCleaner = &startedVMCleanerMock{}
|
|
|
|
func TestCleanupStartVM(t *testing.T) {
|
|
cs := []struct {
|
|
name string
|
|
setVmRef bool
|
|
setSuccess bool
|
|
stopVMErr error
|
|
expectCallStopVM bool
|
|
deleteVMErr error
|
|
expectCallDeleteVM bool
|
|
}{
|
|
{
|
|
name: "when vmRef state is not set, nothing should happen",
|
|
setVmRef: false,
|
|
expectCallStopVM: false,
|
|
},
|
|
{
|
|
name: "when success state is set, nothing should happen",
|
|
setVmRef: true,
|
|
setSuccess: true,
|
|
expectCallStopVM: false,
|
|
},
|
|
{
|
|
name: "when not successful, vm should be stopped and deleted",
|
|
setVmRef: true,
|
|
setSuccess: false,
|
|
expectCallStopVM: true,
|
|
expectCallDeleteVM: true,
|
|
},
|
|
{
|
|
name: "if stopping fails, DeleteVm should not be called",
|
|
setVmRef: true,
|
|
setSuccess: false,
|
|
expectCallStopVM: true,
|
|
stopVMErr: fmt.Errorf("some error"),
|
|
expectCallDeleteVM: false,
|
|
},
|
|
}
|
|
|
|
for _, c := range cs {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
var stopWasCalled, deleteWasCalled bool
|
|
|
|
cleaner := startedVMCleanerMock{
|
|
stopVm: func() (string, error) {
|
|
if !c.expectCallStopVM {
|
|
t.Error("Did not expect StopVm to be called")
|
|
}
|
|
|
|
stopWasCalled = true
|
|
return "", c.stopVMErr
|
|
},
|
|
deleteVm: func() (string, error) {
|
|
if !c.expectCallDeleteVM {
|
|
t.Error("Did not expect DeleteVm to be called")
|
|
}
|
|
|
|
deleteWasCalled = true
|
|
return "", c.deleteVMErr
|
|
},
|
|
}
|
|
|
|
state := new(multistep.BasicStateBag)
|
|
state.Put("ui", packersdk.TestUi(t))
|
|
state.Put("proxmoxClient", cleaner)
|
|
if c.setVmRef {
|
|
state.Put("vmRef", proxmox.NewVmRef(1))
|
|
}
|
|
if c.setSuccess {
|
|
state.Put("success", "true")
|
|
}
|
|
|
|
step := stepStartVM{}
|
|
step.Cleanup(state)
|
|
|
|
if c.expectCallStopVM && !stopWasCalled {
|
|
t.Error("Expected StopVm to be called, but it wasn't")
|
|
}
|
|
if c.expectCallDeleteVM && !deleteWasCalled {
|
|
t.Error("Expected DeleteVm to be called, but it wasn't")
|
|
}
|
|
})
|
|
}
|
|
}
|