From c39332710d3d0e4e17c879ea0b58550fecd12518 Mon Sep 17 00:00:00 2001 From: Tomasz Grabiec Date: Thu, 18 May 2023 00:39:54 +0200 Subject: [PATCH] test: test_tablets: materialize all rows from the result set When paging, iterating twice over the result set is not possible, making the second loop noop. --- test/topology/test_tablets.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/test/topology/test_tablets.py b/test/topology/test_tablets.py index 8b2cf16891..a96a2e6d45 100644 --- a/test/topology/test_tablets.py +++ b/test/topology/test_tablets.py @@ -58,8 +58,8 @@ async def test_tablet_metadata_propagates_with_schema_changes_in_snapshot_mode(m await asyncio.gather(*[manager.cql.run_async(f"INSERT INTO test.test (pk, c) VALUES ({k}, 2);", execution_profile='whitelist') for k in keys]) - rows = manager.cql.execute("SELECT * FROM test.test;") - assert len(list(rows)) == len(keys) + rows = manager.cql.execute("SELECT * FROM test.test;").all() + assert len(rows) == len(keys) for r in rows: assert r.c == 2 @@ -73,8 +73,8 @@ async def test_tablet_metadata_propagates_with_schema_changes_in_snapshot_mode(m await asyncio.gather(*[manager.cql.run_async(f"INSERT INTO test.test (pk, c) VALUES ({k}, 3);", execution_profile='whitelist') for k in keys]) - rows = manager.cql.execute("SELECT * FROM test.test;") - assert len(list(rows)) == len(keys) + rows = manager.cql.execute("SELECT * FROM test.test;").all() + assert len(rows) == len(keys) for r in rows: assert r.c == 3 @@ -94,8 +94,8 @@ async def test_scans(manager: ManagerClient): rows = manager.cql.execute("SELECT count(*) FROM test.test;") assert rows.one().count == len(keys) - rows = manager.cql.execute("SELECT * FROM test.test;") - assert len(list(rows)) == len(keys) + rows = manager.cql.execute("SELECT * FROM test.test;").all() + assert len(rows) == len(keys) for r in rows: assert r.c == r.pk