wj_eventmachine 1.3.0.dev.1

data/tests/test_channel.rb

@@ -0,0 +1,75 @@
+require_relative 'em_test_helper'
+
+class TestEMChannel < Test::Unit::TestCase
+  def test_channel_subscribe
+    s = 0
+    EM.run do
+      c = EM::Channel.new
+      c.subscribe { |v| s = v; EM.stop }
+      c << 1
+    end
+    assert_equal 1, s
+  end
+
+  def test_channel_unsubscribe
+    s = 0
+    EM.run do
+      c = EM::Channel.new
+      subscription = c.subscribe { |v| s = v }
+      c.unsubscribe(subscription)
+      c << 1
+      EM.next_tick { EM.stop }
+    end
+    assert_not_equal 1, s
+  end
+
+  def test_channel_pop
+    s = 0
+    EM.run do
+      c = EM::Channel.new
+      c.pop{ |v| s = v }
+      c.push(1,2,3)
+      c << 4
+      c << 5
+      EM.next_tick { EM.stop }
+    end
+    assert_equal 1, s
+  end
+
+  def test_channel_reactor_thread_push
+    out = []
+    c = EM::Channel.new
+    c.subscribe { |v| out << v }
+    Thread.new { c.push(1,2,3) }.join
+    assert out.empty?
+
+    EM.run { EM.next_tick { EM.stop } }
+
+    assert_equal [1,2,3], out
+  end
+
+  def test_channel_reactor_thread_callback
+    out = []
+    c = EM::Channel.new
+    Thread.new { c.subscribe { |v| out << v } }.join
+    c.push(1,2,3)
+    assert out.empty?
+
+    EM.run { EM.next_tick { EM.stop } }
+
+    assert_equal [1,2,3], out
+  end
+
+  def test_channel_num_subscribers
+     subs = 0
+     EM.run do
+       c = EM::Channel.new
+       c.subscribe { |v| }
+       c.subscribe { |v| }
+       EM.next_tick { EM.stop }
+       subs = c.num_subscribers
+     end
+
+     assert_equal subs, 2
+  end
+end

data/tests/test_completion.rb

