wj_eventmachine 1.3.0.dev.1

data/tests/test_error_handler.rb

@@ -0,0 +1,38 @@
+require_relative 'em_test_helper'
+
+class TestErrorHandler < Test::Unit::TestCase
+  def setup
+    @exception = Class.new(StandardError)
+  end
+
+  def test_error_handler
+    error = nil
+
+    EM.error_handler{ |e|
+      error = e
+      EM.error_handler(nil)
+      EM.stop
+    }
+
+    assert_nothing_raised do
+      EM.run{
+        EM.add_timer(0){
+          raise @exception, 'test'
+        }
+      }
+    end
+
+    assert_equal error.class, @exception
+    assert_equal error.message, 'test'
+  end
+
+  def test_without_error_handler
+    assert_raise @exception do
+      EM.run{
+        EM.add_timer(0){
+          raise @exception, 'test'
+        }
+      }
+    end
+  end
+end

data/tests/test_exc.rb

@@ -0,0 +1,37 @@
+require_relative 'em_test_helper'
+
+class TestSomeExceptions < Test::Unit::TestCase
+  class DoomedConnectionError < StandardError
+  end
+  class DoomedConnection < EventMachine::Connection
+    def unbind
+      raise DoomedConnectionError
+    end
+  end
+
+  # Read the commentary in EM#run.
+  # This test exercises the ensure block in #run that makes sure
+  # EM#release_machine gets called even if an exception is
+  # thrown within the user code. Without the ensured call to release_machine,
+  # the second call to EM#run will fail with a C++ exception
+  # because the machine wasn't cleaned up properly.
+
+  def test_a
+    assert_raises(RuntimeError) {
+      EM.run { raise "some exception" }
+    }
+  end
+
+  def test_b
+    assert_raises(RuntimeError) {
+      EM.run { raise "some exception" }
+    }
+  end
+
+  def test_exception_on_unbind
+    assert_raises(DoomedConnectionError) {
+      EM.run { EM.connect("localhost", 8888, DoomedConnection) }
+    }
+  end
+
+end

data/tests/test_file_watch.rb

@@ -0,0 +1,86 @@
+require_relative 'em_test_helper'
+require 'tempfile'
+
+class TestFileWatch < Test::Unit::TestCase
+  if windows?
+    def test_watch_file_raises_unsupported_error
+      pend("\nFIXME: Windows as of 2018-06-23 on 32 bit >= 2.4 (#{RUBY_VERSION} #{RUBY_PLATFORM})") if RUBY_PLATFORM[/i386-mingw/] && RUBY_VERSION >= '2.4'
+      assert_raises(EM::Unsupported) do
+        EM.run do
+          file = Tempfile.new("fake_file")
+          EM.watch_file(file.path)
+        end
+      end
+    end
+  elsif EM.respond_to? :watch_filename
+    module FileWatcher
+      def file_modified
+        $modified = true
+      end
+      def file_deleted
+        $deleted = true
+      end
+      def unbind
+        $unbind = true
+        EM.stop
+      end
+    end
+
+    def setup
+      EM.kqueue = true if EM.kqueue?
+    end
+
+    def teardown
+      EM.kqueue = false if EM.kqueue?
+    end
+
+    def test_events
+      omit_if(solaris?)
+      EM.run{
+        file = Tempfile.new('em-watch')
+        $tmp_path = file.path
+
+        # watch it
+        watch = EM.watch_file(file.path, FileWatcher)
+        $path = watch.path
+
+        # modify it
+        File.open(file.path, 'w'){ |f| f.puts 'hi' }
+
+        # delete it
+        EM.add_timer(0.01){ file.close; file.delete }
+      }
+
+      assert_equal($path, $tmp_path)
+      assert($modified)
+      assert($deleted)
+      assert($unbind)
+    end
+
+    # Refer: https://github.com/eventmachine/eventmachine/issues/512
+    def test_invalid_signature
+      # This works fine with kqueue, only fails with linux inotify.
+      omit_if(EM.kqueue?)
+
+      EM.run {
+        file = Tempfile.new('foo')
+
+        w1 = EventMachine.watch_file(file.path)
+        w2 = EventMachine.watch_file(file.path)
+
+        assert_raise EventMachine::InvalidSignature do
+          w2.stop_watching
+        end
+        w1.stop_watching rescue nil
+        EM.stop
+      }
+    end
+  else
+    warn "EM.watch_file not implemented, skipping tests in #{__FILE__}"
+
+    # Because some rubies will complain if a TestCase class has no tests
+    def test_em_watch_file_unsupported
+      assert true
+    end
+  end
+end

