Add verify_os_architecture_is_64bit test case

Author: lubaihua33

lisa/base_tools/uname.py

@@ -12,6 +12,7 @@
 
 if TYPE_CHECKING:
     from lisa.node import Node
+    from lisa.operating_system import CpuArchitecture
 
 
 @dataclass
@@ -80,6 +81,21 @@ def get_linux_information(
 
         return result
 
+    def get_machine_architecture(self, force_run: bool = False) -> "CpuArchitecture":
+        # To avoid circular import
+        from lisa.operating_system import CpuArchitecture
+
+        arch_map = {
+            "x86_64": CpuArchitecture.X64,
+            "amd64": CpuArchitecture.X64,
+            "aarch64": CpuArchitecture.ARM64,
+            "arm64": CpuArchitecture.ARM64,
+            "i386": CpuArchitecture.I386,
+        }
+        self.initialize()
+        arch_str = self.run("-m", force_run=force_run).stdout.strip().lower()
+        return arch_map.get(arch_str, CpuArchitecture.UNKNOWN)
+
 
 class FreeBSDUname(Uname):
     _key_info_pattern = re.compile(

lisa/operating_system.py

@@ -77,6 +77,7 @@ class CpuArchitecture(str, Enum):
     X64 = "x86_64"
     ARM64 = "aarch64"
     I386 = "i386"
+    UNKNOWN = "unknown"
 
 
 class AzureCoreRepo(str, Enum):

microsoft/testsuites/core/azure_image_standard.py

@@ -17,6 +17,7 @@
     schema,
     simple_requirement,
 )
+from lisa.base_tools.uname import Uname
 from lisa.features import Disk
 from lisa.operating_system import (
     BSD,
@@ -1365,6 +1366,28 @@ def verify_openssl_version(self, node: Node) -> None:
             library_name="OpenSSL",
         )
 
+    @TestCaseMetadata(
+        description="""
+        This test verifies that the Linux operating system has a 64-bit architecture.
+
+        Steps:
+        1. Retrieve the OS architecture using the Uname tool.
+        2. Verify that the architecture is either x86_64 (AMD64) or aarch64 (ARM64).
+        3. Fail the test if the architecture is not 64-bit.
+        """,
+        priority=1,
+        requirement=simple_requirement(supported_platform_type=[AZURE]),
+    )
+    def verify_azure_64bit_os(self, node: Node) -> None:
+        uname_tool = node.tools[Uname]
+        arch = uname_tool.get_machine_architecture()
+        arch_64bit = [CpuArchitecture.X64, CpuArchitecture.ARM64]
+        if arch not in arch_64bit:
+            raise LisaException(
+                f"Architecture '{arch.value}' is not supported. Azure only supports "
+                f"64-bit architectures: {', '.join(str(a.value) for a in arch_64bit)}."
+            )
+
     def _check_version_by_pattern_value(
         self,
         node: Node,