mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-12-09 00:01:48 -05:00
7143245c37
when users of the library code bugs in callbacks, they should not be ignored (as they were before this change), but they should also not be treated as timeouts and such, in that they should not be wrapped in an error response. they should fail loudly, ,i.e. raise Closes #276
114 lines
3.0 KiB
Ruby
114 lines
3.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "resolv"
|
|
require "httpx/plugins/callbacks"
|
|
|
|
module Requests
|
|
using HTTPX::URIExtensions
|
|
module Callbacks
|
|
def test_callbacks_connection_opened
|
|
uri = URI(build_uri("/get"))
|
|
origin = ip = nil
|
|
opened = 0
|
|
|
|
response = HTTPX.plugin(SessionWithPool).plugin(:callbacks).on_connection_opened do |o, sock|
|
|
origin = o
|
|
ip = sock.to_io.remote_address.ip_address
|
|
opened += 1
|
|
end.get(uri)
|
|
verify_status(response, 200)
|
|
|
|
assert !origin.nil?
|
|
assert origin.to_s == uri.origin
|
|
assert !ip.nil?
|
|
assert opened == 1
|
|
|
|
assert Resolv.getaddresses(uri.host).include?(ip)
|
|
end
|
|
|
|
def test_callbacks_connection_closed
|
|
uri = URI(build_uri("/get"))
|
|
origin = nil
|
|
closed = 0
|
|
|
|
response = HTTPX.plugin(SessionWithPool).plugin(:callbacks).on_connection_closed do |o|
|
|
origin = o
|
|
closed += 1
|
|
end.get(uri)
|
|
verify_status(response, 200)
|
|
|
|
assert !origin.nil?
|
|
assert origin.to_s == uri.origin
|
|
assert closed == 1
|
|
end
|
|
|
|
def test_callbacks_request_error
|
|
uri = URI(build_uri("/get"))
|
|
error = nil
|
|
|
|
http = HTTPX.plugin(:callbacks).on_request_error { |_, err| error = err }
|
|
|
|
response = http.get(uri)
|
|
verify_status(response, 200)
|
|
|
|
assert error.nil?
|
|
|
|
unavailable_host = URI(origin("localhost"))
|
|
unavailable_host.port = next_available_port
|
|
response = http.get(unavailable_host.to_s)
|
|
verify_error_response(response, /Connection refused| not available/)
|
|
|
|
assert !error.nil?
|
|
assert error == response.error
|
|
end
|
|
|
|
def test_callbacks_request
|
|
uri = URI(build_uri("/post"))
|
|
started = completed = false
|
|
chunks = 0
|
|
|
|
http = HTTPX.plugin(:callbacks)
|
|
.on_request_started { |_| started = true }
|
|
.on_request_body_chunk { |_, _chunk| chunks += 1 }
|
|
.on_request_completed { |_| completed = true }
|
|
|
|
response = http.post(uri, body: "data")
|
|
verify_status(response, 200)
|
|
|
|
assert started
|
|
assert completed
|
|
assert chunks.positive?
|
|
end
|
|
|
|
def test_callbacks_response
|
|
uri = URI(build_uri("/get"))
|
|
started = completed = false
|
|
chunks = 0
|
|
|
|
http = HTTPX.plugin(:callbacks)
|
|
.on_response_started { |_, _| started = true }
|
|
.on_response_body_chunk { |_, _, _chunk| chunks += 1 }
|
|
.on_response_completed { |_, _| completed = true }
|
|
|
|
response = http.get(uri)
|
|
verify_status(response, 200)
|
|
|
|
assert started
|
|
assert completed
|
|
assert chunks.positive?
|
|
end
|
|
|
|
def test_callbacks_bug_inside_callback
|
|
%i[
|
|
connection_opened connection_closed
|
|
request_started request_completed
|
|
response_started response_body_chunk response_completed
|
|
].each do |callback|
|
|
assert_raises(NameError) do
|
|
HTTPX.plugin(SessionWithPool).plugin(:callbacks).send(:"on_#{callback}") { i_dont_exist }.get(build_uri("/get"))
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|