Skip to content

tests: fix load balancing policy tests for Scylla Raft topology #779

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
mykaul wants to merge 4 commits into
base: master
Choose a base branch
from
Draft

tests: fix load balancing policy tests for Scylla Raft topology #779

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
3965951
tests: fix test_roundrobin decommission with dead node
mykaul Mar 31, 2026
a883948
tests: fix test_roundrobin_two_dcs bootstrap with dead node
mykaul Mar 31, 2026
d652713
tests: fix test_roundrobin_two_dcs_2 bootstrap with dead node
mykaul Mar 31, 2026
69611fa
tests: fix test_token_aware_with_rf_2 for Scylla shard-aware routing
mykaul Mar 31, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 9 additions & 10 deletions tests/integration/long/test_loadbalancingpolicies.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -215,10 +215,10 @@ def test_roundrobin(self):
self.coordinator_stats.assert_query_count_equals(2, 6)
self.coordinator_stats.assert_query_count_equals(3, 0)

start(3) # Restart before decommission (Raft rejects ops with dead nodes)
self._wait_for_nodes_up([3], cluster)
decommission(1)
start(3)
self._wait_for_nodes_down([1], cluster)
self._wait_for_nodes_up([3], cluster)

self.coordinator_stats.reset_counts()
self._query(session, keyspace)
Expand All @@ -243,14 +243,13 @@ def test_roundrobin_two_dcs(self):
self.coordinator_stats.assert_query_count_equals(3, 3)
self.coordinator_stats.assert_query_count_equals(4, 3)

bootstrap(5, 'dc3') # Bootstrap before force_stop (Raft rejects ops with dead nodes)
self._wait_for_nodes_up([5], cluster)
force_stop(1)
bootstrap(5, 'dc3')

# reset control connection
self._insert(session, keyspace, count=1000)

self._wait_for_nodes_up([5], cluster)

self.coordinator_stats.reset_counts()
self._query(session, keyspace)

Expand All @@ -276,14 +275,13 @@ def test_roundrobin_two_dcs_2(self):
self.coordinator_stats.assert_query_count_equals(3, 3)
self.coordinator_stats.assert_query_count_equals(4, 3)

bootstrap(5, 'dc1') # Bootstrap before force_stop (Raft rejects ops with dead nodes)
self._wait_for_nodes_up([5], cluster)
force_stop(1)
bootstrap(5, 'dc1')

# reset control connection
self._insert(session, keyspace, count=1000)

self._wait_for_nodes_up([5], cluster)

self.coordinator_stats.reset_counts()
self._query(session, keyspace)

Expand Down Expand Up @@ -516,8 +514,9 @@ def test_token_aware_with_rf_2(self, use_prepared=False):
self._query(session, keyspace)

self.coordinator_stats.assert_query_count_equals(1, 0)
self.coordinator_stats.assert_query_count_equals(2, 12)
self.coordinator_stats.assert_query_count_equals(3, 0)
# Scylla may distribute queries across both replicas with shard-aware routing
queried = self.coordinator_stats.get_query_count(2) + self.coordinator_stats.get_query_count(3)
assert queried == 12, "Expected 12 queries to replicas, got %d" % queried

self.coordinator_stats.reset_counts()
stop(2)
Expand Down
Loading