From 655e024a276105add2f238d6b72dd4b1924e9fab Mon Sep 17 00:00:00 2001 From: mayankdaruka-msft <133302048+mayankdaruka-msft@users.noreply.github.com> Date: Sun, 6 Aug 2023 18:13:28 -0700 Subject: [PATCH] Initial commit to enable running tests on ARM64 Mariner in canary (#2881) --- .../vm_extensions/runtime_extensions/run_commandv2.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/microsoft/testsuites/vm_extensions/runtime_extensions/run_commandv2.py b/microsoft/testsuites/vm_extensions/runtime_extensions/run_commandv2.py index 0b0260f6a7..b2c8eed02e 100644 --- a/microsoft/testsuites/vm_extensions/runtime_extensions/run_commandv2.py +++ b/microsoft/testsuites/vm_extensions/runtime_extensions/run_commandv2.py @@ -15,8 +15,9 @@ simple_requirement, ) from lisa.environment import Environment -from lisa.operating_system import BSD, CpuArchitecture +from lisa.operating_system import BSD, CBLMariner, CpuArchitecture from lisa.sut_orchestrator import AZURE +from lisa.sut_orchestrator.azure.common import AzureNodeSchema from lisa.sut_orchestrator.azure.features import AzureExtension from lisa.util import SkippedException from microsoft.testsuites.vm_extensions.runtime_extensions.common import ( @@ -30,6 +31,13 @@ def _check_architecture_supported(node: Node) -> None: arch = node.os.get_kernel_information().hardware_platform # type: ignore if arch == CpuArchitecture.ARM64: + # Support RCv2 on ARM64 Mariner in Canary regions + if isinstance(node.os, CBLMariner): + node_context = node.capability.get_extended_runbook(AzureNodeSchema, AZURE) + canary_locations = ["centraluseuap", "eastus2euap"] + if node_context.location in canary_locations: + return + raise SkippedException("RunCommandv2 Extension not published on ARM64.")