tests: add test to check tdx flag in CPU info (#246)

canonical · Oct 10, 2024 · d9d2dbb · d9d2dbb
1 parent 39cce6b
commit d9d2dbb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/tests/lib/tdx-tools/src/tdxtools/host.py b/tests/lib/tdx-tools/src/tdxtools/host.py
@@ -25,7 +25,7 @@ def support_tdx():
     """
     Check whether support TDX in CPU info
     """
-    return 'tdx' in cpuinfo.get_cpu_info()['flags']
+    return 'tdx_host_platform' in cpuinfo.get_cpu_info()['flags']
 
 
 def support_sgx():

diff --git a/tests/tests/test_host_tdx_software.py b/tests/tests/test_host_tdx_software.py
@@ -19,6 +19,14 @@
 
 import re
 import subprocess
+import tdxtools
+
+def test_host_tdx_cpu():
+    """
+    Check that the CPU has TDX support enabled
+    (the flag tdx_host_platform is present)
+    """
+    assert tdxtools.host.support_tdx()
 
 def test_host_tdx_software():