From 5a8b8ed624856a40fb7babf62c8d85a4040ac633 Mon Sep 17 00:00:00 2001 From: kanchansenlaskar Date: Wed, 13 Mar 2024 18:20:59 +0530 Subject: [PATCH] Use apt-get to install mock --- .../vm_extensions/vmsnapshot_extension.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/microsoft/testsuites/vm_extensions/vmsnapshot_extension.py b/microsoft/testsuites/vm_extensions/vmsnapshot_extension.py index e69847abae..dbfc1ad440 100644 --- a/microsoft/testsuites/vm_extensions/vmsnapshot_extension.py +++ b/microsoft/testsuites/vm_extensions/vmsnapshot_extension.py @@ -19,6 +19,7 @@ TestSuiteMetadata, simple_requirement, ) +from lisa.operating_system import Debian from lisa.sut_orchestrator import AZURE from lisa.sut_orchestrator.azure.common import ( AzureNodeSchema, @@ -144,10 +145,16 @@ def verify_exclude_disk_support_restore_point( break # installing all the required packages - # install python3 and pip + # install python3 python = node.tools[Python] - pip = node.tools[Pip] - pip.install_packages("mock") + if isinstance(node.os, Debian): + package_name = "python3-mock" + node.os.install_packages(package_name) + else: + # install pip + pip = node.tools[Pip] + pip.install_packages("mock") + # copy the file into the vm self._copy_to_node(node, "handle.txt") assert extension_dir, "Unable to find the extension directory."