@@ -0,0 +1,178 @@
+require_relative 'em_test_helper'
+require 'em/completion'
+
+class TestCompletion < Test::Unit::TestCase
+  def completion
+    @completion ||= EM::Completion.new
+  end
+
+  def crank
+    # This is a slow solution, but this just executes the next tick queue
+    # once. It's the easiest way for now.
+    EM.run { EM.stop }
+  end
+
+  def results
+    @results ||= []
+  end
+
+  def test_state
+    assert_equal :unknown, completion.state
+  end
+
+  def test_succeed
+    completion.callback { |val| results << val }
+    completion.succeed :object
+    crank
+    assert_equal :succeeded, completion.state
+    assert_equal [:object], results
+  end
+
+  def test_fail
+    completion.errback { |val| results << val }
+    completion.fail :object
+    crank
+    assert_equal :failed, completion.state
+    assert_equal [:object], results
+  end
+
+  def test_callback
+    completion.callback { results << :callback }
+    completion.errback  { results << :errback  }
+    completion.succeed
+    crank
+    assert_equal [:callback], results
+  end
+
+  def test_errback
+    completion.callback { results << :callback }
+    completion.errback  { results << :errback  }
+    completion.fail
+    crank
+    assert_equal [:errback], results
+  end
+
+  def test_stateback
+    completion.stateback(:magic) { results << :stateback }
+    completion.change_state(:magic)
+    crank
+    assert_equal [:stateback], results
+  end
+
+  def test_does_not_enqueue_when_completed
+    completion.callback { results << :callback }
+    completion.succeed
+    completion.errback  { results << :errback  }
+    completion.fail
+    crank
+    assert_equal [:callback], results
+  end
+
+  def test_completed
+    assert_equal false, completion.completed?
+    completion.succeed
+    assert_equal true, completion.completed?
+    completion.fail
+    assert_equal true, completion.completed?
+    completion.change_state :magic
+    assert_equal false, completion.completed?
+  end
+
+  def test_recursive_callbacks
+    completion.callback do |val|
+      results << val
+      completion.succeed :two
+    end
+    completion.callback do |val|
+      results << val
+      completion.succeed :three
+    end
+    completion.callback do |val|
+      results << val
+    end
+    completion.succeed :one
+    crank
+    assert_equal [:one, :two, :three], results
+  end
+
+  def test_late_defined_callbacks
+    completion.callback { results << :one }
+    completion.succeed
+    crank
+    assert_equal [:one], results
+    completion.callback { results << :two }
+    crank
+    assert_equal [:one, :two], results
+  end
+
+  def test_cleared_completions
+    completion.callback { results << :callback }
+    completion.errback  { results << :errback  }
+
+    completion.succeed
+    crank
+    completion.fail
+    crank
+    completion.succeed
+    crank
+
+    assert_equal [:callback], results
+  end
+
+  def test_skip_completed_callbacks
+    completion.callback { results << :callback }
+    completion.succeed
+    crank
+
+    completion.errback  { results << :errback  }
+    completion.fail
+    crank
+
+    assert_equal [:callback], results
+  end
+
+  def test_completions
+    completion.completion { results << :completion }
+    completion.succeed
+    crank
+    assert_equal [:completion], results
+
+    completion.change_state(:unknown)
+    results.clear
+
+    completion.completion { results << :completion }
+    completion.fail
+    crank
+    assert_equal [:completion], results
+  end
+
+  def test_latent_completion
+    completion.completion { results << :completion }
+    completion.succeed
+    crank
+    completion.completion { results << :completion }
+    crank
+    assert_equal [:completion, :completion], results
+  end
+
+  def test_timeout
+    args = [1, 2, 3]
+    EM.run do
+      completion.timeout(0.0001, *args)
+      completion.errback { |*errargs| results << errargs }
+      completion.completion { EM.stop }
+      EM.add_timer(0.1) { flunk 'test timed out' }
+    end
+    assert_equal [[1,2,3]], results
+  end
+
+  def test_timeout_gets_cancelled
+    EM.run do
+      completion.timeout(0.0001, :timeout)
+      completion.errback  { results << :errback  }
+      completion.succeed
+      EM.add_timer(0.0002) { EM.stop }
+    end
+    assert_equal [], results
+  end
+end

data/tests/test_connection_count.rb

@@ -0,0 +1,83 @@
+require_relative 'em_test_helper'
+
+class TestConnectionCount < Test::Unit::TestCase
+  def teardown
+    EM.epoll = false
+    EM.kqueue = false
+  end
+
+  def test_idle_connection_count
+    count = nil
+    EM.run {
+      count = EM.connection_count
+      EM.stop_event_loop
+    }
+    assert_equal(0, count)
+  end
+
+  # Run this again with epoll enabled (if available)
+  def test_idle_connection_count_epoll
+    EM.epoll if EM.epoll?
+
+    count = nil
+    EM.run {
+      count = EM.connection_count
+      EM.stop_event_loop
+    }
+    assert_equal(0, count)
+  end
+
+  # Run this again with kqueue enabled (if available)
+  def test_idle_connection_count_kqueue
+    EM.kqueue if EM.kqueue?
+
+    count = nil
+    EM.run {
+      count = EM.connection_count
+      EM.stop_event_loop
+    }
+    assert_equal(0, count)
+  end
+
+  module Client
+    def connection_completed
+      $client_conns += 1
+      EM.stop if $client_conns == 3
+    end
+  end
+
+  def test_with_some_connections
+    EM.run {
+      $client_conns = 0
+      $initial_conns = EM.connection_count
+      EM.start_server("127.0.0.1", 9999)
+      $server_conns = EM.connection_count
+      3.times { EM.connect("127.0.0.1", 9999, Client) }
+    }
+
+    assert_equal(0, $initial_conns)
+    assert_equal(1, $server_conns)
+    assert_equal(4, $client_conns + $server_conns)
+  end
+
+  module DoubleCloseClient
+    def unbind
+      close_connection
+      $num_close_scheduled_1 = EM.num_close_scheduled
+      EM.next_tick do
+        $num_close_scheduled_2 = EM.num_close_scheduled
+        EM.stop
+      end
+    end
+  end
+
+  def test_num_close_scheduled
+    omit_if(jruby?)
+    EM.run {
+      assert_equal(0, EM.num_close_scheduled)
+      EM.connect("127.0.0.1", 9999, DoubleCloseClient) # nothing listening on 9999
+    }
+    assert_equal(1, $num_close_scheduled_1)
+    assert_equal(0, $num_close_scheduled_2)
+  end
+end

