refactor: write rest client

Author: NguyenHoangSon96

tests/test_write_api.py

@@ -16,12 +16,13 @@
 _package = "influxdb3-python"
 _sentHeaders = {}
 
+
 class WriteApiTests(unittest.TestCase):
     received_timeout_total = None
 
     def mock_urllib3_timeout_request(method,
-                                 url,
-                                 **urlopen_kw):
+                                     url,
+                                     **urlopen_kw):
         if urlopen_kw.get('timeout', None) is not None:
             WriteApiTests.received_timeout_total = urlopen_kw['timeout'].total
             raise ConnectTimeoutError()
@@ -70,7 +71,7 @@ def test_api_error_oss_with_detail(self):
         with self.assertRaises(InfluxDBError) as err:
             self._test_api_error(response_body)
         self.assertEqual("parsing failed for write_lp endpoint:\n\tinvalid field value in line protocol for field "
-                     "'val' on line 1", err.exception.message)
+                         "'val' on line 1", err.exception.message)
 
     def test_api_error_unknown(self):
         response_body = '{"detail":"no info"}'
@@ -258,7 +259,7 @@ def test_api_error_headers(self):
         self.assertEqual(headers['X-Influxdb-Build'], 'Mock')
 
     @mock.patch("urllib3._request_methods.RequestMethods.request",
-                    side_effect=mock_urllib3_timeout_request)
+                side_effect=mock_urllib3_timeout_request)
     def test_write_timeout(self, mock_request):
         host = "http://localhost:8181"
         timeout = 300
@@ -276,7 +277,7 @@ def test_write_timeout(self, mock_request):
         self.received_timeout_total = None
 
     @mock.patch("urllib3._request_methods.RequestMethods.request",
-                    side_effect=mock_urllib3_timeout_request)
+                side_effect=mock_urllib3_timeout_request)
     def test_request_arg_timeout(self, mock_request):
         host = "http://localhost:8181"
         timeout = 300
@@ -290,7 +291,7 @@ def test_request_arg_timeout(self, mock_request):
 
         with pytest.raises(ConnectTimeoutError):
             client._write_api.write("TEST_ORG", "TEST_BUCKET", "data,foo=bar val=3.14",
-                               _request_timeout=100)
+                                    _request_timeout=100)
         self.assertEqual(0.1, self.received_timeout_total)
         self.received_timeout_total = None