Skip to content

Commit 7dff093

Browse files
refactor: revert code
1 parent 8ebfa53 commit 7dff093

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

tests/test_WriteApi.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -528,8 +528,12 @@ def test_writes_default_tags_dict_without_tag(self):
528528

529529
def test_redirect(self):
530530
from urllib3 import Retry
531+
Retry.DEFAULT_REMOVE_HEADERS_ON_REDIRECT = frozenset()
532+
Retry.DEFAULT.remove_headers_on_redirect = Retry.DEFAULT_REMOVE_HEADERS_ON_REDIRECT
531533
self.influxdb_client.close()
532534

535+
# In the newer urllib3 versions we need to set `redirect` and `remove_headers_on_redirect=[]` to
536+
# make it re-direct POST requests and stop it from remove the `Authorization` header.
533537
retries = Retry(redirect=1, remove_headers_on_redirect=[])
534538
self.influxdb_client = InfluxDBClient(url="http://localhost", token="my-token", org="my-org", retries=retries)
535539

@@ -546,6 +550,9 @@ def test_redirect(self):
546550
self.assertEqual('Token my-token', requests[0].headers['Authorization'])
547551
self.assertEqual('Token my-token', requests[1].headers['Authorization'])
548552

553+
from urllib3 import Retry
554+
Retry.DEFAULT.remove_headers_on_redirect = Retry.DEFAULT_REMOVE_HEADERS_ON_REDIRECT
555+
549556
def test_named_tuple(self):
550557
httpretty.register_uri(httpretty.POST, uri="http://localhost/api/v2/write", status=204)
551558

0 commit comments

Comments
 (0)