feat: improve error handling

Author: NguyenHoangSon96

tests/test_influxdb_client_3_integration.py

@@ -7,7 +7,7 @@
 
 import pyarrow
 import pytest
-from pyarrow._flight import FlightError
+from influxdb_client_3.influxdb_client_error import InfluxdbClientQueryError
 
 from influxdb_client_3 import InfluxDBClient3, InfluxDBError, write_client_options, WriteOptions
 from tests.util import asyncio_run, lp_to_py_object
@@ -225,7 +225,7 @@ def test_queries_w_bad_cert(self):
                 query = "SELECT table_name FROM information_schema.tables"
                 client.query(query, mode="")
                 assert False, "query should throw SSL_ERROR"
-            except FlightError as fe:
+            except InfluxdbClientQueryError as fe:
                 assert str(fe).__contains__('SSL_ERROR_SSL')
             finally:
                 self.remove_test_cert(cert_file)