sonixlabs-eventmachine-java 1.0.0.rc.4-java

data/tests/test_pending_connect_timeout.rb

@@ -0,0 +1,52 @@
+require 'em_test_helper'
+
+class TestPendingConnectTimeout < Test::Unit::TestCase
+
+  if EM.respond_to? :get_pending_connect_timeout
+    def test_default
+      EM.run {
+        c = EM.connect("127.0.0.1", 54321)
+        assert_equal 20.0, c.pending_connect_timeout
+        EM.stop
+      }
+    end
+
+    def test_set_and_get
+      EM.run {
+        c = EM.connect("127.0.0.1", 54321)
+        c.pending_connect_timeout = 2.5
+        assert_equal 2.5, c.pending_connect_timeout
+        EM.stop
+      }
+    end
+
+    def test_for_real
+      start, finish = nil
+
+      timeout_handler = Module.new do
+        define_method :unbind do
+          finish = Time.now
+          EM.stop
+        end
+      end
+
+      EM.run {
+        setup_timeout
+        EM.heartbeat_interval = 0.1
+        start = Time.now
+        c = EM.connect("1.2.3.4", 54321, timeout_handler)
+        c.pending_connect_timeout = 0.2
+      }
+
+      assert_in_delta(0.2, (finish - start), 0.1)
+    end
+  else
+    warn "EM.pending_connect_timeout not implemented, skipping tests in #{__FILE__}"
+
+    # Because some rubies will complain if a TestCase class has no tests
+    def test_em_pending_connect_timeout_not_implemented
+      assert true
+    end
+  end
+
+end

data/tests/test_pool.rb