data/tests/test_connection_write.rb

@@ -0,0 +1,35 @@
+require_relative 'em_test_helper'
+
+class TestConnectionWrite < Test::Unit::TestCase
+
+  # This test takes advantage of the fact that EM::_RunSelectOnce iterates over the connections twice:
+  #   - once to determine which ones to call Write() on
+  #   - and once to call Write() on each of them.
+  #
+  # But state may change in the meantime before Write() is finally called.
+  # And that is what we try to exploit to get Write() to be called when bWatchOnly is true, and bNotifyWritable is false,
+  # to cause an assertion failure.
+
+  module SimpleClient
+    def notify_writable
+      $conn2.notify_writable = false  # Being naughty in callback
+      # If this doesn't crash anything, the test passed!
+    end
+  end
+
+  def test_with_naughty_callback
+    EM.run do
+      r1, _ = IO.pipe
+      r2, _ = IO.pipe
+
+      # Adding EM.watches
+      $conn1 = EM.watch(r1, SimpleClient)
+      $conn2 = EM.watch(r2, SimpleClient)
+
+      $conn1.notify_writable = true
+      $conn2.notify_writable = true
+
+      EM.stop
+    end
+  end
+end

data/tests/test_defer.rb

@@ -0,0 +1,35 @@
+require_relative 'em_test_helper'
+
+class TestDefer < Test::Unit::TestCase
+
+  def test_defers
+    n = 0
+    n_times = 20
+    EM.run {
+      n_times.times {
+        work_proc = proc { n += 1 }
+        callback = proc { EM.stop if n == n_times }
+        EM.defer work_proc, callback
+      }
+    }
+    assert_equal( n, n_times )
+  end
+
+  def test_errbacks
+    iterations = 20
+    callback_parameter = rand(100)
+    callback_parameters = []
+    callback_op = proc { callback_parameter }
+    callback = proc { |result| callback_parameters << result }
+    errback_parameter = Exception.new
+    errback_parameters = []
+    errback_op = proc { raise errback_parameter }
+    errback = proc { |error| errback_parameters << error }
+    EventMachine.run do
+      (1..iterations).each { |index| EventMachine.defer(index.even? ? callback_op : errback_op, callback, errback) }
+      EventMachine.add_periodic_timer(0.1) { EventMachine.stop if EventMachine.defers_finished? }
+    end
+    assert_equal(callback_parameters.select { |parameter| parameter == callback_parameter }.length, iterations * 0.5)
+    assert_equal(errback_parameters.select{ |parameter| parameter == errback_parameter }.length, iterations * 0.5)
+  end
+end

data/tests/test_deferrable.rb

@@ -0,0 +1,35 @@
+require_relative 'em_test_helper'
+
+class TestDeferrable < Test::Unit::TestCase
+  class Later
+    include EM::Deferrable
+  end
+
+  def test_timeout_without_args
+    assert_nothing_raised do
+      EM.run {
+        df = Later.new
+        df.timeout(0)
+        df.errback { EM.stop }
+        EM.add_timer(0.01) { flunk "Deferrable was not timed out." }
+      }
+    end
+  end
+
+  def test_timeout_with_args
+    args = nil
+
+    EM.run {
+      df = Later.new
+      df.timeout(0, :timeout, :foo)
+      df.errback do |type, name|
+        args = [type, name]
+        EM.stop
+      end
+
+      EM.add_timer(0.01) { flunk "Deferrable was not timed out." }
+    }
+
+    assert_equal [:timeout, :foo], args
+  end
+end

data/tests/test_epoll.rb

