diff --git a/src/runtime/cmd/kata-runtime/kata-check_amd64_test.go b/src/runtime/cmd/kata-runtime/kata-check_amd64_test.go index 24fdf2c338..f2b1077786 100644 --- a/src/runtime/cmd/kata-runtime/kata-check_amd64_test.go +++ b/src/runtime/cmd/kata-runtime/kata-check_amd64_test.go @@ -467,7 +467,12 @@ func TestSetCPUtype(t *testing.T) { assert.NotEmpty(archRequiredCPUAttribs) assert.NotEmpty(archRequiredKernelModules) - assert.Equal(archRequiredCPUFlags["vmx"], "Virtualization support") + cpuType = getCPUtype() + if cpuType == cpuTypeIntel { + assert.Equal(archRequiredCPUFlags["vmx"], "Virtualization support") + } else if cpuType == cpuTypeAMD { + assert.Equal(archRequiredCPUFlags["svm"], "Virtualization support") + } _, ok := archRequiredKernelModules["kvm"] assert.True(ok) diff --git a/src/runtime/cmd/kata-runtime/kata-env_amd64_test.go b/src/runtime/cmd/kata-runtime/kata-env_amd64_test.go index 7c1fe849db..97245a22bd 100644 --- a/src/runtime/cmd/kata-runtime/kata-env_amd64_test.go +++ b/src/runtime/cmd/kata-runtime/kata-env_amd64_test.go @@ -55,7 +55,12 @@ func TestEnvGetEnvInfoSetsCPUType(t *testing.T) { assert.NotEmpty(archRequiredCPUAttribs) assert.NotEmpty(archRequiredKernelModules) - assert.Equal(archRequiredCPUFlags["vmx"], "Virtualization support") + cpuType = getCPUtype() + if cpuType == cpuTypeIntel { + assert.Equal(archRequiredCPUFlags["vmx"], "Virtualization support") + } else if cpuType == cpuTypeAMD { + assert.Equal(archRequiredCPUFlags["svm"], "Virtualization support") + } _, ok := archRequiredKernelModules["kvm"] assert.True(ok)