@@ -0,0 +1,194 @@
+class TestPool < Test::Unit::TestCase
+  def pool
+    @pool ||= EM::Pool.new
+  end
+
+  def go
+    EM.run { yield }
+  end
+
+  def stop
+    EM.stop
+  end
+
+  def deferrable
+    @deferrable ||= EM::DefaultDeferrable.new
+  end
+
+  def test_supports_more_work_than_resources
+    ran = false
+    go do
+      pool.perform do
+        ran = true
+        deferrable
+      end
+      stop
+    end
+    assert_equal false, ran
+    go do
+      pool.add :resource
+      stop
+    end
+    assert_equal true, ran
+  end
+
+  def test_reques_resources_on_error
+    pooled_res, pooled_res2 = nil
+    pool.add :res
+    go do
+      pool.perform do |res|
+        pooled_res = res
+        deferrable
+      end
+      stop
+    end
+    deferrable.fail
+    go do
+      pool.perform do |res|
+        pooled_res2 = res
+        deferrable
+      end
+      stop
+    end
+    assert_equal :res, pooled_res
+    assert_equal pooled_res, pooled_res2
+  end
+
+  def test_supports_custom_error_handler
+    eres = nil
+    pool.on_error do |res|
+      eres = res
+    end
+    performs = []
+    pool.add :res
+    go do
+      pool.perform do |res|
+        performs << res
+        deferrable
+      end
+      pool.perform do |res|
+        performs << res
+        deferrable
+      end
+      deferrable.fail
+      stop
+    end
+    assert_equal :res, eres
+    # manual requeues required when error handler is installed:
+    assert_equal 1, performs.size
+    assert_equal :res, performs.first
+  end
+
+  def test_catches_successful_deferrables
+    performs = []
+    pool.add :res
+    go do
+      pool.perform { |res| performs << res; deferrable }
+      pool.perform { |res| performs << res; deferrable }
+      stop
+    end
+    assert_equal [:res], performs
+    deferrable.succeed
+    go { stop }
+    assert_equal [:res, :res], performs
+  end
+
+  def test_prunes_locked_and_removed_resources
+    performs = []
+    pool.add :res
+    deferrable.succeed
+    go do
+      pool.perform { |res| performs << res; pool.remove res; deferrable }
+      pool.perform { |res| performs << res; pool.remove res; deferrable }
+      stop
+    end
+    assert_equal [:res], performs
+  end
+
+  # Contents is only to be used for inspection of the pool!
+  def test_contents
+    pool.add :res
+    assert_equal [:res], pool.contents
+    # Assert that modifying the contents list does not affect the pools
+    # contents.
+    pool.contents.delete(:res)
+    assert_equal [:res], pool.contents
+  end
+
+  def test_contents_when_perform_errors_and_on_error_is_not_set
+    pool.add :res
+    assert_equal [:res], pool.contents
+
+    pool.perform do |r|
+      d = EM::DefaultDeferrable.new
+      d.fail
+      d
+    end
+
+    EM.run { EM.next_tick { EM.stop } }
+
+    assert_equal [:res], pool.contents
+  end
+
+  def test_contents_when_perform_errors_and_on_error_is_set
+    pool.add :res
+    res = nil
+    pool.on_error do |r|
+      res = r
+    end
+    assert_equal [:res], pool.contents
+
+    pool.perform do |r|
+      d = EM::DefaultDeferrable.new
+      d.fail 'foo'
+      d
+    end
+
+    EM.run { EM.next_tick { EM.stop } }
+
+    assert_equal :res, res
+    assert_equal [], pool.contents
+  end
+
+  def test_num_waiting
+    pool.add :res
+    assert_equal 0, pool.num_waiting
+    pool.perform { |r| EM::DefaultDeferrable.new }
+    assert_equal 0, pool.num_waiting
+    10.times { pool.perform { |r| EM::DefaultDeferrable.new } }
+    EM.run { EM.next_tick { EM.stop } }
+    assert_equal 10, pool.num_waiting
+  end
+
+  def test_exceptions_in_the_work_block_bubble_up_raise_and_fail_the_resource
+    pool.add :res
+
+    res = nil
+    pool.on_error { |r| res = r }
+    pool.perform { raise 'boom' }
+
+    assert_raises(RuntimeError) do
+      EM.run { EM.next_tick { EM.stop } }
+    end
+
+    assert_equal [], pool.contents
+    assert_equal :res, res
+  end
+
+  def test_removed_list_does_not_leak_on_errors
+    pool.add :res
+
+    pool.on_error do |r|
+      # This is actually the wrong thing to do, and not required, but some users
+      # might do it. When they do, they would find that @removed would cause a
+      # slow leak.
+      pool.remove r
+    end
+
+    pool.perform { d = EM::DefaultDeferrable.new; d.fail; d }
+
+    EM.run { EM.next_tick { EM.stop } }
+    assert_equal [], pool.instance_variable_get(:@removed)
+  end
+
+end

data/tests/test_process_watch.rb

@@ -0,0 +1,48 @@
+require 'em_test_helper'
+
+if EM.kqueue?
+  class TestProcessWatch < Test::Unit::TestCase
+    module ParentProcessWatcher
+      def process_forked
+        $forked = true
+      end
+    end
+
+    module ChildProcessWatcher
+      def process_exited
+        $exited = true
+      end
+      def unbind
+        $unbind = true
+        EM.stop
+      end
+    end
+
+    def setup
+      EM.kqueue = true
+    end
+
+    def teardown
+      EM.kqueue = false
+    end
+
+    def test_events
+      EM.run{
+        # watch ourselves for a fork notification
+        EM.watch_process(Process.pid, ParentProcessWatcher)
+        $fork_pid = fork{ sleep }
+        child = EM.watch_process($fork_pid, ChildProcessWatcher)
+        $pid = child.pid
+
+        EM.add_timer(0.2){
+          Process.kill('TERM', $fork_pid)
+        }
+      }
+
+      assert_equal($pid, $fork_pid)
+      assert($forked)
+      assert($exited)
+      assert($unbind)
+    end
+  end
+end

data/tests/test_processes.rb