data/tests/test_fork.rb

@@ -0,0 +1,75 @@
+require_relative 'em_test_helper'
+
+class TestFork < Test::Unit::TestCase
+
+  def test_fork_safe
+    omit_if(jruby?)
+    omit_if(windows?)
+
+    fork_pid = nil
+    read, write = IO.pipe
+    EM.run do
+      fork_pid = fork do
+        write.puts "forked"
+        EM.run do
+          EM.next_tick do
+            write.puts "EM ran"
+            EM.stop
+          end
+        end
+      end
+      EM.stop
+    end
+
+    sleep 0.1
+    begin
+      Timeout::timeout 1 do
+        assert_equal "forked\n", read.readline
+        assert_equal "EM ran\n", read.readline
+      end
+    rescue Timeout::Error
+      Process.kill 'TERM', fork_pid
+      flunk "Timeout waiting for next_tick in new fork reactor"
+    end
+  ensure
+    read.close rescue nil
+    write.close rescue nil
+  end
+
+  def test_fork_reactor
+    omit_if(jruby?)
+    omit_if(windows?)
+
+    fork_pid = nil
+    read, write = IO.pipe
+    EM.run do
+      EM.defer do
+        write.puts Process.pid
+        EM.defer do
+          EM.stop
+        end
+      end
+      fork_pid = EM.fork_reactor do
+        EM.defer do
+          write.puts Process.pid
+          EM.stop
+        end
+      end
+    end
+
+    sleep 0.1
+    begin
+      Timeout::timeout 1 do
+        assert_equal Process.pid.to_s, read.readline.chomp
+        assert_equal fork_pid.to_s, read.readline.chomp
+      end
+    rescue Timeout::Error
+      Process.kill 'TERM', fork_pid
+      flunk "Timeout waiting for deferred block in fork_reactor"
+    end
+  ensure
+    read.close rescue nil
+    write.close rescue nil
+  end
+
+end

data/tests/test_futures.rb

