Skip to content

Commit

Permalink
black formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
jmeickle committed Jun 3, 2019
1 parent 4431efa commit 75c9b69
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 18 deletions.
2 changes: 1 addition & 1 deletion src/urllib3/util/retry.py
Expand Up @@ -210,7 +210,7 @@ def new(self, **kw):
raise_on_status=self.raise_on_status,
history=self.history,
remove_headers_on_redirect=self.remove_headers_on_redirect,
respect_retry_after_header=self.respect_retry_after_header
respect_retry_after_header=self.respect_retry_after_header,
)
params.update(kw)
return type(self)(**params)
Expand Down
26 changes: 9 additions & 17 deletions test/test_retry.py
Expand Up @@ -288,34 +288,26 @@ def test_parse_retry_after(self, value, expected):
retry = Retry()
assert retry.parse_retry_after(value) == expected

@pytest.mark.parametrize('respect_retry_after_header', [
True,
False
])
def test_respect_retry_after_header_propagated(self,
respect_retry_after_header):
@pytest.mark.parametrize("respect_retry_after_header", [True, False])
def test_respect_retry_after_header_propagated(self, respect_retry_after_header):

retry = Retry(respect_retry_after_header=respect_retry_after_header)
new_retry = retry.new()
assert new_retry.respect_retry_after_header \
== respect_retry_after_header
assert new_retry.respect_retry_after_header == respect_retry_after_header

@pytest.mark.parametrize(
'retry_after_header,respect_retry_after_header,sleep_duration', [
("3600", True, 3600),
("3600", False, None)
]
"retry_after_header,respect_retry_after_header,sleep_duration",
[("3600", True, 3600), ("3600", False, None)],
)
def test_respect_retry_after_header_sleep(self, retry_after_header,
respect_retry_after_header,
sleep_duration):
def test_respect_retry_after_header_sleep(
self, retry_after_header, respect_retry_after_header, sleep_duration
):
retry = Retry(respect_retry_after_header=respect_retry_after_header)

with patch("time.sleep") as sleep_mock:
# for the default behavior, it must be in RETRY_AFTER_STATUS_CODES
response = HTTPResponse(
status=503,
headers={"Retry-After": retry_after_header}
status=503, headers={"Retry-After": retry_after_header}
)

retry.sleep(response)
Expand Down

0 comments on commit 75c9b69

Please sign in to comment.