From d6d511817013baf8ea1b02bc642b2a0ed8bc62c6 Mon Sep 17 00:00:00 2001 From: Lili Deng Date: Fri, 31 Jan 2025 10:32:59 +0800 Subject: [PATCH] Fix runtime error `ModuleNotFoundError: No module named 'libvirt'` --- microsoft/testsuites/cvm/cvm_attestation.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/microsoft/testsuites/cvm/cvm_attestation.py b/microsoft/testsuites/cvm/cvm_attestation.py index 3b1e73d60e..6023daac1b 100644 --- a/microsoft/testsuites/cvm/cvm_attestation.py +++ b/microsoft/testsuites/cvm/cvm_attestation.py @@ -2,7 +2,7 @@ # Licensed under the MIT license. import base64 from pathlib import Path -from typing import Any, Dict +from typing import TYPE_CHECKING, Any, Dict from lisa import ( Environment, @@ -15,7 +15,6 @@ from lisa.features.security_profile import CvmEnabled from lisa.operating_system import CBLMariner, Ubuntu from lisa.sut_orchestrator import AZURE, CLOUD_HYPERVISOR -from lisa.sut_orchestrator.libvirt.context import NodeContext from lisa.testsuite import TestResult, simple_requirement from lisa.tools import Ls, Lscpu from lisa.tools.lscpu import CpuType @@ -26,6 +25,9 @@ SnpGuest, ) +if TYPE_CHECKING: + from lisa.sut_orchestrator.libvirt.context import NodeContext + @TestSuiteMetadata( area="cvm", @@ -136,7 +138,7 @@ def verify_nested_cvm_attestation_report( host_data, ) - def _get_host_data(self, node_context: NodeContext) -> str: + def _get_host_data(self, node_context: "NodeContext") -> str: # Based on libvirt version our libvirt platform will set # either plain text or b64 encoded string as host data. # We need to decode it as this test would get host_data