Skip to content

Commit e1ac214

Browse files
Apply suggestions from code review
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent 8bd4163 commit e1ac214

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

segment/analytics/request.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import json
66
import base64
77
from dateutil.tz import tzutc
8-
from requests.auth import HTTPBasicAuth
8+
99
from requests import sessions
1010

1111
from segment.analytics.version import VERSION
@@ -30,7 +30,8 @@ def parse_retry_after(response):
3030
try:
3131
# Try parsing as integer (delay in seconds)
3232
delay = int(retry_after)
33-
return min(delay, MAX_RETRY_AFTER_SECONDS)
33+
# Ensure delay is non-negative before applying upper bound
34+
return min(max(delay, 0), MAX_RETRY_AFTER_SECONDS)
3435
except ValueError:
3536
# Could be HTTP-date format, but for simplicity we'll skip that
3637
# Most APIs use integer seconds

segment/analytics/test/test_consumer.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -490,10 +490,8 @@ def mock_post_fn(*args, **kwargs):
490490
raise FatalError('Fatal error occurred')
491491

492492
with mock.patch('segment.analytics.consumer.post', side_effect=mock_post_fn):
493-
try:
493+
with self.assertRaises(FatalError):
494494
consumer.request([track])
495-
except FatalError:
496-
pass
497495

498496
# Should only be called once (no retries)
499497
self.assertEqual(call_count, 1)

0 commit comments

Comments
 (0)