eventmachine-eventmachine 0.12.3

data/tests/test_attach.rb

@@ -0,0 +1,66 @@
+# $Id$
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+#
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+
+$:.unshift "../lib"
+require 'eventmachine'
+require 'socket'
+require 'test/unit'
+
+
+class TestAttach < Test::Unit::TestCase
+
+  Host = "127.0.0.1"
+  Port = 9550
+
+  class EchoServer < EM::Connection
+    def receive_data data
+      send_data data
+    end
+  end
+
+  class EchoClient < EM::Connection
+    def connection_completed
+      $sock.write("abc\n")
+    end
+
+    def notify_readable
+      $read = $sock.readline
+      $fd = detach
+    end
+
+    def unbind
+      EM.next_tick do
+        $sock.write("def\n")
+        EM.add_timer(0.5){ EM.stop }
+      end
+    end
+  end
+
+  def test_attach
+    EM.run{
+      EM.start_server Host, Port, EchoServer
+      $sock = TCPSocket.new Host, Port
+      EM.attach $sock, EchoClient
+    }
+
+    assert_equal $read, "abc\n"
+    assert_equal $fd, $sock.fileno
+    assert_equal false, $sock.closed?
+    assert_equal $sock.readline, "def\n"
+  end
+
+end

data/tests/test_basic.rb

@@ -0,0 +1,231 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 8 April 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+#
+# 
+
+$:.unshift File.expand_path(File.dirname(__FILE__) + "/../lib")
+require 'eventmachine'
+require 'test/unit'
+
+class TestBasic < Test::Unit::TestCase
+
+  def setup
+    assert(!EM.reactor_running?)
+  end
+
+  def teardown
+    assert(!EM.reactor_running?)
+  end
+
+  #-------------------------------------
+
+  def test_libtype
+    lt = EventMachine.library_type
+		em_lib = (ENV["EVENTMACHINE_LIBRARY"] || $eventmachine_library || :xxx).to_sym
+		
+		# Running from test runner, under jruby.
+		if RUBY_PLATFORM == 'java'
+			unless em_lib == :pure_ruby
+				assert_equal( :java, lt )
+				return
+			end
+		end
+		
+    case em_lib
+    when :pure_ruby
+      assert_equal( :pure_ruby, lt )
+    when :extension
+      assert_equal( :extension, lt )
+    when :java
+      assert_equal( :java, lt )
+    else
+			# Running from jruby as a standalone test.
+			if RUBY_PLATFORM == 'java'
+				assert_equal( :java, lt )
+			else
+      	assert_equal( :extension, lt )
+			end
+    end
+  end
+
+  #-------------------------------------
+
+
+  def test_em
+    EventMachine.run {
+      EventMachine.add_timer 0 do
+        EventMachine.stop
+      end
+    }
+  end
+
+  #-------------------------------------
+
+  def test_timer
+    n = 0
+    EventMachine.run {
+      EventMachine.add_periodic_timer(0.1) {
+        n += 1
+        EventMachine.stop if n == 2
+      }
+    }
+  end
+
+  #-------------------------------------
+
+  # This test once threw an already-running exception.
+  module Trivial
+    def post_init
+      EventMachine.stop
+    end
+  end
+
+  def test_server
+    EventMachine.run {
+      EventMachine.start_server "localhost", 9000, Trivial
+      EventMachine.connect "localhost", 9000
+    }
+    assert( true ) # make sure it halts
+  end
+
+  #--------------------------------------
+
+  # EventMachine#run_block starts the reactor loop, runs the supplied block, and then STOPS
+  # the loop automatically. Contrast with EventMachine#run, which keeps running the reactor
+  # even after the supplied block completes.
+  def test_run_block
+    assert !EM.reactor_running?
+      a = nil
+      EM.run_block { a = "Worked" }
+      assert a
+      assert !EM.reactor_running?
+  end
+
+
+  #--------------------------------------
+
+  # TODO! This is an unfinished edge case.
+  # EM mishandles uncaught Ruby exceptions that fire from within #unbind handlers.
+  # A uncaught Ruby exception results in a call to EM::release_machine (which is in an ensure
+  # block in EM::run). But if EM is processing an unbind request, the release_machine call
+  # will cause a segmentation fault.
+  #
+
+  TestHost = "127.0.0.1"
+  TestPort = 9070
+
+  class UnbindError < EM::Connection
+	  def initialize *args
+		  super
+	  end
+	  def connection_completed
+		  close_connection_after_writing
+	  end
+	  def unbind
+		  raise "Blooey"
+	  end
+  end
+
+  def xxx_test_unbind_error
+	  assert_raise( RuntimeError ) {
+		  EM.run {
+			  EM.start_server TestHost, TestPort
+			  EM.connect TestHost, TestPort, UnbindError
+		  }
+	  }
+  end
+
+  #------------------------------------
+  #
+  # TODO. This is an unfinished bug fix.
+  # This case was originally reported by Dan Aquino. If you throw a Ruby exception
+  # in a post_init handler, it gets rethrown as a confusing reactor exception.
+  # The problem is in eventmachine.rb, which calls post_init within the private
+  # initialize method of the EM::Connection class. This happens in both the EM::connect
+  # method and in the code that responds to connection-accepted events.
+  # What happens is that we instantiate the new connection object, which calls
+  # initialize, and then after initialize returns, we stick the new connection object
+  # into EM's @conns hashtable.
+  # But the problem is that Connection::initialize calls #post_init before it returns,
+  # and this may be user-written code that may throw an uncaught Ruby exception.
+  # If that happens, the reactor will abort, and it will then try to run down open
+  # connections. Because @conns never got a chance to properly reflect the new connection
+  # (because initialize never returned), we throw a ConnectionNotBound error
+  # (eventmachine.rb line 1080).
+  # When the bug is fixed, activate this test case.
+  #
+
+  class PostInitError < EM::Connection
+	  def post_init
+		  aaa bbb # should produce a Ruby exception
+	  end
+  end
+  # This test causes issues, the machine becomes unreleasable after 
+  # release_machine suffers an exception in event_callback.
+  def xxx_test_post_init_error
+	  assert_raise( EventMachine::ConnectionNotBound ) {
+		  EM.run {
+		  	EM::Timer.new(1) {EM.stop}
+			EM.start_server TestHost, TestPort
+			EM.connect TestHost, TestPort, PostInitError
+		  }
+	  }
+	  EM.run {
+	    EM.stop
+	  }
+	  assert !EM.reactor_running?
+  end
+  
+  module BrsTestSrv
+    def receive_data data
+      $received << data
+    end
+    def unbind
+      EM.stop
+    end
+  end
+  module BrsTestCli
+    def post_init
+      send_data $sent
+      close_connection_after_writing
+    end
+  end
+  
+  # From ticket #50
+  def test_byte_range_send
+    $received = ''
+    $sent = (0..255).to_a.pack('C*')
+    EM::run {
+      
+      EM::start_server TestHost, TestPort, BrsTestSrv
+      
+      EM::connect TestHost, TestPort, BrsTestCli
+      
+      EM::add_timer(0.5) { assert(false, 'test timed out'); EM.stop; Kernel.warn "test timed out!" }
+    }
+    assert_equal($sent, $received)
+  end
+
+end
+

