Skip to content

Commit c2f04bd

Browse files
committed
test: add simple unit test for new proxy code (WIP)
1 parent e111367 commit c2f04bd

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-1
lines changed

influxdb_client_3/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def __init__(
121121
possible.
122122
:key str proxy: Set this to configure the http proxy to be used (ex. http://localhost:3128)
123123
:key str proxy_headers: A dictionary containing headers that will be sent to the proxy. Could be used for proxy
124-
authentication.
124+
authentication. (Applies to Write API only)
125125
:key int connection_pool_maxsize: Number of connections to save that can be reused by urllib3.
126126
Defaults to "multiprocessing.cpu_count() * 5".
127127
:key urllib3.util.retry.Retry retries: Set the default retry strategy that is used for all HTTP requests

tests/test_query.py

+16
Original file line numberDiff line numberDiff line change
@@ -150,3 +150,19 @@ def test_query_with_parameters(self):
150150
)
151151

152152
mock_do_get.assert_called_once_with(expected_ticket, ANY)
153+
154+
def test_query_proxy_base_client(self):
155+
test_proxy = "http://testproxy:5432"
156+
client = InfluxDBClient3(
157+
host="http://localhost:8443",
158+
token="my-token",
159+
org="my-org",
160+
database="my-database",
161+
proxy=test_proxy
162+
)
163+
164+
assert client._query_api._proxy == test_proxy
165+
assert ('grpc.http_proxy', test_proxy) in\
166+
client._query_api._flight_client_options.get('generic_options')
167+
168+

0 commit comments

Comments
 (0)