eventmachine-mkroman 1.3.0.dev.1

data/tests/test_ltp2.rb

@@ -0,0 +1,332 @@
+require_relative 'em_test_helper'
+
+# TODO!!! Need tests for overlength headers and text bodies.
+
+class TestLineText2 < Test::Unit::TestCase
+
+  # Run each of these tests two ways: passing in the whole test-dataset in one chunk,
+  # and passing it in one character at a time.
+
+  class Basic
+    include EM::Protocols::LineText2
+    attr_reader :lines
+    def receive_line line
+      (@lines ||= []) << line
+    end
+  end
+  def test_basic
+    testdata = "Line 1\nLine 2\r\nLine 3\n"
+
+    a = Basic.new
+    a.receive_data testdata
+    assert_equal( ["Line 1", "Line 2", "Line 3"], a.lines )
+
+    a = Basic.new
+    testdata.length.times {|i| a.receive_data( testdata[i...i+1] ) }
+    assert_equal( ["Line 1", "Line 2", "Line 3"], a.lines )
+  end
+
+  # The basic test above shows that extra newlines are chomped
+  # This test shows that newlines are preserved if the delimiter isn't \n
+  class PreserveNewlines
+    include EM::Protocols::LineText2
+    attr_reader :lines
+    def initialize *args
+      super
+      @delim = "|"
+      set_delimiter @delim
+    end
+    def receive_line line
+      (@lines ||= []) << line
+    end
+  end
+  def test_preserve_newlines
+    a = PreserveNewlines.new
+    a.receive_data "aaa|bbb|ccc|\n|\r\n| \t ||"
+    assert_equal( ["aaa", "bbb", "ccc", "\n", "\r\n", " \t ", ""], a.lines )
+  end
+
+  class ChangeDelimiter
+    include EM::Protocols::LineText2
+    attr_reader :lines
+    def initialize *args
+      super
+      @delim = "A"
+      set_delimiter @delim
+    end
+    def receive_line line
+      (@lines ||= []) << line
+      set_delimiter( @delim.succ! )
+    end
+  end
+
+  def test_change_delimiter
+    testdata = %Q(LineaALinebBLinecCLinedD)
+
+    a = ChangeDelimiter.new
+    a.receive_data testdata
+    assert_equal( ["Linea", "Lineb", "Linec", "Lined"], a.lines )
+
+    a = ChangeDelimiter.new
+    testdata.length.times {|i| a.receive_data( testdata[i...i+1] ) }
+    assert_equal( ["Linea", "Lineb", "Linec", "Lined"], a.lines )
+  end
+
+  class RegexDelimiter
+    include EM::Protocols::LineText2
+    attr_reader :lines
+    def initialize *args
+      super
+      @delim = /[A-D]/
+      set_delimiter @delim
+    end
+    def receive_line line
+      (@lines ||= []) << line
+    end
+  end
+
+  def test_regex_delimiter
+    testdata = %Q(LineaALinebBLinecCLinedD)
+
+    a = RegexDelimiter.new
+    a.receive_data testdata
+    assert_equal( ["Linea", "Lineb", "Linec", "Lined"], a.lines )
+
+    a = RegexDelimiter.new
+    testdata.length.times {|i| a.receive_data( testdata[i...i+1] ) }
+    assert_equal( ["Linea", "Lineb", "Linec", "Lined"], a.lines )
+  end
+
+  #--
+  # Test two lines followed by an empty line, ten bytes of binary data, then
+  # two more lines.
+
+  class Binary
+    include EM::Protocols::LineText2
+    attr_reader :lines, :body
+    def initialize *args
+      super
+      @lines = []
+      @body = nil
+    end
+    def receive_line ln
+      if ln == ""
+        set_text_mode 10
+      else
+        @lines << ln
+      end
+    end
+    def receive_binary_data data
+      @body = data
+    end
+  end
+
+  def test_binary
+    testdata = %Q(Line 1
+Line 2
+
+0000000000Line 3
+Line 4
+)
+
+    a = Binary.new
+    a.receive_data testdata
+    assert_equal( ["Line 1", "Line 2", "Line 3", "Line 4"], a.lines)
+    assert_equal( "0000000000", a.body )
+
+    a = Binary.new
+    testdata.length.times {|i| a.receive_data( testdata[i...i+1] ) }
+    assert_equal( ["Line 1", "Line 2", "Line 3", "Line 4"], a.lines)
+    assert_equal( "0000000000", a.body )
+  end
+
+
+  # Test unsized binary data. The expectation is that each chunk of it
+  # will be passed to us as it it received.
+  class UnsizedBinary
+    include EM::Protocols::LineText2
+    attr_reader :n_calls, :body
+    def initialize *args
+      super
+      set_text_mode
+    end
+    def receive_binary_data data
+      @n_calls ||= 0
+      @n_calls += 1
+      (@body ||= "") << data
+    end
+  end
+
+  def test_unsized_binary
+    testdata = "X\0" * 1000
+
+    a = UnsizedBinary.new
+    a.receive_data testdata
+    assert_equal( 1, a.n_calls )
+    assert_equal( testdata, a.body )
+
+    a = UnsizedBinary.new
+    testdata.length.times {|i| a.receive_data( testdata[i...i+1] ) }
+    assert_equal( 2000, a.n_calls )
+    assert_equal( testdata, a.body )
+  end
+
+
+  # Test binary data with a "throw back" into line-mode.
+  class ThrowBack
+    include EM::Protocols::LineText2
+    attr_reader :headers
+    def initialize *args
+      super
+      @headers = []
+      @n_bytes = 0
+      set_text_mode
+    end
+    def receive_binary_data data
+      wanted = 25 - @n_bytes
+      will_take = if data.length > wanted
+                    data.length - wanted
+                  else
+                    data.length
+                  end
+      @n_bytes += will_take
+
+      if @n_bytes == 25
+        set_line_mode( data[will_take..-1] )
+      end
+    end
+    def receive_line ln
+      @headers << ln
+    end
+  end
+  def test_throw_back
+    testdata = "Line\n" * 10
+
+    a = ThrowBack.new
+    a.receive_data testdata
+    assert_equal( ["Line"] * 5, a.headers )
+
+    a = ThrowBack.new
+    testdata.length.times {|i| a.receive_data( testdata[i...i+1] ) }
+    assert_equal( ["Line"] * 5, a.headers )
+  end
+
+  # Test multi-character line delimiters.
+  # Also note that the test data has a "tail" with no delimiter, that will be
+  # discarded, but cf. the BinaryTail test.
+  # TODO!!! This test doesn't work in the byte-by-byte case.
+  class Multichar
+    include EM::Protocols::LineText2
+    attr_reader :lines
+    def initialize *args
+      super
+      @lines = []
+      set_delimiter "012"
+    end
+    def receive_line ln
+      @lines << ln
+    end
+  end
+  def test_multichar
+    testdata = "Line012Line012Line012Line"
+
+    a = Multichar.new
+    a.receive_data testdata
+    assert_equal( ["Line"]*3, a.lines )
+
+    a = Multichar.new
+    testdata.length.times {|i| a.receive_data( testdata[i...i+1] ) }
+    # DOESN'T WORK in this case. Multi-character delimiters are broken.
+    #assert_equal( ["Line"]*3, a.lines )
+  end
+
+  # Test a binary "tail," when a sized binary transfer doesn't complete because
+  # of an unbind. We get a partial result.
+  class BinaryTail
+    include EM::Protocols::LineText2
+    attr_reader :data
+    def initialize *args
+      super
+      @data = ""
+      set_text_mode 1000
+    end
+    def receive_binary_data data
+      # we expect to get all the data in one chunk, even in the byte-by-byte case,
+      # because sized transfers by definition give us exactly one call to
+      # #receive_binary_data.
+      @data = data
+    end
+  end
+  def test_binary_tail
+    testdata = "0" * 500
+
+    a = BinaryTail.new
+    a.receive_data testdata
+    a.unbind
+    assert_equal( "0" * 500, a.data )
+
+    a = BinaryTail.new
+    testdata.length.times {|i| a.receive_data( testdata[i...i+1] ) }
+    a.unbind
+    assert_equal( "0" * 500, a.data )
+  end
+
+
+  # Test an end-of-binary call. Arrange to receive binary data but don't bother counting it
+  # as it comes. Rely on getting receive_end_of_binary_data to signal the transition back to
+  # line mode.
+  # At the present time, this isn't strictly necessary with sized binary chunks because by
+  # definition we accumulate them and make exactly one call to receive_binary_data, but
+  # we may want to support a mode in the future that would break up large chunks into multiple
+  # calls.
+  class LazyBinary
+    include EM::Protocols::LineText2
+    attr_reader :data, :end
+    def initialize *args
+      super
+      @data = ""
+      set_text_mode 1000
+    end
+    def receive_binary_data data
+      # we expect to get all the data in one chunk, even in the byte-by-byte case,
+      # because sized transfers by definition give us exactly one call to
+      # #receive_binary_data.
+      @data = data
+    end
+    def receive_end_of_binary_data
+      @end = true
+    end
+  end
+  def test_receive_end_of_binary_data
+    testdata = "_" * 1000
+    a = LazyBinary.new
+    testdata.length.times {|i| a.receive_data( testdata[i...i+1] ) }
+    assert_equal( "_" * 1000, a.data )
+    assert( a.end )
+  end
+
+
+  # This tests a bug fix in which calling set_text_mode failed when called
+  # inside receive_binary_data.
+  #
+  class BinaryPair
+    include EM::Protocols::LineText2
+    attr_reader :sizes
+    def initialize *args
+      super
+      set_text_mode 1
+      @sizes = []
+    end
+    def receive_binary_data dt
+      @sizes <<  dt.length
+      set_text_mode( (dt.length == 1) ? 2 : 1 )
+    end
+  end
+  def test_binary_pairs
+    test_data = "123" * 5
+    a = BinaryPair.new
+    a.receive_data test_data
+    assert_equal( [1,2,1,2,1,2,1,2,1,2], a.sizes )
+  end
+
+end