@@ -0,0 +1,170 @@
+require_relative 'em_test_helper'
+
+class TestFutures < Test::Unit::TestCase
+
+  def setup
+  end
+
+  def teardown
+  end
+
+  def test_future
+      assert_equal(100, EM::Deferrable.future(100) )
+
+      p1 = proc { 100 + 1 }
+      assert_equal(101, EM::Deferrable.future(p1) )
+  end
+
+  class MyFuture
+      include EM::Deferrable
+      def initialize *args
+        super
+        set_deferred_status :succeeded, 40
+      end
+  end
+
+  class MyErrorFuture
+      include EM::Deferrable
+      def initialize *args
+        super
+        set_deferred_status :failed, 41
+      end
+  end
+
+
+  def test_future_1
+      # Call future with one additional argument and it will be treated as a callback.
+      def my_future
+        MyFuture.new
+      end
+
+      value = nil
+      EM::Deferrable.future my_future, proc {|v| value=v}
+      assert_equal( 40, value )
+  end
+
+
+  def test_future_2
+      # Call future with two additional arguments and they will be treated as a callback
+      # and an errback.
+      value = nil
+      EM::Deferrable.future MyErrorFuture.new, nil, proc {|v| value=v}
+      assert_equal( 41, value )
+  end
+
+
+  def test_future_3
+      # Call future with no additional arguments but with a block, and the block will be
+      # treated as a callback.
+      value = nil
+      EM::Deferrable.future MyFuture.new do |v|
+        value=v
+      end
+      assert_equal( 40, value )
+  end
+
+
+  class RecursiveCallback
+      include EM::Deferrable
+  end
+
+  # A Deferrable callback can call #set_deferred_status to change the values
+  # passed to subsequent callbacks.
+  #
+  def test_recursive_callbacks
+      n = 0 # counter assures that all the tests actually run.
+      rc = RecursiveCallback.new
+      rc.callback {|a|
+        assert_equal(100, a)
+        n += 1
+        rc.set_deferred_status :succeeded, 101, 101
+      }
+      rc.callback {|a,b|
+        assert_equal(101, a)
+        assert_equal(101, b)
+        n += 1
+        rc.set_deferred_status :succeeded, 102, 102, 102
+      }
+      rc.callback {|a,b,c|
+        assert_equal(102, a)
+        assert_equal(102, b)
+        assert_equal(102, c)
+        n += 1
+      }
+      rc.set_deferred_status :succeeded, 100
+      assert_equal(3, n)
+  end
+
+  def test_syntactic_sugar
+    rc = RecursiveCallback.new
+    rc.set_deferred_success 100
+    rc.set_deferred_failure 200
+  end
+
+  # It doesn't raise an error to set deferred status more than once.
+  # In fact, this is a desired and useful idiom when it happens INSIDE
+  # a callback or errback.
+  # However, it's less useful otherwise, and in fact would generally be
+  # indicative of a programming error. However, we would like to be resistant
+  # to such errors. So whenever we set deferred status, we also clear BOTH
+  # stacks of handlers.
+  #
+  def test_double_calls
+    s = 0
+    e = 0
+
+    d = EM::DefaultDeferrable.new
+    d.callback {s += 1}
+    d.errback {e += 1}
+
+    d.succeed	# We expect the callback to be called, and the errback to be DISCARDED.
+    d.fail	  # Presumably an error. We expect the errback NOT to be called.
+    d.succeed	# We expect the callback to have been discarded and NOT to be called again.
+
+    assert_equal(1, s)
+    assert_equal(0, e)
+  end
+
+  # Adding a callback to a Deferrable that is already in a success state executes the callback
+  # immediately. The same applies to a an errback added to an already-failed Deferrable.
+  # HOWEVER, we expect NOT to be able to add errbacks to succeeded Deferrables, or callbacks
+  # to failed ones.
+  #
+  # We illustrate this with a rather contrived test. The test calls #fail after #succeed,
+  # which ordinarily would not happen in a real program.
+  #
+  # What we're NOT attempting to specify is what happens if a Deferrable is succeeded and then
+  # failed (or vice-versa). Should we then be able to add callbacks/errbacks of the appropriate
+  # type for immediate execution? For now at least, the official answer is "don't do that."
+  #
+  def test_delayed_callbacks
+    s1 = 0
+    s2 = 0
+    e = 0
+
+    d = EM::DefaultDeferrable.new
+    d.callback {s1 += 1}
+
+    d.succeed # Triggers and discards the callback.
+
+    d.callback {s2 += 1} # This callback is executed immediately and discarded.
+
+    d.errback {e += 1} # This errback should be DISCARDED and never execute.
+    d.fail # To prove it, fail and assert e is 0
+
+    assert_equal( [1,1], [s1,s2] )
+    assert_equal( 0, e )
+  end
+
+  def test_timeout
+    n = 0
+    EM.run {
+      d = EM::DefaultDeferrable.new
+      d.callback {n = 1; EM.stop}
+      d.errback {n = 2; EM.stop}
+      d.timeout(0.01)
+    }
+    assert_equal( 2, n )
+  end
+
+end

data/tests/test_handler_check.rb

@@ -0,0 +1,35 @@
+require_relative 'em_test_helper'
+
+class TestHandlerCheck < Test::Unit::TestCase
+
+  class Foo < EM::Connection; end;
+  module TestModule; end;
+
+  def test_with_correct_class
+    assert_nothing_raised do
+      EM.run {
+        EM.connect("127.0.0.1", 80, Foo)
+        EM.stop_event_loop
+      }
+    end
+  end
+
+  def test_with_incorrect_class
+    assert_raise(ArgumentError) do
+      EM.run {
+        EM.connect("127.0.0.1", 80, String)
+        EM.stop_event_loop
+      }
+    end
+  end
+
+  def test_with_module
+    assert_nothing_raised do
+      EM.run {
+        EM.connect("127.0.0.1", 80, TestModule)
+        EM.stop_event_loop
+      }
+    end
+  end
+
+end

data/tests/test_hc.rb

