Make OSDiskSizeGB actually optional

paulmey · Jun 3, 2019 · 2e07c84 · 2e07c84
1 parent 7b6ed8c
commit 2e07c84
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/builder/azure/chroot/step_create_new_disk.go b/builder/azure/chroot/step_create_new_disk.go
@@ -16,7 +16,7 @@ var _ multistep.Step = &StepCreateNewDisk{}
 
 type StepCreateNewDisk struct {
 	SubscriptionID, ResourceGroup, DiskName string
-	DiskSizeGB                              int32
+	DiskSizeGB                              int32 // optional, ignored if 0
 	DiskStorageAccountType                  string
 	HyperVGeneration                        string
 	Location                                string
@@ -46,10 +46,14 @@ func (s StepCreateNewDisk) Run(ctx context.Context, state multistep.StateBag) mu
 			OsType:           "Linux",
 			HyperVGeneration: compute.HyperVGeneration(s.HyperVGeneration),
 			CreationData:     &compute.CreationData{},
-			DiskSizeGB:       to.Int32Ptr(s.DiskSizeGB),
 		},
 		//Tags: map[string]*string{
 	}
+
+	if s.DiskSizeGB > 0 {
+		disk.DiskProperties.DiskSizeGB = to.Int32Ptr(s.DiskSizeGB)
+	}
+
 	if s.PlatformImage == nil {
 		disk.CreationData.CreateOption = compute.Empty
 	} else {