feat: support-env

Author: NguyenHoangSon96

tests/test_influxdb_client_3.py

@@ -1,7 +1,7 @@
 import unittest
 from unittest.mock import patch
 
-from influxdb_client_3 import InfluxDBClient3, WritePrecision, DefaultWriteOptions
+from influxdb_client_3 import InfluxDBClient3, WritePrecision, DefaultWriteOptions, Point, WriteOptions, WriteType
 from tests.util import asyncio_run
 from tests.util.mocks import ConstantFlightServer, ConstantData
 
@@ -74,6 +74,16 @@ async def test_query_async(self):
             assert {'data': 'sql_query', 'reference': query, 'value': -1.0} in result_list
             assert {'data': 'query_type', 'reference': 'sql', 'value': -1.0} in result_list
 
+    def test_write_api_custom_options_no_error(self):
+        write_options = WriteOptions(write_type=WriteType.batching)
+        write_client_option = {'write_options': write_options}
+        client = InfluxDBClient3(write_client_options=write_client_option)
+        try:
+            client._write_api._write_batching("bucket", "org", Point.measurement("test"), None)
+            self.assertTrue(True)
+        except Exception as e:
+            self.fail(f"Write API with default options raised an exception: {str(e)}")
+
     def test_default_client(self):
         expected_precision = DefaultWriteOptions.write_precision.value
         expected_write_type = DefaultWriteOptions.write_type.value