From e01eea8f313965f713bb437019567e63ada6e5f6 Mon Sep 17 00:00:00 2001
From: Matthew Booth <mbooth@redhat.com>
Date: Fri, 6 Jun 2025 09:34:13 +0100
Subject: [PATCH] Support data disks on ASH

Co-Authored-By: George Campbell <gcampbell@palantir.com>
---
 .../azure/services/virtualmachines/virtualmachines_stack.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pkg/cloud/azure/services/virtualmachines/virtualmachines_stack.go b/pkg/cloud/azure/services/virtualmachines/virtualmachines_stack.go
index 681d7d74f..af9020bd4 100644
--- a/pkg/cloud/azure/services/virtualmachines/virtualmachines_stack.go
+++ b/pkg/cloud/azure/services/virtualmachines/virtualmachines_stack.go
@@ -44,6 +44,11 @@ func (s *Service) deriveVirtualMachineParametersStackHub(vmSpec *Spec, nicID str
 		}
 	}
 
+	dataDisks, err := generateDataDisks(vmSpec)
+	if err != nil {
+		return nil, fmt.Errorf("failed to generate data disk spec: %w", err)
+	}
+
 	virtualMachine := &compute.VirtualMachine{
 		Location: ptr.To(s.Scope.MachineConfig.Location),
 		Tags:     vmSpec.Tags,
@@ -62,6 +67,7 @@ func (s *Service) deriveVirtualMachineParametersStackHub(vmSpec *Spec, nicID str
 						StorageAccountType: ptr.To(compute.StorageAccountTypes(vmSpec.OSDisk.ManagedDisk.StorageAccountType)),
 					},
 				},
+				DataDisks: dataDisks,
 			},
 			OSProfile: osProfile,
 			NetworkProfile: &compute.NetworkProfile{