From 7cbd57faa677e40b2cd3ed9a0068847074b5ba74 Mon Sep 17 00:00:00 2001
From: Peter Mounce <petermounce@improbable.io>
Date: Tue, 19 Sep 2017 11:40:01 +0100
Subject: [PATCH] Filling it back

---
 builder/googlecompute/driver_mock.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/builder/googlecompute/driver_mock.go b/builder/googlecompute/driver_mock.go
index 89ea0b8de..895775da5 100644
--- a/builder/googlecompute/driver_mock.go
+++ b/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,