From 810085c01820e005666e6b357b70a890485febe0 Mon Sep 17 00:00:00 2001
From: David <djoshy@redhat.com>
Date: Wed, 10 Sep 2025 10:22:00 -0400
Subject: [PATCH] testutils: add image option to resourcebuilder

---
 .../resourcebuilder/machine/v1beta1/azure_provider_spec.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/testutils/resourcebuilder/machine/v1beta1/azure_provider_spec.go b/testutils/resourcebuilder/machine/v1beta1/azure_provider_spec.go
index f2057cbf8..a578bcaa2 100644
--- a/testutils/resourcebuilder/machine/v1beta1/azure_provider_spec.go
+++ b/testutils/resourcebuilder/machine/v1beta1/azure_provider_spec.go
@@ -41,6 +41,7 @@ type AzureProviderSpecBuilder struct {
 	vmSize               string
 	zone                 string
 	subnet               string
+	image                machinev1beta1.Image
 }
 
 // Build builds a new Azure machine config based on the configuration provided.
@@ -120,3 +121,9 @@ func (m AzureProviderSpecBuilder) WithSubnet(subnet string) AzureProviderSpecBui
 	m.subnet = subnet
 	return m
 }
+
+// WithImage sets the image for the Azure machine config builder.
+func (m AzureProviderSpecBuilder) WithImage(image machinev1beta1.Image) AzureProviderSpecBuilder {
+	m.image = image
+	return m
+}