Skip to content
Permalink
Browse files
Add test for CASSANDRA-15905
Patch by Yifan Cai, reviewed by brandonwilliams for CASSANDRA-15905
  • Loading branch information
yifan-c authored and driftx committed Jun 30, 2020
1 parent 8c6332d commit f5bc21c40ccd4bc2b9bc118ec5888bad3cc15b16
Showing 1 changed file with 27 additions and 0 deletions.
@@ -2031,6 +2031,33 @@ def test_execute_statement(self):

assert_all(session, "SELECT * FROM ks.cf", [[0]])

def test_fetch_all_rows_in_batch_mode(self):
"""
Test: cqlsh -e "<SELECT STATEMENT>" with more rows than 1 page
@jira_ticket CASSANDRA-15905
"""
self.cluster.populate(1)
self.cluster.start(wait_for_binary_proto=True)
node1, = self.cluster.nodelist()
session = self.patient_cql_connection(node1)

session.execute("CREATE KEYSPACE ks WITH REPLICATION={'class':'SimpleStrategy','replication_factor':1};")
session.execute("CREATE TABLE ks.test (key uuid primary key);")

num_rows = 200
expected_lines = num_rows + 5 # 5: header + empty lines

for i in range(num_rows):
session.execute("INSERT INTO ks.test (key) VALUES (uuid())")

stdout, err = self.run_cqlsh(node1, cmds="", cqlsh_options=['-e', 'SELECT * FROM ks.test;'])
assert err == ""
output_lines = stdout.splitlines()
assert expected_lines == len(output_lines)
assert output_lines[0].strip() == ''
assert output_lines[-2].strip() == ''
assert output_lines[-1].strip() == "({} rows)".format(num_rows)

def run_cqlsh(self, node, cmds, cqlsh_options=None, env_vars=None):
"""
Local version of run_cqlsh to open a cqlsh subprocess with

0 comments on commit f5bc21c

Please sign in to comment.