Index: client/tests/kvm/tests/shutdown.py |
diff --git a/client/tests/kvm/tests/shutdown.py b/client/tests/kvm/tests/shutdown.py |
index bfc5477b5b48610cece2fb750308664b353979cf..fc0407fdb03e90f909f7753049e886f040957595 100644 |
--- a/client/tests/kvm/tests/shutdown.py |
+++ b/client/tests/kvm/tests/shutdown.py |
@@ -15,9 +15,10 @@ def run_shutdown(test, params, env): |
@param params: Dictionary with the test parameters |
@param env: Dictionary with test environment |
""" |
- vm = kvm_test_utils.get_living_vm(env, params.get("main_vm")) |
+ vm = env.get_vm(params["main_vm"]) |
+ vm.verify_alive() |
timeout = int(params.get("login_timeout", 360)) |
- session = kvm_test_utils.wait_for_login(vm, timeout=timeout) |
+ session = vm.wait_for_login(timeout=timeout) |
try: |
if params.get("shutdown_method") == "shell": |