data/tests/test_many_fds.rb

@@ -0,0 +1,21 @@
+require_relative 'em_test_helper'
+
+class TestManyFDs < Test::Unit::TestCase
+  def setup
+    @port = next_port
+  end
+
+  def test_connection_class_cache
+    mod = Module.new
+    a = nil
+    Process.setrlimit(Process::RLIMIT_NOFILE, 4096) rescue nil
+    EM.run {
+      EM.start_server '127.0.0.1', @port, mod
+      1100.times do
+        a = EM.connect '127.0.0.1', @port, mod
+        assert_kind_of EM::Connection, a
+      end
+      EM.stop
+    }
+  end
+end

data/tests/test_next_tick.rb

@@ -0,0 +1,104 @@
+require_relative 'em_test_helper'
+
+class TestNextTick < Test::Unit::TestCase
+
+  def test_tick_arg
+    pr = proc {EM.stop}
+    EM.run {
+      EM.next_tick pr
+    }
+    assert true
+  end
+
+  def test_tick_block
+    EM.run {
+      EM.next_tick {EM.stop}
+    }
+    assert true
+  end
+
+  # This illustrates the solution to a long-standing problem.
+  # It's now possible to correctly nest calls to EM#run.
+  # See the source code commentary for EM#run for more info.
+  #
+  def test_run_run
+    EM.run {
+      EM.run {
+        EM.next_tick {EM.stop}
+      }
+    }
+  end
+
+  def test_pre_run_queue
+    x = false
+    EM.next_tick { EM.stop; x = true }
+    EM.run { EM.add_timer(0.01) { EM.stop } }
+    assert x
+  end
+
+  def test_cleanup_after_stop
+    x = true
+    EM.run{
+      EM.next_tick{
+        EM.stop
+        EM.next_tick{ x=false }
+      }
+    }
+    EM.run{
+      EM.next_tick{ EM.stop }
+    }
+    assert x
+  end
+
+  # We now support an additional parameter for EM#run.
+  # You can pass two procs to EM#run now. The first is executed as the normal
+  # run block. The second (if given) is scheduled for execution after the
+  # reactor loop completes.
+  # The reason for supporting this is subtle. There has always been an expectation
+  # that EM#run doesn't return until after the reactor loop ends. But now it's
+  # possible to nest calls to EM#run, which means that a nested call WILL
+  # RETURN. In order to write code that will run correctly either way, it's
+  # recommended to put any code which must execute after the reactor completes
+  # in the second parameter.
+  #
+  def test_run_run_2
+    a = proc {EM.stop}
+    b = proc {assert true}
+    EM.run a, b
+  end
+
+
+  # This illustrates that EM#run returns when it's called nested.
+  # This isn't a feature, rather it's something to be wary of when writing code
+  # that must run correctly even if EM#run is called while a reactor is already
+  # running.
+  def test_run_run_3
+    a = []
+    EM.run {
+      EM.run proc {EM.stop}, proc {a << 2}
+      a << 1
+    }
+    assert_equal( [1,2], a )
+  end
+
+
+  def test_schedule_on_reactor_thread
+    x = false
+    EM.run do
+      EM.schedule { x = true }
+      EM.stop
+    end
+    assert x
+  end
+
+  def test_schedule_from_thread
+    x = false
+    EM.run do
+      Thread.new { EM.schedule { x = true } }.join
+      assert !x
+      EM.next_tick { EM.stop }
+    end
+    assert x
+  end
+
+end

