From b7532538fee6746511c9f61e9798e219bdfaa45d Mon Sep 17 00:00:00 2001 From: Borys Date: Mon, 30 Dec 2024 14:02:12 +0200 Subject: [PATCH] test: fix test_network_disconnect_during_migration (#4378) --- tests/dragonfly/cluster_test.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/dragonfly/cluster_test.py b/tests/dragonfly/cluster_test.py index 230dc710d..b90d48665 100644 --- a/tests/dragonfly/cluster_test.py +++ b/tests/dragonfly/cluster_test.py @@ -1433,8 +1433,7 @@ async def test_migration_with_key_ttl(df_factory): assert await nodes[1].client.execute_command("stick k_sticky") == 0 -@pytest.mark.skip("Flaky test") -@dfly_args({"proactor_threads": 4, "cluster_mode": "yes"}) +@dfly_args({"proactor_threads": 4, "cluster_mode": "yes", "migration_finalization_timeout_ms": 5}) async def test_network_disconnect_during_migration(df_factory): instances = [ df_factory.create( @@ -1473,7 +1472,7 @@ async def test_network_disconnect_during_migration(df_factory): await nodes[0].admin_client.execute_command("DFLYCLUSTER", "SLOT-MIGRATION-STATUS") ) - await wait_for_status(nodes[0].admin_client, nodes[1].id, "SYNC") + await wait_for_status(nodes[0].admin_client, nodes[1].id, "SYNC", 20) finally: await proxy.close(task)