Skip to content

Commit 1b77e57

Browse files
ai: apply changes for #1007 (1 review thread)
Addresses: - #4418426396 at tests/unit/test_sea_result_set.py:531 Signed-off-by: peco-engineer-bot[bot] <peco-engineer-bot[bot]@users.noreply.github.com>
1 parent bf198d2 commit 1b77e57

1 file changed

Lines changed: 18 additions & 0 deletions

File tree

‎tests/unit/test_sea_result_set.py‎

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -527,6 +527,24 @@ def test_json_fetchone_converts_in_batches(
527527
assert [r.col2 for r in rows] == list(range(100))
528528
assert m.call_count <= 100 // 10 + 2
529529

530+
def test_json_row_fetches_keep_order_and_rownumber(
531+
self, mock_connection, mock_sea_client, execute_response
532+
):
533+
"""Mixed fetchone/fetchmany/fetchall ordering without any Arrow path."""
534+
rs = self._make_json_result_set(
535+
mock_connection, mock_sea_client, execute_response, 10, arraysize=4
536+
)
537+
assert rs.fetchone().col2 == 0
538+
assert rs.fetchone().col2 == 1
539+
assert rs.rownumber == 2
540+
assert [r.col2 for r in rs.fetchmany(3)] == [2, 3, 4]
541+
assert rs.rownumber == 5
542+
assert rs.fetchone().col2 == 5
543+
assert rs.rownumber == 6
544+
assert [r.col2 for r in rs.fetchall()] == [6, 7, 8, 9]
545+
assert rs.rownumber == 10
546+
assert rs.fetchone() is None
547+
530548
@pytest.mark.skipif(pyarrow is None, reason="PyArrow is not installed")
531549
def test_json_mixed_fetches_keep_order_and_rownumber(
532550
self, mock_connection, mock_sea_client, execute_response

0 commit comments

Comments
 (0)