Merge pull request #494 from Mause/bugfix/disable-server-side-cursors

fix: disable server side cursors

Author: kodiakhq[bot]

duckdb_engine/__init__.py

@@ -149,6 +149,7 @@ class Dialect(PGDialect_psycopg2):
     supports_statement_cache = False
     supports_comments = False
     supports_sane_rowcount = False
+    supports_server_side_cursors = False
     inspector = DuckDBInspector
     # colspecs TODO: remap types to duckdb types
     colspecs = util.update_copy(

duckdb_engine/tests/test_basic.py

@@ -118,6 +118,17 @@ def test_foreign(session: Session) -> None:
     assert owner.owned.name == "Walter"
 
 
+def test_disabled_server_side_cursors(engine: Engine) -> None:
+    connection = engine.connect().execution_options(stream_results=True)
+
+    session = sessionmaker(bind=connection)()
+
+    session.add(FakeModel(name="Walter"))
+    session.commit()
+
+    assert list(session.query(FakeModel).yield_per(1))
+
+
 @given(text_strat())
 @settings(deadline=timedelta(seconds=1))
 def test_simple_string(s: str) -> None: