step_create_disks_test.go

//  Copyright 2017 Google Inc. All Rights Reserved.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.

package daisy

import (
	"context"
	"testing"

	daisyCompute "github.com/GoogleCloudPlatform/compute-image-tools/daisy/compute"
	"google.golang.org/api/compute/v1"
)

func TestCreateDisksRun(t *testing.T) {
	ctx := context.Background()
	w := testWorkflow()
	s := &Step{w: w}
	w.images.m = map[string]*Resource{"i1": {RealName: "i1", link: "i1link"}}
	w.snapshots.m = map[string]*Resource{"ss1": {RealName: "ss1", link: "ss1link"}}

	e := Errf("error")
	quotaExceededErr := Errf("Some error\nCode: QUOTA_EXCEEDED\nMessage: some message.")
	tests := []struct {
		desc                 string
		d                    compute.Disk
		wantD                compute.Disk
		clientErr            []error
		wantErr              DError
		fallbackToPdStandard bool
	}{
		{"blank case", compute.Disk{}, compute.Disk{}, nil, nil, false},
		{"resolve source image case", compute.Disk{SourceImage: "i1"}, compute.Disk{SourceImage: "i1link"}, nil, nil, false},
		{"client error case", compute.Disk{}, compute.Disk{}, []error{e}, e, false},
		{"not fallback to pd-standard case", compute.Disk{Type: "prefix/pd-ssd"}, compute.Disk{Type: "prefix/pd-ssd"}, []error{e}, e, true},
		{"fallback to pd-standard case", compute.Disk{Type: "prefix/pd-ssd"}, compute.Disk{Type: "prefix/pd-standard"}, []error{quotaExceededErr, nil}, nil, true},
		{"create from snapshot case", compute.Disk{SourceSnapshot: "ss1"}, compute.Disk{SourceSnapshot: "ss1link"}, nil, nil, false},
	}
	for _, tt := range tests {
		var gotD compute.Disk
		var errIndex = 0
		fake := func(_, _ string, d *compute.Disk) error {
			gotD = *d
			if tt.clientErr == nil {
				return nil
			}
			var ret = tt.clientErr[errIndex]
			errIndex = (errIndex + 1) % len(tt.clientErr)
			return ret
		}
		w.ComputeClient = &daisyCompute.TestClient{CreateDiskFn: fake}
		cds := &CreateDisks{{Disk: tt.d, FallbackToPdStandard: tt.fallbackToPdStandard}}
		if err := cds.run(ctx, s); err != tt.wantErr {
			t.Errorf("%s: unexpected error returned, got: %v, want: %v", tt.desc, err, tt.wantErr)
		}
		if diffRes := diff(gotD, tt.wantD, 0); diffRes != "" {
			t.Errorf("%s: client got incorrect disk, got: %v, want: %v", tt.desc, gotD, tt.wantD)
		}
	}
}