@@ -0,0 +1,155 @@
+require_relative 'em_test_helper'
+
+class TestHeaderAndContentProtocol < Test::Unit::TestCase
+
+  class SimpleTest < EM::P::HeaderAndContentProtocol
+    attr_reader :first_header, :my_headers, :request
+
+    def receive_first_header_line hdr
+      @first_header ||= []
+      @first_header << hdr
+    end
+    def receive_headers hdrs
+      @my_headers ||= []
+      @my_headers << hdrs
+    end
+    def receive_request hdrs, content
+      @request ||= []
+      @request << [hdrs, content]
+    end
+  end
+  
+  class StopOnUnbind < EM::Connection
+    def unbind
+      EM.add_timer(0.01) { EM.stop }
+    end
+  end
+
+  def setup
+    @port = next_port
+  end
+
+  def test_no_content
+    the_connection = nil
+    EM.run {
+      EM.start_server( "127.0.0.1", @port, SimpleTest ) do |conn|
+        the_connection = conn
+      end
+      setup_timeout
+
+      EM.connect "127.0.0.1", @port, StopOnUnbind do |c|
+        c.send_data [ "aaa\n", "bbb\r\n", "ccc\n", "\n" ].join
+        c.close_connection_after_writing
+      end
+    }
+    assert_equal( ["aaa"], the_connection.first_header )
+    assert_equal( [%w(aaa bbb ccc)], the_connection.my_headers )
+    assert_equal( [[%w(aaa bbb ccc), ""]], the_connection.request )
+  end
+
+  def test_content
+    the_connection = nil
+    content = "A" * 50
+    headers = ["aaa", "bbb", "Content-length: #{content.length}", "ccc"]
+    EM.run {
+      EM.start_server( "127.0.0.1", @port, SimpleTest ) do |conn|
+        the_connection = conn
+      end
+      setup_timeout
+
+      EM.connect "127.0.0.1", @port, StopOnUnbind do |c|
+        headers.each { |h| c.send_data "#{h}\r\n" }
+        c.send_data "\n"
+        c.send_data content
+        c.close_connection_after_writing
+      end
+    }
+    assert_equal( ["aaa"], the_connection.first_header )
+    assert_equal( [headers], the_connection.my_headers )
+    assert_equal( [[headers, content]], the_connection.request )
+  end
+
+  def test_several_requests
+    the_connection = nil
+    content = "A" * 50
+    headers = ["aaa", "bbb", "Content-length: #{content.length}", "ccc"]
+    EM.run {
+      EM.start_server( "127.0.0.1", @port, SimpleTest ) do |conn|
+        the_connection = conn
+      end
+      setup_timeout
+
+      EM.connect( "127.0.0.1", @port, StopOnUnbind ) do |c|
+        5.times do
+          headers.each { |h| c.send_data "#{h}\r\n" }
+          c.send_data "\n"
+          c.send_data content
+        end
+        c.close_connection_after_writing
+      end
+    }
+    assert_equal( ["aaa"] * 5, the_connection.first_header )
+    assert_equal( [headers] * 5, the_connection.my_headers )
+    assert_equal( [[headers, content]] * 5, the_connection.request )
+  end
+
+
+  # def x_test_multiple_content_length_headers
+  #   # This is supposed to throw a RuntimeError but it throws a C++ exception instead.
+  #   the_connection = nil
+  #   content = "A" * 50
+  #   headers = ["aaa", "bbb", ["Content-length: #{content.length}"]*2, "ccc"].flatten
+  #   EM.run {
+  #     EM.start_server( "127.0.0.1", @port, SimpleTest ) do |conn|
+  #       the_connection = conn
+  #     end
+  #     EM.add_timer(4) {raise "test timed out"}
+  #     test_proc = proc {
+  #       t = TCPSocket.new "127.0.0.1", @port
+  #       headers.each {|h| t.write "#{h}\r\n" }
+  #       t.write "\n"
+  #       t.write content
+  #       t.close
+  #     }
+  #     EM.defer test_proc, proc {
+  #       EM.stop
+  #     }
+  #   }
+  # end
+
+  def test_interpret_headers
+    the_connection = nil
+    content = "A" * 50
+    headers = [
+      "GET / HTTP/1.0",
+      "Accept: aaa",
+      "User-Agent: bbb",
+      "Host: ccc",
+      "x-tempest-header:ddd"
+    ]
+
+    EM.run {
+      EM.start_server( "127.0.0.1", @port, SimpleTest ) do |conn|
+        the_connection = conn
+      end
+      setup_timeout
+
+      EM.connect( "127.0.0.1", @port, StopOnUnbind ) do |c|
+        headers.each { |h| c.send_data "#{h}\r\n" }
+        c.send_data "\n"
+        c.send_data content
+        c.close_connection_after_writing
+      end
+    }
+
+    hsh = the_connection.headers_2_hash( the_connection.my_headers.shift )
+    expect = {
+      :accept => "aaa",
+      :user_agent => "bbb",
+      :host => "ccc",
+      :x_tempest_header => "ddd"
+    }
+    assert_equal(expect, hsh)
+  end
+
+end