@@ -0,0 +1,141 @@
+require_relative 'em_test_helper'
+
+class TestEpoll < Test::Unit::TestCase
+
+  module TestEchoServer
+    def receive_data data
+      send_data data
+      close_connection_after_writing
+    end
+  end
+
+  module TestEchoClient
+    def connection_completed
+      send_data "ABCDE"
+      $max += 1
+    end
+    def receive_data data
+      raise "bad response" unless data == "ABCDE"
+    end
+    def unbind
+      $n -= 1
+      EM.stop if $n == 0
+    end
+  end
+
+
+  # We can set the rlimit/nofile of a process but we can only set it
+  # higher if we're running as root.
+  # On most systems, the default value is 1024.
+  def test_rlimit
+    omit_if(windows? || jruby?)
+    unless EM.set_descriptor_table_size >= 1024
+      a = EM.set_descriptor_table_size
+      assert( a <= 1024 )
+      a = EM.set_descriptor_table_size( 1024 )
+      assert( a == 1024 )
+    end
+  end
+
+  # Run a high-volume version of this test by kicking the number of connections
+  # up past 512. (Each connection uses two sockets, a client and a server.)
+  # (Will require running the test as root)
+  # This test exercises TCP clients and servers.
+  #
+  # XXX this test causes all sort of weird issues on OSX (when run as part of the suite)
+  def _test_descriptors
+    EM.epoll
+    EM.set_descriptor_table_size 60000
+    EM.run {
+      EM.start_server "127.0.0.1", 9800, TestEchoServer
+      $n = 0
+      $max = 0
+      100.times {
+        EM.connect("127.0.0.1", 9800, TestEchoClient) {$n += 1}
+      }
+    }
+    assert_equal(0, $n)
+    assert_equal(100, $max)
+  end
+
+  def setup
+    @port = next_port
+  end
+
+  module TestDatagramServer
+    def receive_data dgm
+      $in = dgm
+      send_data "abcdefghij"
+    end
+  end
+  module TestDatagramClient
+    def initialize port
+      @port = port
+    end
+
+    def post_init
+      send_datagram "1234567890", "127.0.0.1", @port
+    end
+
+    def receive_data dgm
+      $out = dgm
+      EM.stop
+    end
+  end
+
+  def test_datagrams
+    $in = $out = ""
+    EM.run {
+      EM.open_datagram_socket "127.0.0.1", @port, TestDatagramServer
+      EM.open_datagram_socket "127.0.0.1", 0, TestDatagramClient, @port
+    }
+    assert_equal( "1234567890", $in )
+    assert_equal( "abcdefghij", $out )
+  end
+
+  # XXX this test fails randomly...
+  def _test_unix_domain
+    fn = "/tmp/xxx.chain"
+    EM.epoll
+    EM.set_descriptor_table_size 60000
+    EM.run {
+      # The pure-Ruby version won't let us open the socket if the node already exists.
+      # Not sure, that actually may be correct and the compiled version is wrong.
+      # Pure Ruby also oddly won't let us make that many connections. This test used
+      # to run 100 times. Not sure where that lower connection-limit is coming from in
+      # pure Ruby.
+      # Let's not sweat the Unix-ness of the filename, since this test can't possibly
+      # work on Windows anyway.
+      #
+      File.unlink(fn) if File.exist?(fn)
+      EM.start_unix_domain_server fn, TestEchoServer
+      $n = 0
+      $max = 0
+      50.times {
+        EM.connect_unix_domain(fn, TestEchoClient) {$n += 1}
+      }
+      EM::add_timer(1) { $stderr.puts("test_unix_domain timed out!"); EM::stop }
+    }
+    assert_equal(0, $n)
+    assert_equal(50, $max)
+  ensure
+    File.unlink(fn) if File.exist?(fn)
+  end
+
+  def test_attach_detach
+    EM.epoll
+    EM.run {
+      EM.add_timer(0.01) { EM.stop }
+
+      r, _ = IO.pipe
+
+      # This tests a regression where detach in the same tick as attach crashes EM
+      EM.watch(r) do |connection|
+        connection.detach
+      end
+    }
+
+    assert true
+  end
+end
+