Merge pull request #327 from carlosfy/326-fix-handle-progress-packet

Fix handle Progress packet while insert from input (#326)

clickhouse_driver/client.py

@@ -589,6 +589,9 @@ def receive_end_of_query(self):
             if packet.type == ServerPacketTypes.END_OF_STREAM:
                 break
 
+            elif packet.type == ServerPacketTypes.PROGRESS:
+                continue
+
             elif packet.type == ServerPacketTypes.EXCEPTION:
                 raise packet.exception
 

tests/test_insert.py

@@ -3,6 +3,7 @@
 from tests.testcase import BaseTestCase
 from clickhouse_driver import errors
 from clickhouse_driver.errors import ServerException
+from tests.util import require_server_version
 
 
 class InsertTestCase(BaseTestCase):
@@ -148,6 +149,20 @@ def test_insert_return(self):
             )
             self.assertEqual(rv, 5)
 
+    @require_server_version(22, 3, 6)
+    def test_insert_from_input(self):
+        with self.create_table('a Int8'):
+            data = [{'a': 1}]
+            self.client.execute(
+                "INSERT INTO test (a) "
+                "SELECT a from input ('a Int8')",
+                data
+            )
+
+            query = 'SELECT * FROM test'
+            inserted = self.emit_cli(query)
+            self.assertEqual(inserted, '1\n')
+
 
 class InsertColumnarTestCase(BaseTestCase):
     def test_insert_tuple_ok(self):