data/tests/test_object_protocol.rb

@@ -0,0 +1,36 @@
+require_relative 'em_test_helper'
+
+class TestObjectProtocol < Test::Unit::TestCase
+  module Server
+    include EM::P::ObjectProtocol
+    def post_init
+      send_object :hello=>'world'
+    end
+    def receive_object obj
+      $server = obj
+      EM.stop
+    end
+  end
+
+  module Client
+    include EM::P::ObjectProtocol
+    def receive_object obj
+      $client = obj
+      send_object 'you_said'=>obj
+    end
+  end
+
+  def setup
+    @port = next_port
+  end
+
+  def test_send_receive
+    EM.run{
+      EM.start_server "127.0.0.1", @port, Server
+      EM.connect "127.0.0.1", @port, Client
+    }
+
+    assert($client == {:hello=>'world'})
+    assert($server == {'you_said'=>{:hello=>'world'}})
+  end
+end

data/tests/test_pause.rb

@@ -0,0 +1,109 @@
+require_relative 'em_test_helper'
+
+class TestPause < Test::Unit::TestCase
+  if EM.respond_to? :pause_connection
+    def setup
+      @port = next_port
+    end
+
+    def teardown
+      assert(!EM.reactor_running?)
+    end
+
+    def test_pause_resume
+      server = nil
+
+      s_rx = c_rx = 0
+
+      test_server = Module.new do
+        define_method :post_init do
+          server = self
+        end
+
+        define_method :receive_data do |data|
+          s_rx += 1
+
+          EM.add_periodic_timer(0.01) { send_data 'hi' }
+          send_data 'hi'
+
+          # pause server, now no outgoing data will actually
+          # be sent and no more incoming data will be received
+          pause
+        end
+      end
+
+      test_client = Module.new do
+        def post_init
+          EM.add_periodic_timer(0.01) do
+            send_data 'hello'
+          end
+        end
+
+        define_method :receive_data do |data|
+          c_rx += 1
+        end
+      end
+
+      EM.run do
+        EM.start_server "127.0.0.1", @port, test_server
+        EM.connect "127.0.0.1", @port, test_client
+
+        tmr = darwin? ? 0.10 : 0.05
+
+        EM.add_timer(tmr) do
+          assert_equal 1, s_rx
+          assert_equal 0, c_rx
+          assert server.paused?
+
+          # resume server, queued outgoing and incoming data will be flushed
+          server.resume
+
+          assert !server.paused?
+
+          EM.add_timer(tmr) do
+            assert server.paused?
+            assert s_rx > 1
+            assert c_rx > 0
+            EM.stop
+          end
+        end
+      end
+    end
+
+    def test_pause_in_receive_data
+      incoming = []
+
+      test_server = Module.new do
+        define_method(:receive_data) do |data|
+          incoming << data
+          pause
+          EM.add_timer(0.5){ close_connection }
+        end
+        define_method(:unbind) do
+          EM.stop
+        end
+      end
+
+      buf = 'a' * 1024
+
+      EM.run do
+        EM.start_server "127.0.0.1", @port, test_server
+        cli = EM.connect "127.0.0.1", @port
+        128.times do
+          cli.send_data buf
+        end
+      end
+
+      assert_equal 1, incoming.size
+      assert incoming[0].bytesize > buf.bytesize
+      assert incoming[0].bytesize < buf.bytesize * 128
+    end
+  else
+    warn "EM.pause_connection not implemented, skipping tests in #{__FILE__}"
+
+    # Because some rubies will complain if a TestCase class has no tests
+    def test_em_pause_connection_not_implemented
+      assert true
+    end
+  end
+end

data/tests/test_pending_connect_timeout.rb

@@ -0,0 +1,52 @@
+require_relative '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 = EM.current_time
+          EM.stop
+        end
+      end
+
+      EM.run {
+        setup_timeout 0.4
+        EM.heartbeat_interval = 0.1
+        start = EM.current_time
+        c = EM.connect('192.0.2.0', 54321, timeout_handler)
+        c.pending_connect_timeout = 0.2
+      }
+      # Travis can vary from 0.10 to 0.40
+      assert_in_delta(0.25, (finish - start), 0.15)
+    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