Fix PR #276

mymarilyn · Dec 26, 2021 · a91b7c5 · a91b7c5
1 parent d60603c
commit a91b7c5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/clickhouse_driver/dbapi/cursor.py b/clickhouse_driver/dbapi/cursor.py
@@ -312,6 +312,8 @@ def _process_response(self, response, executemany=False):
 
         if not response or isinstance(response, int):
             self._columns = self._types = self._rows = []
+            if isinstance(response, int):
+                self._rowcount = response
             return
 
         if self._stream_results:

diff --git a/tests/test_dbapi.py b/tests/test_dbapi.py
@@ -153,10 +153,8 @@ def test_rowcount_insert_from_select(self):
 
     def test_execute_insert(self):
         with self.created_cursor() as cursor, self.create_table('a UInt8'):
-            cursor.execute(
-                'INSERT INTO test VALUES',
-                parameters=[[1]]
-            )
+            cursor.execute('INSERT INTO test VALUES', [[4]])
+            self.assertEqual(cursor.rowcount, 1)
 
     def test_description(self):
         with self.created_cursor() as cursor: