chore: fix broken tests and flake

Author: karel-rehor

influxdb_client_3/write_client/client/write_api.py

@@ -547,13 +547,12 @@ def _retry_callback_delegate(exception):
         return _BatchResponse(data=batch_item)
 
     def _post_write(self, _async_req, bucket, org, body, precision, no_sync, **kwargs):
-            return self._write_service.post_write(org=org, bucket=bucket, body=body, precision=precision,
+        return self._write_service.post_write(org=org, bucket=bucket, body=body, precision=precision,
                                               no_sync=no_sync,
                                               async_req=_async_req,
                                               content_type="text/plain; charset=utf-8",
                                               **kwargs)
 
-
     def _to_response(self, data: _BatchItem, delay: timedelta):
 
         return rx.of(data).pipe(

tests/test_influxdb_client_3_integration.py

@@ -361,15 +361,14 @@ def retry_cb(args, data, excp):
             host=self.host,
             database=self.database,
             token=self.token,
+            write_timeout=1,
             write_client_options=write_client_options(
                 error_callback=set_error_result,
                 retry_callback=retry_cb,
-                write_timeout=20,
                 write_options=WriteOptions(
                     max_retry_time=10000,
                     max_retry_delay=100,
                     retry_interval=100,
-                    # timeout=20,
                     max_retries=3,
                     batch_size=1,
                 )

tests/test_polars.py

@@ -96,4 +96,6 @@ def test_write_polars_batching(self):
             async_req=ANY,
             content_type=ANY,
             urlopen_kw=ANY,
+            data_frame_measurement_name='measurement',
+            data_frame_timestamp_column='time',
             body=b'measurement temperature=22.4 1722470400000000000\nmeasurement temperature=21.8 1722474000000000000')