132 lines
3.2 KiB
Go
132 lines
3.2 KiB
Go
package chroot
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2019-07-01/compute"
|
|
"github.com/Azure/go-autorest/autorest"
|
|
"github.com/hashicorp/packer/builder/azure/common/client"
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
"github.com/hashicorp/packer/packer"
|
|
)
|
|
|
|
func TestStepAttachDisk_Run(t *testing.T) {
|
|
type fields struct {
|
|
GetDiskResponseCode int
|
|
GetDiskResponseBody string
|
|
|
|
attachError error
|
|
waitForDeviceError error
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
want multistep.StepAction
|
|
}{
|
|
{
|
|
name: "HappyPath",
|
|
want: multistep.ActionContinue,
|
|
},
|
|
{
|
|
name: "AttachError",
|
|
fields: fields{
|
|
attachError: errors.New("unit test"),
|
|
},
|
|
want: multistep.ActionHalt,
|
|
},
|
|
{
|
|
name: "WaitForDeviceError",
|
|
fields: fields{
|
|
waitForDeviceError: errors.New("unit test"),
|
|
},
|
|
want: multistep.ActionHalt,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s := &StepAttachDisk{}
|
|
|
|
NewDiskAttacher = func(azcli client.AzureClientSet) DiskAttacher {
|
|
return &fakeDiskAttacher{
|
|
attachError: tt.fields.attachError,
|
|
waitForDeviceError: tt.fields.waitForDeviceError,
|
|
}
|
|
}
|
|
|
|
dm := compute.NewDisksClient("subscriptionId")
|
|
dm.Sender = autorest.SenderFunc(func(r *http.Request) (*http.Response, error) {
|
|
return &http.Response{
|
|
Request: r,
|
|
Body: ioutil.NopCloser(strings.NewReader(tt.fields.GetDiskResponseBody)),
|
|
StatusCode: tt.fields.GetDiskResponseCode,
|
|
}, nil
|
|
})
|
|
|
|
errorBuffer := &strings.Builder{}
|
|
ui := &packer.BasicUi{
|
|
Reader: strings.NewReader(""),
|
|
Writer: ioutil.Discard,
|
|
ErrorWriter: errorBuffer,
|
|
}
|
|
|
|
state := new(multistep.BasicStateBag)
|
|
state.Put("azureclient", &client.AzureClientSetMock{})
|
|
state.Put("ui", ui)
|
|
state.Put(stateBagKey_OSDiskResourceID, "/subscriptions/12345/resourceGroups/group1/providers/Microsoft.Compute/disks/disk1")
|
|
|
|
got := s.Run(context.TODO(), state)
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("StepAttachDisk.Run() = %v, want %v", got, tt.want)
|
|
}
|
|
|
|
if got == multistep.ActionHalt {
|
|
if _, ok := state.GetOk("error"); !ok {
|
|
t.Fatal("Expected 'error' to be set in statebag after failure")
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
type fakeDiskAttacher struct {
|
|
attachError error
|
|
waitForDeviceError error
|
|
}
|
|
|
|
var _ DiskAttacher = &fakeDiskAttacher{}
|
|
|
|
func (da *fakeDiskAttacher) AttachDisk(ctx context.Context, disk string) (lun int32, err error) {
|
|
if da.attachError != nil {
|
|
return 0, da.attachError
|
|
}
|
|
return 3, nil
|
|
}
|
|
|
|
func (da *fakeDiskAttacher) DiskPathForLun(lun int32) string {
|
|
panic("not implemented")
|
|
}
|
|
|
|
func (da *fakeDiskAttacher) WaitForDevice(ctx context.Context, lun int32) (device string, err error) {
|
|
if da.waitForDeviceError != nil {
|
|
return "", da.waitForDeviceError
|
|
}
|
|
if lun == 3 {
|
|
return "/dev/sdq", nil
|
|
}
|
|
panic("expected lun==3")
|
|
}
|
|
|
|
func (da *fakeDiskAttacher) DetachDisk(ctx context.Context, disk string) (err error) {
|
|
panic("not implemented")
|
|
}
|
|
|
|
func (da *fakeDiskAttacher) WaitForDetach(ctx context.Context, diskID string) error {
|
|
panic("not implemented")
|
|
}
|