httpx/test/support/requests/plugins/proxy.rb

# frozen_string_literal: true

module Requests
  module Plugins
    module Proxy
      include ProxyHelper

      def test_plugin_http_proxy
        client = HTTPX.plugin(:proxy).with_proxy(uri: http_proxy)
        uri = build_uri("/get")
        response = client.get(uri)
        verify_status(response, 200)
        verify_body_length(response)
      end

      def test_plugin_socks4_proxy
        client = HTTPX.plugin(:proxy).with_proxy(uri: socks4_proxy)
        uri = build_uri("/get")
        response = client.get(uri)
        verify_status(response, 200)
        verify_body_length(response)
      end

      def test_plugin_socks4a_proxy
        client = HTTPX.plugin(:proxy).with_proxy(uri: socks4a_proxy)
        uri = build_uri("/get")
        response = client.get(uri)
        verify_status(response, 200)
        verify_body_length(response)
      end

      def test_plugin_socks5_proxy
        client = HTTPX.plugin(:proxy).with_proxy(uri: socks5_proxy)
        uri = build_uri("/get")
        response = client.get(uri)
        verify_status(response, 200)
        verify_body_length(response)
      end

      def test_plugin_ssh_proxy
        client = HTTPX.plugin(:"proxy/ssh").with_proxy(uri: ssh_proxy,
                                                       username: "root",
                                                       auth_methods: %w[publickey],
                                                       host_key: "ssh-rsa",
                                                       keys: %w[test/support/ssh/ssh_host_ed25519_key])
        uri = build_uri("/get")
        response = client.get(uri)
        verify_status(response, 200)
        verify_body_length(response)
      end if ENV.key?("HTTPX_SSH_PROXY")
    end
  end
end