mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-10-06 00:02:08 -04:00
6b9a737756
this eliminates the overuse of Connection#origin, which in the case of proxied connections was broken in the previous commit the proxy implementation got simpler, despite this large changeset
93 lines
2.8 KiB
Ruby
93 lines
2.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "uri"
|
|
|
|
HTTP_PROXY = ENV["HTTPX_HTTP_PROXY"]
|
|
ENV["http_proxy"] = HTTP_PROXY
|
|
HTTPS_PROXY = ENV["HTTPX_HTTPS_PROXY"]
|
|
ENV["https_proxy"] = HTTPS_PROXY
|
|
NO_PROXY = ENV["HTTPBIN_NO_PROXY_HOST"]
|
|
ENV["no_proxy"] = URI(NO_PROXY).authority
|
|
|
|
require "test_helper"
|
|
require "support/http_helpers"
|
|
require "support/minitest_extensions"
|
|
|
|
class EnvProxyTest < Minitest::Test
|
|
include HTTPHelpers
|
|
using HTTPX::URIExtensions
|
|
|
|
def test_plugin_http_no_proxy
|
|
HTTPX.plugin(SessionWithPool).plugin(ProxyResponseDetector).wrap do |session|
|
|
# proxy
|
|
response = session.get("https://#{httpbin}/get")
|
|
verify_status(response, 200)
|
|
verify_body_length(response)
|
|
assert response.proxied?
|
|
|
|
# no proxy
|
|
no_proxy_response = session.get("#{NO_PROXY}/get")
|
|
verify_status(no_proxy_response, 200)
|
|
verify_body_length(no_proxy_response)
|
|
assert !no_proxy_response.proxied?
|
|
end
|
|
end
|
|
|
|
def test_env_proxy_coalescing
|
|
HTTPX.plugin(SessionWithPool).wrap do |http|
|
|
response = http.get("https://#{httpbin}/get")
|
|
verify_status(response, 200)
|
|
verify_body_length(response)
|
|
|
|
connections = http.connections
|
|
|
|
assert connections.size == 1
|
|
connection = connections.first
|
|
assert HTTPS_PROXY.end_with?(connection.peer.authority), "#{connection.peer.authority} not found in #{HTTPS_PROXY}"
|
|
end
|
|
end
|
|
|
|
def test_multiple_get_no_concurrency
|
|
uri = "https://nghttp2/get"
|
|
|
|
HTTPX.plugin(SessionWithPool).plugin(:persistent).wrap do |http|
|
|
response1, response2 = http.get(uri, uri, max_concurrent_requests: 1)
|
|
|
|
verify_status(response1, 200)
|
|
verify_body_length(response1)
|
|
|
|
verify_status(response2, 200)
|
|
verify_body_length(response2)
|
|
|
|
connections = http.connections
|
|
|
|
assert connections.size == 1
|
|
connections.each do |connection|
|
|
assert HTTP_PROXY.end_with?(connection.peer.authority), "#{connection.peer.authority} not found in #{HTTPS_PROXY}"
|
|
end
|
|
end
|
|
end
|
|
|
|
# def test_env_proxy_altsvc_get
|
|
# altsvc_host = ENV["HTTPBIN_ALTSVC_HOST"]
|
|
|
|
# HTTPX.plugin(SessionWithPool).wrap do |http|
|
|
# altsvc_uri = "https://#{altsvc_host}/get"
|
|
# response = http.get(altsvc_uri)
|
|
# verify_status(response, 200)
|
|
# verify_header(response.headers, "alt-svc", "h2=\"nghttp2:443\"")
|
|
# response2 = http.get(altsvc_uri)
|
|
# verify_status(response2, 200)
|
|
# verify_no_header(response2.headers, "alt-svc")
|
|
# # introspection time
|
|
# pool = session.pool
|
|
# connections = pool.connections
|
|
|
|
# assert connections.size == 1
|
|
# connections.each do |connection|
|
|
# assert HTTPS_PROXY.end_with?(connection.origin.authority), "#{connection.origin.authority} not found in #{HTTPS_PROXY}"
|
|
# end
|
|
# end
|
|
# end
|
|
end
|