Signed-off-by: Juan Quintela <quint...@redhat.com> --- tests/qtest/migration-test.c | 20 +++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-)
diff --git a/tests/qtest/migration-test.c b/tests/qtest/migration-test.c index 6438379dcf..0afe871cfb 100644 --- a/tests/qtest/migration-test.c +++ b/tests/qtest/migration-test.c @@ -667,12 +667,22 @@ static void migrate_postcopy_start(GuestState *from, GuestState *to) static void do_migrate(GuestState *from, GuestState *to, const gchar *uri) { - if (!uri) { - g_autofree char *tcp_uri = - migrate_get_socket_address(to->qs, "socket-address"); - migrate_qmp(from->qs, tcp_uri, "{}"); + if (to->uri) { + if (strncmp(to->uri, "tcp:", strlen("tcp:")) == 0) { + g_autofree char *tcp_uri = + migrate_get_socket_address(to->qs, "socket-address"); + migrate_qmp(from->qs, tcp_uri, "{}"); + } else { + migrate_qmp(from->qs, to->uri, "{}"); + } } else { - migrate_qmp(from->qs, uri, "{}"); + if (!uri) { + g_autofree char *tcp_uri = + migrate_get_socket_address(to->qs, "socket-address"); + migrate_qmp(from->qs, tcp_uri, "{}"); + } else { + migrate_qmp(from->qs, uri, "{}"); + } } } -- 2.40.1