mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-10-06 00:02:08 -04:00
74 lines
2.6 KiB
Ruby
74 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Requests
|
|
module Plugins
|
|
module RateLimiter
|
|
def test_plugin_rate_limiter_429
|
|
rate_limiter_session = HTTPX.plugin(RequestInspector)
|
|
.plugin(SessionWithMockResponse[429])
|
|
.plugin(:rate_limiter)
|
|
|
|
uri = build_uri("/get")
|
|
|
|
rate_limiter_session.get(uri)
|
|
|
|
verify_rated_responses(rate_limiter_session, 429)
|
|
end
|
|
|
|
def test_plugin_rate_limiter_503
|
|
rate_limiter_session = HTTPX.plugin(RequestInspector)
|
|
.plugin(SessionWithMockResponse[503])
|
|
.plugin(:rate_limiter)
|
|
|
|
uri = build_uri("/get")
|
|
|
|
rate_limiter_session.get(uri)
|
|
|
|
verify_rated_responses(rate_limiter_session, 503)
|
|
end
|
|
|
|
def test_plugin_rate_limiter_retry_after_integer
|
|
rate_limiter_session = HTTPX.plugin(RequestInspector)
|
|
.plugin(SessionWithMockResponse[429, "retry-after" => "2"])
|
|
.plugin(:rate_limiter)
|
|
|
|
uri = build_uri("/get")
|
|
|
|
before_time = Process.clock_gettime(Process::CLOCK_MONOTONIC, :second)
|
|
rate_limiter_session.get(uri)
|
|
after_time = Process.clock_gettime(Process::CLOCK_MONOTONIC, :second)
|
|
|
|
verify_rated_responses(rate_limiter_session, 429)
|
|
|
|
total_time = after_time - before_time
|
|
assert total_time >= 2, "request didn't take as expected to retry (#{total_time} secs)"
|
|
end
|
|
|
|
def test_plugin_rate_limiter_retry_after_date
|
|
rate_limiter_session = HTTPX.plugin(RequestInspector)
|
|
.plugin(SessionWithMockResponse[429, "retry-after" => (Time.now + 3).httpdate])
|
|
.plugin(:rate_limiter)
|
|
|
|
uri = build_uri("/get")
|
|
|
|
before_time = Process.clock_gettime(Process::CLOCK_MONOTONIC, :second)
|
|
rate_limiter_session.get(uri)
|
|
after_time = Process.clock_gettime(Process::CLOCK_MONOTONIC, :second)
|
|
|
|
verify_rated_responses(rate_limiter_session, 429)
|
|
total_time = after_time - before_time
|
|
assert total_time >= 2, "request didn't take as expected to retry (#{total_time} secs)"
|
|
end
|
|
|
|
private
|
|
|
|
def verify_rated_responses(session, rated_status)
|
|
assert session.total_responses.size == 2, "expected 2 responses(was #{session.total_responses.size})"
|
|
rated_response, response = session.total_responses
|
|
verify_status(rated_response, rated_status)
|
|
verify_status(response, 200)
|
|
end
|
|
end
|
|
end
|
|
end
|