Set SkipGeneralize to false

hashicorp · Apr 20, 2023 · fc2d64b · fc2d64b
1 parent 2c79b6f
commit fc2d64b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/builder/azure/arm/builder.go b/builder/azure/arm/builder.go
@@ -512,6 +512,8 @@ func (b *Builder) configureStateBag(stateBag multistep.StateBag) {
 	stateBag.Put(constants.ArmKeepOSDisk, b.config.KeepOSDisk)
 
 	stateBag.Put(constants.ArmIsSIGImage, b.config.isPublishToSIG())
+	// Set SkipGeneralization as false so that we can pull it from the state later even if we're not publishing to SIG
+	stateBag.Put(constants.ArmSharedImageGalleryDestinationSkipGeneralization, false)
 	if b.config.isPublishToSIG() {
 		stateBag.Put(constants.ArmManagedImageSigPublishResourceGroup, b.config.SharedGalleryDestination.SigDestinationResourceGroup)
 		stateBag.Put(constants.ArmManagedImageSharedGalleryName, b.config.SharedGalleryDestination.SigDestinationGalleryName)

diff --git a/builder/azure/arm/step_capture_image_test.go b/builder/azure/arm/step_capture_image_test.go
@@ -136,6 +136,7 @@ func createTestStateBagStepCaptureImage() multistep.StateBag {
 	stateBag.Put(constants.ArmManagedImageName, "")
 	stateBag.Put(constants.ArmImageParameters, &compute.Image{})
 	stateBag.Put(constants.ArmIsSIGImage, false)
+	stateBag.Put(constants.ArmSharedImageGalleryDestinationSkipGeneralization, false)
 
 	return stateBag
 }