mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-10-08 00:02:42 -04:00
101 lines
4.2 KiB
Ruby
101 lines
4.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Requests
|
|
module Plugins
|
|
module Retries
|
|
def test_plugin_retries
|
|
no_retries_session = HTTPX.plugin(RequestInspector).timeout(total_timeout: 3)
|
|
no_retries_response = no_retries_session.get(build_uri("/delay/10"))
|
|
assert no_retries_response.is_a?(HTTPX::ErrorResponse)
|
|
assert no_retries_session.calls.zero?, "expect request to be built 1 times (was #{no_retries_session.calls})"
|
|
retries_session = HTTPX
|
|
.plugin(RequestInspector)
|
|
.plugin(:retries)
|
|
.timeout(total_timeout: 3)
|
|
retries_response = retries_session.get(build_uri("/delay/10"))
|
|
assert retries_response.is_a?(HTTPX::ErrorResponse)
|
|
assert retries_session.calls == 3, "expect request to be built 4 times (was #{retries_session.calls})"
|
|
end
|
|
|
|
def test_plugin_retries_max_retries
|
|
retries_session = HTTPX
|
|
.plugin(RequestInspector)
|
|
.plugin(:retries)
|
|
.timeout(total_timeout: 3)
|
|
.max_retries(2)
|
|
retries_response = retries_session.get(build_uri("/delay/10"))
|
|
assert retries_response.is_a?(HTTPX::ErrorResponse)
|
|
# we're comparing against max-retries + 1, because the calls increment will happen
|
|
# also in the last call, where the request is not going to be retried.
|
|
assert retries_session.calls == 2, "expect request to be built 3 times (was #{retries_session.calls})"
|
|
end
|
|
|
|
def test_plugin_retries_retry_on
|
|
retry_callback = lambda do |response|
|
|
!response.error.is_a?(HTTPX::TimeoutError)
|
|
end
|
|
|
|
retries_session = HTTPX
|
|
.plugin(RequestInspector)
|
|
.plugin(:retries, retry_on: retry_callback)
|
|
.timeout(total_timeout: 3)
|
|
.max_retries(2)
|
|
|
|
retries_response = retries_session.get(build_uri("/delay/10"))
|
|
assert retries_response.is_a?(HTTPX::ErrorResponse)
|
|
assert retries_session.calls.zero?, "expect request not to be retried (it was, #{retries_session.calls} times)"
|
|
end
|
|
|
|
def test_plugin_retries_retry_after
|
|
before_time = Process.clock_gettime(Process::CLOCK_MONOTONIC, :second)
|
|
retries_session = HTTPX
|
|
.plugin(RequestInspector)
|
|
.plugin(:retries, retry_after: 2)
|
|
.timeout(total_timeout: 3)
|
|
.max_retries(1)
|
|
retries_response = retries_session.get(build_uri("/delay/10"))
|
|
after_time = Process.clock_gettime(Process::CLOCK_MONOTONIC, :second)
|
|
total_time = after_time - before_time
|
|
|
|
assert retries_response.is_a?(HTTPX::ErrorResponse)
|
|
assert_in_delta 3 + 2 + 3, total_time, 1, "request didn't take as expected to retry (#{total_time} secs)"
|
|
end
|
|
|
|
def test_plugin_retries_retry_after_callable
|
|
retries = 0
|
|
exponential = ->(_) { (retries += 1) * 2 }
|
|
before_time = Process.clock_gettime(Process::CLOCK_MONOTONIC, :second)
|
|
retries_session = HTTPX
|
|
.plugin(RequestInspector)
|
|
.plugin(:retries, retry_after: exponential)
|
|
.timeout(total_timeout: 3)
|
|
.max_retries(2)
|
|
retries_response = retries_session.get(build_uri("/delay/10"))
|
|
after_time = Process.clock_gettime(Process::CLOCK_MONOTONIC, :second)
|
|
total_time = after_time - before_time
|
|
|
|
assert retries_response.is_a?(HTTPX::ErrorResponse)
|
|
assert_in_delta 3 + 2 + 3 + 4 + 3, total_time, 1, "request didn't take as expected to retry (#{total_time} secs)"
|
|
end
|
|
|
|
module RequestInspector
|
|
module InstanceMethods
|
|
attr_reader :calls
|
|
def initialize(*args)
|
|
super
|
|
# we're comparing against max-retries + 1, because the calls increment will happen
|
|
# also in the last call, where the request is not going to be retried.
|
|
@calls = -1
|
|
end
|
|
|
|
def fetch_response(*)
|
|
response = super
|
|
@calls += 1 if response
|
|
response
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|