Filling it back

builder/googlecompute/driver_mock.go

@@ -8,6 +8,7 @@ type DriverMock struct {
 	CreateImageName            string
 	CreateImageDesc            string
 	CreateImageFamily          string
+	CreateImageLabels          map[string]string
 	CreateImageZone            string
 	CreateImageDisk            string
 	CreateImageResultLicenses  []string
@@ -85,6 +86,7 @@ func (d *DriverMock) CreateImage(name, description, family, zone, disk string, i
 	d.CreateImageName = name
 	d.CreateImageDesc = description
 	d.CreateImageFamily = family
+	d.CreateImageLabels = image_labels
 	d.CreateImageZone = zone
 	d.CreateImageDisk = disk
 	if d.CreateImageResultProjectId == "" {
@@ -103,6 +105,7 @@ func (d *DriverMock) CreateImage(name, description, family, zone, disk string, i
 	if resultCh == nil {
 		ch := make(chan *Image, 1)
 		ch <- &Image{
+			Labels:    d.CreateImageLabels,
 			Licenses:  d.CreateImageResultLicenses,
 			Name:      name,
 			ProjectId: d.CreateImageResultProjectId,