data/tests/test_defer.rb

@@ -0,0 +1,47 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 8 April 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+#
+#
+
+$:.unshift "../lib"
+require 'eventmachine'
+require 'test/unit'
+
+class TestDeferUsage < 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 unless RUBY_VERSION >= '1.9.0'
+
+end
+

data/tests/test_epoll.rb

@@ -0,0 +1,161 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 8 April 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) the GNU General Public License
+# as published by the Free Software Foundation; either version 2 of the
+# License, or (at your option) any later version; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+#
+#
+# TODO, and I know this doesn't belong here, but if a datagram calls
+# send_data outside of a receive_data, there is no return address, and
+# the result is a very confusing error message.
+#
+
+require 'eventmachine'
+require 'test/unit'
+
+
+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.
+	# Java doesn't (currently) implement this.
+	def test_rlimit
+		unless RUBY_PLATFORM =~ /java/
+			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.
+	#
+	def test_descriptors
+		EM.epoll
+		s = 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 test_defer
+    n = 0
+    work_proc = proc {n += 1}
+    callback_proc = proc {EM.stop}
+    EM.epoll
+    EM.run {
+      EM.defer work_proc, callback_proc
+    }
+    assert_equal( 1, n )
+  end unless RUBY_VERSION >= '1.9.0'
+
+
+	module TestDatagramServer
+		def receive_data dgm
+			$in = dgm
+			send_data "abcdefghij"
+		end
+	end
+	module TestDatagramClient
+		def post_init
+			send_datagram "1234567890", "127.0.0.1", 9500
+		end
+		def receive_data dgm
+			$out = dgm
+			EM.stop
+		end
+	end
+
+	def test_datagrams
+		$in = $out = ""
+		EM.epoll
+		EM.run {
+			EM.open_datagram_socket "127.0.0.1", 9500, TestDatagramServer
+			EM.open_datagram_socket "127.0.0.1", 0, TestDatagramClient
+		}
+		assert_equal( "1234567890", $in )
+		assert_equal( "abcdefghij", $out )
+	end
+
+
+
+
+	def test_unix_domain
+		EM.epoll
+		s = 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.
+			#
+			fn = "/tmp/xxx.chain"
+			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)
+	end
+
+end
+