From fcbf63e62c627deae76c1b8cb8c0876c536ed811 Mon Sep 17 00:00:00 2001
From: Jari Vetoniemi <jari.vetoniemi@indooratlas.com>
Date: Mon, 16 Mar 2020 18:49:26 +0900
Subject: Fresh start

---
 .../rubygems/test_gem_request_connection_pools.rb  | 120 +++++++++++++++++++++
 1 file changed, 120 insertions(+)
 create mode 100644 jni/ruby/test/rubygems/test_gem_request_connection_pools.rb

(limited to 'jni/ruby/test/rubygems/test_gem_request_connection_pools.rb')

diff --git a/jni/ruby/test/rubygems/test_gem_request_connection_pools.rb b/jni/ruby/test/rubygems/test_gem_request_connection_pools.rb
new file mode 100644
index 0000000..1cf6b27
--- /dev/null
+++ b/jni/ruby/test/rubygems/test_gem_request_connection_pools.rb
@@ -0,0 +1,120 @@
+require 'rubygems/test_case'
+require 'rubygems/request'
+require 'timeout'
+
+class TestGemRequestConnectionPool < Gem::TestCase
+  class FakeHttp
+    def initialize *args
+    end
+
+    def start
+    end
+  end
+
+  def setup
+    super
+    @old_client = Gem::Request::ConnectionPools.client
+    Gem::Request::ConnectionPools.client = FakeHttp
+
+    @proxy = URI 'http://proxy.example'
+  end
+
+  def teardown
+    Gem::Request::ConnectionPools.client = @old_client
+    super
+  end
+
+  def test_checkout_same_connection
+    uri = URI.parse('http://example/some_endpoint')
+
+    pools = Gem::Request::ConnectionPools.new nil, []
+    pool = pools.pool_for uri
+    conn = pool.checkout
+    pool.checkin conn
+
+    assert_equal conn, pool.checkout
+  end
+
+  def test_to_proxy_eh
+    pools = Gem::Request::ConnectionPools.new nil, []
+
+    env_no_proxy = %w[
+      1.no-proxy.example
+      2.no-proxy.example
+    ]
+
+    no_proxy = pools.send :no_proxy?, '2.no-proxy.example', env_no_proxy
+
+    assert no_proxy, 'match'
+
+    no_proxy = pools.send :no_proxy?, 'proxy.example', env_no_proxy
+
+    refute no_proxy, 'mismatch'
+  end
+
+  def test_to_proxy_eh_wildcard
+    pools = Gem::Request::ConnectionPools.new nil, []
+
+    env_no_proxy = %w[
+      .no-proxy.example
+    ]
+
+    no_proxy = pools.send :no_proxy?, '2.no-proxy.example', env_no_proxy
+
+    assert no_proxy, 'wildcard matching subdomain'
+
+    no_proxy = pools.send :no_proxy?, 'no-proxy.example', env_no_proxy
+
+    assert no_proxy, 'wildcard matching dotless domain'
+
+    no_proxy = pools.send :no_proxy?, 'proxy.example', env_no_proxy
+
+    refute no_proxy, 'wildcard mismatch'
+  end
+
+  def test_net_http_args
+    pools = Gem::Request::ConnectionPools.new nil, []
+
+    net_http_args = pools.send :net_http_args, URI('http://example'), nil
+
+    assert_equal ['example', 80], net_http_args
+  end
+
+  def test_net_http_args_proxy
+    pools = Gem::Request::ConnectionPools.new nil, []
+
+    net_http_args = pools.send :net_http_args, URI('http://example'), @proxy
+
+    assert_equal ['example', 80, 'proxy.example', 80, nil, nil], net_http_args
+  end
+
+  def test_net_http_args_no_proxy
+    orig_no_proxy, ENV['no_proxy'] = ENV['no_proxy'], 'example'
+
+    pools = Gem::Request::ConnectionPools.new nil, []
+
+    net_http_args = pools.send :net_http_args, URI('http://example'), @proxy
+
+    assert_equal ['example', 80, nil, nil], net_http_args
+
+  ensure
+    ENV['no_proxy'] = orig_no_proxy
+  end
+
+  def test_thread_waits_for_connection
+    uri = URI.parse('http://example/some_endpoint')
+    pools = Gem::Request::ConnectionPools.new nil, []
+    pool  = pools.pool_for uri
+
+    pool.checkout
+
+    t1 = Thread.new {
+      timeout(1) do
+        pool.checkout
+      end
+    }
+    assert_raises(Timeout::Error) do
+      t1.join
+    end
+  end
+end
-- 
cgit v1.2.3