feat: add test for writing CSV files with batching

Author: NguyenHoangSon96

tests/test_influxdb_client_3_integration.py

@@ -61,7 +61,7 @@ def test_write_dataframe(self):
 
         result = self.client.query(query=f"select * from {measurement}", mode="pandas")
 
-        self.assertEqual(2, len(result))
+        self.assertIsNotNone(result)
         self.assertEqual(2, len(result.get('city')))
         self.assertEqual(2, len(result.get('temperature')))
 
@@ -92,6 +92,26 @@ def test_write_dataframe_with_batch(self):
         self.assertEqual(2, len(result.get('city')))
         self.assertEqual(2, len(result.get('temperature')))
 
+    def test_write_csv_file_with_batch(self):
+        client = InfluxDBClient3(host=self.host,
+                                 database=self.database,
+                                 token=self.token,
+                                 write_client_options=write_client_options(
+                                     write_options=WriteOptions(batch_size=100)
+                                 ))
+        measurement = f'test{random_hex(3)}'.lower()
+        client.write_file(
+            measurement_name=measurement,
+            file='tests/data/iot.csv',
+            timestamp_column='time', tag_columns=["name"])
+        client.flush()
+
+        result = client.query(query=f"select * from {measurement}", mode="pandas")
+        self.assertIsNotNone(result)
+        self.assertEqual(3, len(result.get('building')))
+        self.assertEqual(3, len(result.get('temperature')))
+
+
     def test_write_and_query(self):
         test_id = time.time_ns()
         self.client.write(f"integration_test_python,type=used value=123.0,test_id={test_id}i")