fix: add missing gzip support for v3 api

Author: jansimonb

influxdb_client_3/write_client/client/_base.py

@@ -282,7 +282,7 @@ def update_request_header_params(self, path: str, params: dict, should_gzip: boo
         super().update_request_header_params(path, params, should_gzip)
         if should_gzip:
             # GZIP Request
-            if path == '/api/v2/write':
+            if path == '/api/v2/write' or path == '/api/v3/write_lp':
                 params["Content-Encoding"] = "gzip"
                 params["Accept-Encoding"] = "identity"
                 pass
@@ -298,7 +298,7 @@ def update_request_body(self, path: str, body, should_gzip: bool = False):
         _body = super().update_request_body(path, body, should_gzip)
         if should_gzip:
             # GZIP Request
-            if path == '/api/v2/write':
+            if path == '/api/v2/write' or path == '/api/v3/write_lp':
                 import gzip
                 if isinstance(_body, bytes):
                     return gzip.compress(data=_body)

influxdb_client_3/write_client/client/write_api.py

@@ -209,7 +209,7 @@ def _body_reduce(batch_items):
 
 class WriteApi(_BaseWriteApi):
     """
-    Implementation for '/api/v2/write' endpoint.
+    Implementation for '/api/v2/write' and '/api/v3/write_lp' endpoint.
 
     Example:
         .. code-block:: python

tests/test_write_integration.py

@@ -45,7 +45,7 @@ def test_write_with_write_options(self, httpserver: HTTPServer):
                     write_precision=WritePrecision.US,
                     no_sync=False
                 )
-            )
+            ),
         ).write(self.SAMPLE_RECORD)
 
         self.assert_request_made(httpserver, RequestMatcher(
@@ -87,3 +87,43 @@ def test_write_with_no_sync_true_on_v2_server(self, httpserver: HTTPServer):
         self.assert_request_made(httpserver, RequestMatcher(
             method="POST", uri="/api/v3/write_lp",
             query_string={"org": "ORG", "db": "DB", "precision": "nanosecond", "no_sync": "true"}))
+
+    def test_write_with_no_sync_false_and_gzip(self, httpserver: HTTPServer):
+        self.set_response_status(httpserver, 200)
+
+        InfluxDBClient3(
+            host=(httpserver.url_for("/")), org="ORG", database="DB", token="TOKEN",
+            write_client_options=write_client_options(
+                write_options=WriteOptions(
+                    write_type=WriteType.synchronous,
+                    write_precision=WritePrecision.US,
+                    no_sync=False
+                )
+            ),
+            enable_gzip=True
+        ).write(self.SAMPLE_RECORD)
+
+        self.assert_request_made(httpserver, RequestMatcher(
+            method="POST", uri="/api/v2/write",
+            query_string={"org": "ORG", "bucket": "DB", "precision": "us"},
+            headers={"Content-Encoding": "gzip"}, ))
+
+    def test_write_with_no_sync_true_and_gzip(self, httpserver: HTTPServer):
+        self.set_response_status(httpserver, 200)
+
+        InfluxDBClient3(
+            host=(httpserver.url_for("/")), org="ORG", database="DB", token="TOKEN",
+            write_client_options=write_client_options(
+                write_options=WriteOptions(
+                    write_type=WriteType.synchronous,
+                    write_precision=WritePrecision.US,
+                    no_sync=True
+                )
+            ),
+            enable_gzip=True
+        ).write(self.SAMPLE_RECORD)
+
+        self.assert_request_made(httpserver, RequestMatcher(
+            method="POST", uri="/api/v3/write_lp",
+            query_string={"org": "ORG", "db": "DB", "precision": "microsecond", "no_sync": "true"},
+            headers={"Content-Encoding": "gzip"}, ))