diff --git a/tests/integration/ha_tests/test_ha.py b/tests/integration/ha_tests/test_ha.py index 9edd2b812..f5102eef7 100644 --- a/tests/integration/ha_tests/test_ha.py +++ b/tests/integration/ha_tests/test_ha.py @@ -662,17 +662,17 @@ async def test_scale_up_down(ops_test: OpsTest, continuous_writes): scales = [3, -3, 4, -4, 5, -5, 6, -6, 7, -7] for count in scales: await scale_and_verify(ops_test, count=count) - await _verify_writes(ops_test) + await _verify_writes(ops_test, continuous_writes) @pytest.mark.abort_on_fail @pytest.mark.unstable -async def test_scale_up_down_removing_leader(ops_test: OpsTest): +async def test_scale_up_down_removing_leader(ops_test: OpsTest, continuous_writes): """Scale up and down the application and verify the replica set is healthy.""" scales = [3, -3, 4, -4, 5, -5, 6, -6, 7, -7] for count in scales: await scale_and_verify(ops_test, count=count, remove_leader=True) - await _verify_writes(ops_test) + await _verify_writes(ops_test, continuous_writes) async def scale_and_verify(ops_test: OpsTest, count: int, remove_leader: bool = False):