@@ -0,0 +1,128 @@
+require 'em_test_helper'
+
+class TestProcesses < Test::Unit::TestCase
+
+  if !windows? && !jruby?
+
+    # EM::DeferrableChildProcess is a sugaring of a common use-case
+    # involving EM::popen.
+    # Call the #open method on EM::DeferrableChildProcess, passing
+    # a command-string. #open immediately returns an EM::Deferrable
+    # object. It also schedules the forking of a child process, which
+    # will execute the command passed to #open.
+    # When the forked child terminates, the Deferrable will be signalled
+    # and execute its callbacks, passing the data that the child process
+    # wrote to stdout.
+    #
+    def test_deferrable_child_process
+      ls = ""
+      EM.run {
+        d = EM::DeferrableChildProcess.open( "ls -ltr" )
+        d.callback {|data_from_child|
+          ls = data_from_child
+          EM.stop
+        }
+      }
+      assert( ls.length > 0)
+    end
+
+    def setup
+      $out = nil
+      $status = nil
+    end
+
+    def test_em_system
+      EM.run{
+        EM.system('ls'){ |out,status| $out, $status = out, status; EM.stop }
+      }
+
+      assert( $out.length > 0 )
+      assert_equal($status.exitstatus, 0)
+      assert_equal($status.class, Process::Status)
+    end
+
+    def test_em_system_pid
+      $pids = []
+
+      EM.run{
+        $pids << EM.system('echo hi', proc{ |out,status|$pids << status.pid; EM.stop })
+      }
+
+      assert_equal $pids[0], $pids[1]
+    end
+
+    def test_em_system_with_proc
+      EM.run{
+        EM.system('ls', proc{ |out,status| $out, $status = out, status; EM.stop })
+      }
+
+      assert( $out.length > 0 )
+      assert_equal($status.exitstatus, 0)
+      assert_equal($status.class, Process::Status)
+    end
+
+    def test_em_system_with_two_procs
+      EM.run{
+        EM.system('sh', proc{ |process|
+          process.send_data("echo hello\n")
+          process.send_data("exit\n")
+        }, proc{ |out,status|
+          $out = out
+          $status = status
+          EM.stop
+        })
+      }
+
+      assert_equal("hello\n", $out)
+    end
+
+    def test_em_system_cmd_arguments
+      EM.run{
+        EM.system('echo', '1', '2', 'version', proc{ |process|
+        }, proc{ |out,status|
+          $out = out
+          $status = status
+          EM.stop
+        })
+      }
+
+      assert_match(/1 2 version/i, $out)
+    end
+
+    def test_em_system_spaced_arguments
+      EM.run{
+        EM.system('ruby', '-e', 'puts "hello"', proc{ |out,status|
+          $out = out
+          EM.stop
+        })
+      }
+
+      assert_equal("hello\n", $out)
+    end
+
+    def test_em_popen_pause_resume
+      c_rx = 0
+
+      test_client = Module.new do
+        define_method :receive_data do |data|
+          c_rx += 1
+          pause
+          EM.add_timer(0.5) { EM.stop }
+        end
+      end
+
+      EM.run{
+        EM.popen('cat /dev/random', test_client)
+      }
+
+      assert_equal 1, c_rx
+    end
+  else
+    warn "EM.popen not implemented, skipping tests in #{__FILE__}"
+
+    # Because some rubies will complain if a TestCase class has no tests
+    def test_em_popen_unsupported
+      assert true
+    end
+  end
+end

data/tests/test_proxy_connection.rb

