|
1 |
| -from virttest import env_process, error_context |
| 1 | +from virttest import env_process, error_context, qemu_migration |
2 | 2 | from virttest.qemu_monitor import QMPCmdError
|
3 | 3 |
|
4 | 4 | from provider import win_driver_utils
|
@@ -34,6 +34,39 @@ def get_buses_and_serial_devices(vm, params, char_devices, serials):
|
34 | 34 | return buses, serial_devices
|
35 | 35 |
|
36 | 36 |
|
| 37 | +@error_context.context_aware |
| 38 | +def reboot_guest(test, params, vm, session): |
| 39 | + """ |
| 40 | + Reboot guest from system_reset or shell. |
| 41 | + """ |
| 42 | + |
| 43 | + vm.reboot(session, method=params["reboot_method"]) |
| 44 | + |
| 45 | + |
| 46 | +@error_context.context_aware |
| 47 | +def shutdown_guest(test, params, vm, session): |
| 48 | + """ |
| 49 | + Shutdown guest via system_powerdown or shell. |
| 50 | + """ |
| 51 | + |
| 52 | + if params.get("shutdown_method") == "shell": |
| 53 | + session.sendline(params["shutdown_command"]) |
| 54 | + elif params.get("shutdown_method") == "system_powerdown": |
| 55 | + vm.monitor.system_powerdown() |
| 56 | + if not vm.wait_for_shutdown(int(params.get("shutdown_timeout", 360))): |
| 57 | + test.fail("guest refuses to go down") |
| 58 | + |
| 59 | + |
| 60 | +@error_context.context_aware |
| 61 | +def live_migration_guest(test, params, vm, session): |
| 62 | + """ |
| 63 | + Run migrate_set_speed, then migrate guest. |
| 64 | + """ |
| 65 | + |
| 66 | + qemu_migration.set_speed(vm, params.get("mig_speed", "1G")) |
| 67 | + vm.migrate() |
| 68 | + |
| 69 | + |
37 | 70 | @error_context.context_aware
|
38 | 71 | def run(test, params, env):
|
39 | 72 | """
|
|
0 commit comments