Skip to content

Commit e844cab

Browse files
lubaihua33LiliDeng
authored andcommitted
Add gallery_image_disk_controller_types in SIG transformer
1 parent afdea68 commit e844cab

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

lisa/sut_orchestrator/azure/transformers.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -473,6 +473,7 @@ class SigTransformerSchema(schema.Transformer):
473473
- westus3
474474
- westus2
475475
gallery_image_hyperv_generation: 2
476+
gallery_image_disk_controller_types: NVMe
476477
gallery_image_name: image_name
477478
gallery_image_architecture: Arm64
478479
gallery_image_fullname: Microsoft Linux arm64 0.0.1
@@ -515,6 +516,9 @@ class SigTransformerSchema(schema.Transformer):
515516
)
516517
# image security type, can be empty, TrustedLaunch
517518
gallery_image_securitytype: str = ""
519+
# Disk controller type feature for the gallery image definition.
520+
# Common values: NVMe, SCSI, or comma-separated (e.g. SCSI,NVMe)
521+
gallery_image_disk_controller_types: str = ""
518522
gallery_image_osstate: str = field(
519523
default="Generalized",
520524
metadata=field_metadata(
@@ -629,8 +633,13 @@ def _internal_run(self) -> Dict[str, Any]:
629633
"architecture", runbook.gallery_image_architecture
630634
)
631635
features.pop("os_type", None)
632-
elif runbook.gallery_image_securitytype:
633-
features = {"SecurityType": runbook.gallery_image_securitytype}
636+
else:
637+
if runbook.gallery_image_securitytype:
638+
features["SecurityType"] = runbook.gallery_image_securitytype
639+
640+
disk_controller_types = runbook.gallery_image_disk_controller_types
641+
if disk_controller_types:
642+
features["DiskControllerTypes"] = disk_controller_types
634643

635644
(
636645
gallery_image_publisher,

0 commit comments

Comments
 (0)