@@ -0,0 +1,180 @@
+require 'em_test_helper'
+
+class TestProxyConnection < Test::Unit::TestCase
+
+  if EM.respond_to?(:start_proxy)
+    module ProxyConnection
+      def initialize(client, request)
+        @client, @request = client, request
+      end
+
+      def post_init
+        EM::enable_proxy(self, @client)
+      end
+
+      def connection_completed
+        EM.next_tick {
+          send_data @request
+        }
+      end
+
+      def proxy_target_unbound
+        $unbound_early = true
+        EM.stop
+      end
+
+      def unbind
+        $proxied_bytes = self.get_proxied_bytes
+        @client.close_connection_after_writing
+      end
+    end
+
+    module PartialProxyConnection
+      def initialize(client, request, length)
+        @client, @request, @length = client, request, length
+      end
+
+      def post_init
+        EM::enable_proxy(self, @client, 0, @length)
+      end
+
+      def receive_data(data)
+        $unproxied_data = data
+        @client.send_data(data)
+      end
+
+      def connection_completed
+        EM.next_tick {
+          send_data @request
+        }
+      end
+
+      def proxy_target_unbound
+        $unbound_early = true
+        EM.stop
+      end
+
+      def proxy_completed
+        $proxy_completed = true
+      end
+
+      def unbind
+        @client.close_connection_after_writing
+      end
+    end
+
+    module Client
+      def connection_completed
+        send_data "EM rocks!"
+      end
+
+      def receive_data(data)
+        $client_data = data
+      end
+
+      def unbind
+        EM.stop
+      end
+    end
+
+    module Client2
+      include Client
+      def unbind; end
+    end
+
+    module Server
+      def receive_data(data)
+        send_data "I know!" if data == "EM rocks!"
+        close_connection_after_writing
+      end
+    end
+
+    module ProxyServer
+      def initialize port
+        @port = port
+      end
+
+      def receive_data(data)
+        @proxy = EM.connect("127.0.0.1", @port, ProxyConnection, self, data)
+      end
+    end
+
+    module PartialProxyServer
+      def initialize port
+        @port = port
+      end
+
+      def receive_data(data)
+        EM.connect("127.0.0.1", @port, PartialProxyConnection, self, data, 1)
+      end
+    end
+
+    module EarlyClosingProxy
+      def initialize port
+        @port = port
+      end
+
+      def receive_data(data)
+        EM.connect("127.0.0.1", @port, ProxyConnection, self, data)
+        close_connection
+      end
+    end
+
+    def setup
+      @port = next_port
+      @proxy_port = next_port
+    end
+
+    def test_proxy_connection
+      EM.run {
+        EM.start_server("127.0.0.1", @port, Server)
+        EM.start_server("127.0.0.1", @proxy_port, ProxyServer, @port)
+        EM.connect("127.0.0.1", @proxy_port, Client)
+      }
+
+      assert_equal("I know!", $client_data)
+    end
+
+    def test_proxied_bytes
+      EM.run {
+        EM.start_server("127.0.0.1", @port, Server)
+        EM.start_server("127.0.0.1", @proxy_port, ProxyServer, @port)
+        EM.connect("127.0.0.1", @proxy_port, Client)
+      }
+
+      assert_equal("I know!", $client_data)
+      assert_equal("I know!".bytesize, $proxied_bytes)
+    end
+
+    def test_partial_proxy_connection
+      EM.run {
+        EM.start_server("127.0.0.1", @port, Server)
+        EM.start_server("127.0.0.1", @proxy_port, PartialProxyServer, @port)
+        EM.connect("127.0.0.1", @proxy_port, Client)
+      }
+
+      assert_equal("I know!", $client_data)
+      assert_equal(" know!", $unproxied_data)
+      assert($proxy_completed)
+    end
+
+    def test_early_close
+      $client_data = nil
+      EM.run {
+        EM.start_server("127.0.0.1", @port, Server)
+        EM.start_server("127.0.0.1", @proxy_port, EarlyClosingProxy, @port)
+        EM.connect("127.0.0.1", @proxy_port, Client2)
+      }
+
+      assert($unbound_early)
+    end
+  else
+    warn "EM.start_proxy not implemented, skipping tests in #{__FILE__}"
+
+    # Because some rubies will complain if a TestCase class has no tests
+    def test_em_start_proxy_not_implemented
+      assert !EM.respond_to?(:start_proxy)
+    end
+  end
+
+end