Skip to content

Commit ba53f59

Browse files
Lorak-mmkdkropachev
authored andcommitted
assert_query_count_equals: Use pytest.fail
1 parent ae16aa1 commit ba53f59

File tree

3 files changed

+109
-108
lines changed

3 files changed

+109
-108
lines changed

tests/integration/long/test_consistency.py

Lines changed: 23 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -101,9 +101,9 @@ def _assert_reads_succeed(self, session, keyspace, consistency_levels, expected_
101101
self._query(session, keyspace, 1, cl)
102102
for i in range(3):
103103
if i == expected_reader:
104-
self.coordinator_stats.assert_query_count_equals(self, i, 1)
104+
self.coordinator_stats.assert_query_count_equals(i, 1)
105105
else:
106-
self.coordinator_stats.assert_query_count_equals(self, i, 0)
106+
self.coordinator_stats.assert_query_count_equals(i, 0)
107107
except Exception as e:
108108
self._cl_failure(cl, e)
109109

@@ -136,9 +136,9 @@ def _test_tokenaware_one_node_down(self, keyspace, rf, accepted):
136136
create_schema(cluster, session, keyspace, replication_factor=rf)
137137
self._insert(session, keyspace, count=1)
138138
self._query(session, keyspace, count=1)
139-
self.coordinator_stats.assert_query_count_equals(self, 1, 0)
140-
self.coordinator_stats.assert_query_count_equals(self, 2, 1)
141-
self.coordinator_stats.assert_query_count_equals(self, 3, 0)
139+
self.coordinator_stats.assert_query_count_equals(1, 0)
140+
self.coordinator_stats.assert_query_count_equals(2, 1)
141+
self.coordinator_stats.assert_query_count_equals(3, 0)
142142

143143
try:
144144
force_stop(2)
@@ -188,9 +188,9 @@ def test_rfthree_tokenaware_none_down(self):
188188
create_schema(cluster, session, keyspace, replication_factor=3)
189189
self._insert(session, keyspace, count=1)
190190
self._query(session, keyspace, count=1)
191-
self.coordinator_stats.assert_query_count_equals(self, 1, 0)
192-
self.coordinator_stats.assert_query_count_equals(self, 2, 1)
193-
self.coordinator_stats.assert_query_count_equals(self, 3, 0)
191+
self.coordinator_stats.assert_query_count_equals(1, 0)
192+
self.coordinator_stats.assert_query_count_equals(2, 1)
193+
self.coordinator_stats.assert_query_count_equals(3, 0)
194194

195195
self.coordinator_stats.reset_counts()
196196

@@ -211,9 +211,9 @@ def _test_downgrading_cl(self, keyspace, rf, accepted):
211211
create_schema(cluster, session, keyspace, replication_factor=rf)
212212
self._insert(session, keyspace, 1)
213213
self._query(session, keyspace, 1)
214-
self.coordinator_stats.assert_query_count_equals(self, 1, 0)
215-
self.coordinator_stats.assert_query_count_equals(self, 2, 1)
216-
self.coordinator_stats.assert_query_count_equals(self, 3, 0)
214+
self.coordinator_stats.assert_query_count_equals(1, 0)
215+
self.coordinator_stats.assert_query_count_equals(2, 1)
216+
self.coordinator_stats.assert_query_count_equals(3, 0)
217217

218218
try:
219219
force_stop(2)
@@ -268,13 +268,13 @@ def rfthree_downgradingcl(self, cluster, keyspace, roundrobin):
268268
self._query(session, keyspace, count=12)
269269

270270
if roundrobin:
271-
self.coordinator_stats.assert_query_count_equals(self, 1, 4)
272-
self.coordinator_stats.assert_query_count_equals(self, 2, 4)
273-
self.coordinator_stats.assert_query_count_equals(self, 3, 4)
271+
self.coordinator_stats.assert_query_count_equals(1, 4)
272+
self.coordinator_stats.assert_query_count_equals(2, 4)
273+
self.coordinator_stats.assert_query_count_equals(3, 4)
274274
else:
275-
self.coordinator_stats.assert_query_count_equals(self, 1, 0)
276-
self.coordinator_stats.assert_query_count_equals(self, 2, 12)
277-
self.coordinator_stats.assert_query_count_equals(self, 3, 0)
275+
self.coordinator_stats.assert_query_count_equals(1, 0)
276+
self.coordinator_stats.assert_query_count_equals(2, 12)
277+
self.coordinator_stats.assert_query_count_equals(3, 0)
278278

279279
try:
280280
self.coordinator_stats.reset_counts()
@@ -288,13 +288,13 @@ def rfthree_downgradingcl(self, cluster, keyspace, roundrobin):
288288
self.coordinator_stats.reset_counts()
289289
self._query(session, keyspace, 12, consistency_level=cl)
290290
if roundrobin:
291-
self.coordinator_stats.assert_query_count_equals(self, 1, 6)
292-
self.coordinator_stats.assert_query_count_equals(self, 2, 0)
293-
self.coordinator_stats.assert_query_count_equals(self, 3, 6)
291+
self.coordinator_stats.assert_query_count_equals(1, 6)
292+
self.coordinator_stats.assert_query_count_equals(2, 0)
293+
self.coordinator_stats.assert_query_count_equals(3, 6)
294294
else:
295-
self.coordinator_stats.assert_query_count_equals(self, 1, 0)
296-
self.coordinator_stats.assert_query_count_equals(self, 2, 0)
297-
self.coordinator_stats.assert_query_count_equals(self, 3, 12)
295+
self.coordinator_stats.assert_query_count_equals(1, 0)
296+
self.coordinator_stats.assert_query_count_equals(2, 0)
297+
self.coordinator_stats.assert_query_count_equals(3, 12)
298298
finally:
299299
start(2)
300300
wait_for_up(cluster, 2)

0 commit comments

Comments
 (0)