Add a COLO migration test for COLO migration and failover. COLO does not support q35 machine at this time.
Signed-off-by: Lukas Straub <[email protected]> --- tests/qtest/meson.build | 7 ++- tests/qtest/migration-test.c | 1 + tests/qtest/migration/colo-tests.c | 115 +++++++++++++++++++++++++++++++++++++ tests/qtest/migration/framework.c | 65 ++++++++++++++++++++- tests/qtest/migration/framework.h | 9 +++ 5 files changed, 195 insertions(+), 2 deletions(-) diff --git a/tests/qtest/meson.build b/tests/qtest/meson.build index 08fba9695b9813dc0b6b6554ef8c40c9615918fa..e68ce6c193ce2a2c244fa072ebb24738380f844a 100644 --- a/tests/qtest/meson.build +++ b/tests/qtest/meson.build @@ -366,6 +366,11 @@ if gnutls.found() endif endif +migration_colo_files = [] +if get_option('replication').allowed() + migration_colo_files = [files('migration/colo-tests.c')] +endif + qtests = { 'aspeed_hace-test': files('aspeed-hace-utils.c', 'aspeed_hace-test.c'), 'aspeed_smc-test': files('aspeed-smc-utils.c', 'aspeed_smc-test.c'), @@ -377,7 +382,7 @@ qtests = { 'migration/migration-util.c') + dbus_vmstate1, 'erst-test': files('erst-test.c'), 'ivshmem-test': [rt, '../../contrib/ivshmem-server/ivshmem-server.c'], - 'migration-test': test_migration_files + migration_tls_files, + 'migration-test': test_migration_files + migration_tls_files + migration_colo_files, 'pxe-test': files('boot-sector.c'), 'pnv-xive2-test': files('pnv-xive2-common.c', 'pnv-xive2-flush-sync.c', 'pnv-xive2-nvpg_bar.c'), diff --git a/tests/qtest/migration-test.c b/tests/qtest/migration-test.c index 08936871741535c926eeac40a7d7c3f461c72fd0..e582f05c7dc2673dbd05a936df8feb6c964b5bbc 100644 --- a/tests/qtest/migration-test.c +++ b/tests/qtest/migration-test.c @@ -55,6 +55,7 @@ int main(int argc, char **argv) migration_test_add_precopy(env); migration_test_add_cpr(env); migration_test_add_misc(env); + migration_test_add_colo(env); ret = g_test_run(); diff --git a/tests/qtest/migration/colo-tests.c b/tests/qtest/migration/colo-tests.c new file mode 100644 index 0000000000000000000000000000000000000000..adec41c1c0473539d02e488b1d0baa663d7743b1 --- /dev/null +++ b/tests/qtest/migration/colo-tests.c @@ -0,0 +1,115 @@ +/* + * SPDX-License-Identifier: GPL-2.0-or-later + * + * QTest testcases for COLO migration + * + * Copyright (c) 2025 Lukas Straub <[email protected]> + * + * This work is licensed under the terms of the GNU GPL, version 2 or later. + * See the COPYING file in the top-level directory. + * + */ + +#include "qemu/osdep.h" +#include "libqtest.h" +#include "migration/framework.h" +#include "migration/migration-qmp.h" +#include "migration/migration-util.h" +#include "qemu/module.h" + +static void test_colo_plain_primary_failover(char *name, MigrateCommon *args) +{ + args->listen_uri = "tcp:127.0.0.1:0"; + args->colo_primary_failover = true; + + test_colo_common(args); +} + +static void test_colo_plain_secondary_failover(char *name, MigrateCommon *args) +{ + args->listen_uri = "tcp:127.0.0.1:0"; + + test_colo_common(args); +} + +static void *hook_start_multifd(QTestState *from, QTestState *to) +{ + return migrate_hook_start_precopy_tcp_multifd_common(from, to, "none"); +} + +static void test_colo_multifd_primary_failover(char *name, MigrateCommon *args) +{ + args->listen_uri = "defer"; + args->start_hook = hook_start_multifd; + args->colo_primary_failover = true; + args->start.caps[MIGRATION_CAPABILITY_MULTIFD] = true; + + test_colo_common(args); +} + +static void test_colo_multifd_secondary_failover(char *name, + MigrateCommon *args) +{ + args->listen_uri = "defer"; + args->start_hook = hook_start_multifd; + args->start.caps[MIGRATION_CAPABILITY_MULTIFD] = true; + + test_colo_common(args); +} + +static void test_colo_plain_primary_failover_checkpoint(char *name, + MigrateCommon *args) +{ + args->colo_failover_during_checkpoint = true; + test_colo_plain_primary_failover(name, args); +} + +static void test_colo_plain_secondary_failover_checkpoint(char *name, + MigrateCommon *args) +{ + args->colo_failover_during_checkpoint = true; + test_colo_plain_secondary_failover(name, args); +} + +static void test_colo_multifd_primary_failover_checkpoint(char *name, + MigrateCommon *args) +{ + args->colo_failover_during_checkpoint = true; + test_colo_multifd_primary_failover(name, args); +} + +static void test_colo_multifd_secondary_failover_checkpoint(char *name, + MigrateCommon *args) +{ + args->colo_failover_during_checkpoint = true; + test_colo_multifd_secondary_failover(name, args); +} + +void migration_test_add_colo(MigrationTestEnv *env) +{ + migration_test_add("/migration/colo/plain/secondary_failover", + test_colo_plain_secondary_failover); + + migration_test_add("/migration/colo/multifd/secondary_failover", + test_colo_multifd_secondary_failover); + + if (!env->full_set) { + return; + } + + migration_test_add("/migration/colo/plain/primary_failover", + test_colo_plain_primary_failover); + + migration_test_add("/migration/colo/multifd/primary_failover", + test_colo_multifd_primary_failover); + + migration_test_add("/migration/colo/plain/primary_failover_checkpoint", + test_colo_plain_primary_failover_checkpoint); + migration_test_add("/migration/colo/plain/secondary_failover_checkpoint", + test_colo_plain_secondary_failover_checkpoint); + + migration_test_add("/migration/colo/multifd/primary_failover_checkpoint", + test_colo_multifd_primary_failover_checkpoint); + migration_test_add("/migration/colo/multifd/secondary_failover_checkpoint", + test_colo_multifd_secondary_failover_checkpoint); +} diff --git a/tests/qtest/migration/framework.c b/tests/qtest/migration/framework.c index 8c1fc6e009f16dc05a47e917167f62e0250ca992..08bca49a8980f9988be9447acf54b17acd56da94 100644 --- a/tests/qtest/migration/framework.c +++ b/tests/qtest/migration/framework.c @@ -315,7 +315,7 @@ int migrate_args(char **from, char **to, const char *uri, MigrateStart *args) if (strcmp(arch, "i386") == 0 || strcmp(arch, "x86_64") == 0) { memory_size = "150M"; - if (g_str_equal(arch, "i386")) { + if (g_str_equal(arch, "i386") || args->is_colo) { machine_alias = "pc"; } else { machine_alias = "q35"; @@ -1068,6 +1068,69 @@ void *migrate_hook_start_precopy_tcp_multifd_common(QTestState *from, return NULL; } +int test_colo_common(MigrateCommon *args) +{ + QTestState *from, *to; + void *data_hook = NULL; + + args->start.oob = true; + args->start.is_colo = true; + args->start.caps[MIGRATION_CAPABILITY_X_COLO] = true; + + if (migrate_start(&from, &to, args->listen_uri, &args->start)) { + return -1; + } + + migrate_set_parameter_int(from, "x-checkpoint-delay", 300); + + if (args->start_hook) { + data_hook = args->start_hook(from, to); + } + + migrate_ensure_converge(from); + wait_for_serial("src_serial"); + + migrate_qmp(from, to, args->connect_uri, NULL, "{}"); + + wait_for_migration_status(from, "colo", NULL); + wait_for_resume(to, &dst_state); + + wait_for_serial("src_serial"); + wait_for_serial("dest_serial"); + + /* wait for 3 checkpoints */ + for (int i = 0; i < 3; i++) { + qtest_qmp_eventwait(to, "RESUME"); + wait_for_serial("src_serial"); + wait_for_serial("dest_serial"); + } + + if (args->colo_failover_during_checkpoint) { + qtest_qmp_eventwait(to, "STOP"); + } + if (args->colo_primary_failover) { + qtest_qmp_assert_success(from, "{'exec-oob': 'yank', 'id': 'yank-cmd', " + "'arguments': {'instances':" + "[{'type': 'migration'}]}}"); + qtest_qmp_assert_success(from, "{'execute': 'x-colo-lost-heartbeat'}"); + wait_for_serial("src_serial"); + } else { + qtest_qmp_assert_success(to, "{'exec-oob': 'yank', 'id': 'yank-cmd', " + "'arguments': {'instances':" + "[{'type': 'migration'}]}}"); + qtest_qmp_assert_success(to, "{'execute': 'x-colo-lost-heartbeat'}"); + wait_for_serial("dest_serial"); + } + + if (args->end_hook) { + args->end_hook(from, to, data_hook); + } + + migrate_end(from, to, !args->colo_primary_failover); + + return 0; +} + QTestMigrationState *get_src(void) { return &src_state; diff --git a/tests/qtest/migration/framework.h b/tests/qtest/migration/framework.h index 2ea13e7758697550b5531737e66d6d939dd800d1..1da532569d8c5941c99d83ae7da88ccab2bdcbe2 100644 --- a/tests/qtest/migration/framework.h +++ b/tests/qtest/migration/framework.h @@ -230,6 +230,9 @@ typedef struct { */ bool live; + bool colo_primary_failover; + bool colo_failover_during_checkpoint; + /* Postcopy specific fields */ void *postcopy_data; PostcopyRecoveryFailStage postcopy_recovery_fail_stage; @@ -248,6 +251,7 @@ void test_postcopy_common(MigrateCommon *args); void test_postcopy_recovery_common(MigrateCommon *args); int test_precopy_common(MigrateCommon *args); void test_file_common(MigrateCommon *args, bool stop_src); +int test_colo_common(MigrateCommon *args); void *migrate_hook_start_precopy_tcp_multifd_common(QTestState *from, QTestState *to, const char *method); @@ -267,5 +271,10 @@ void migration_test_add_file(MigrationTestEnv *env); void migration_test_add_precopy(MigrationTestEnv *env); void migration_test_add_cpr(MigrationTestEnv *env); void migration_test_add_misc(MigrationTestEnv *env); +#ifdef CONFIG_REPLICATION +void migration_test_add_colo(MigrationTestEnv *env); +#else +static inline void migration_test_add_colo(MigrationTestEnv *env) {}; +#endif #endif /* TEST_FRAMEWORK_H */ -- 2.39.5
