crusher-eventmachine 0.12.11

data/tests/test_ssl_args.rb

@@ -0,0 +1,79 @@
+require "test/unit"
+require 'tempfile'
+
+$:.unshift File.expand_path(File.dirname(__FILE__) + '/../lib')
+require "eventmachine"
+
+module EventMachine
+  def self._set_mocks
+    class <<self
+      alias set_tls_parms_old set_tls_parms
+      alias start_tls_old start_tls
+      begin
+        old, $VERBOSE = $VERBOSE, nil
+        def set_tls_parms *args; end
+        def start_tls *args; end
+      ensure
+        $VERBOSE = old
+      end
+    end
+  end
+
+  def self._clear_mocks
+    class <<self
+      begin
+        old, $VERBOSE = $VERBOSE, nil
+        alias set_tls_parms set_tls_parms_old
+        alias start_tls start_tls_old
+      ensure
+        $VERBOSE = old
+      end
+    end
+  end
+end
+
+  
+
+class TestSslArgs < Test::Unit::TestCase
+  def setup
+    EventMachine._set_mocks
+  end
+  
+  def teardown
+    EventMachine._clear_mocks
+  end
+  
+  def test_tls_params_file_doesnt_exist
+    priv_file, cert_file = 'foo_priv_key', 'bar_cert_file'
+    [priv_file, cert_file].all? do |f|
+      assert(!File.exists?(f), "Cert file #{f} seems to exist, and should not for the tests")
+    end
+    
+    # associate_callback_target is a pain! (build!)
+    conn = EventMachine::Connection.new('foo')
+    
+    assert_raises(EventMachine::FileNotFoundException) do
+      conn.start_tls(:private_key_file => priv_file)
+    end
+    assert_raises(EventMachine::FileNotFoundException) do
+      conn.start_tls(:cert_chain_file => cert_file)
+    end
+    assert_raises(EventMachine::FileNotFoundException) do
+      conn.start_tls(:private_key_file => priv_file, :cert_chain_file => cert_file)
+    end
+  end
+  
+  def test_tls_params_file_does_exist
+    priv_file = Tempfile.new('em_test')
+    cert_file = Tempfile.new('em_test')
+    priv_file_path = priv_file.path
+    cert_file_path = cert_file.path
+    conn = EventMachine::Connection.new('foo')
+    params = {:private_key_file => priv_file_path, :cert_chain_file => cert_file_path}
+    begin
+      conn.start_tls params
+    rescue Object
+      assert(false, 'should not have raised an exception')
+    end
+  end
+end if EM.ssl?

data/tests/test_ssl_methods.rb

@@ -0,0 +1,50 @@
+$:.unshift "../lib"
+require 'eventmachine'
+require 'test/unit'
+
+class TestSSLMethods < Test::Unit::TestCase
+
+  module ServerHandler
+
+    def post_init
+      start_tls
+    end
+
+    def ssl_handshake_completed
+      $server_called_back = true
+      $server_cert_value = get_peer_cert
+    end
+
+  end
+
+  module ClientHandler
+
+    def post_init
+      start_tls
+    end
+
+    def ssl_handshake_completed
+      $client_called_back = true
+      $client_cert_value = get_peer_cert
+      EM.stop_event_loop
+    end
+
+  end
+
+  def test_ssl_methods
+    $server_called_back, $client_called_back = false, false
+    $server_cert_value, $client_cert_value = nil, nil
+
+    EM.run {
+      EM.start_server("127.0.0.1", 9999, ServerHandler)
+      EM.connect("127.0.0.1", 9999, ClientHandler)
+    }
+
+    assert($server_called_back)
+    assert($client_called_back)
+
+    assert($server_cert_value.is_a?(NilClass))
+    assert($client_cert_value.is_a?(String))
+  end
+
+end if EM.ssl?

data/tests/test_ssl_verify.rb

@@ -0,0 +1,82 @@
+$:.unshift "../lib"
+require 'eventmachine'
+require 'test/unit'
+
+class TestSslVerify < Test::Unit::TestCase
+
+  def setup
+    $dir = File.dirname(File.expand_path(__FILE__)) + '/'
+    $cert_from_file = File.read($dir+'client.crt')
+  end
+
+  module Client
+    def connection_completed
+      start_tls(:private_key_file => $dir+'client.key', :cert_chain_file => $dir+'client.crt')
+    end
+
+    def ssl_handshake_completed
+      $client_handshake_completed = true
+      close_connection
+    end
+
+    def unbind
+      EM.stop_event_loop
+    end
+  end
+
+  module AcceptServer
+    def post_init
+      start_tls(:verify_peer => true)
+    end
+
+    def ssl_verify_peer(cert)
+      $cert_from_server = cert
+      true
+    end
+
+    def ssl_handshake_completed
+      $server_handshake_completed = true
+    end
+  end
+
+  module DenyServer
+    def post_init
+      start_tls(:verify_peer => true)
+    end
+
+    def ssl_verify_peer(cert)
+      $cert_from_server = cert
+      # Do not accept the peer. This should now cause the connection to shut down without the SSL handshake being completed.
+      false
+    end
+
+    def ssl_handshake_completed
+      $server_handshake_completed = true
+    end
+  end
+
+  def test_accept_server
+    $client_handshake_completed, $server_handshake_completed = false, false
+    EM.run {
+      EM.start_server("127.0.0.1", 16784, AcceptServer)
+      EM.connect("127.0.0.1", 16784, Client).instance_variable_get("@signature")
+    }
+
+    assert_equal($cert_from_file, $cert_from_server)
+    assert($client_handshake_completed)
+    assert($server_handshake_completed)
+  end
+
+  def test_deny_server
+    $client_handshake_completed, $server_handshake_completed = false, false
+    EM.run {
+      EM.start_server("127.0.0.1", 16784, DenyServer)
+      EM.connect("127.0.0.1", 16784, Client)
+    }
+
+    assert_equal($cert_from_file, $cert_from_server)
+    assert(!$client_handshake_completed)
+    assert(!$server_handshake_completed)
+  end
+
+end

data/tests/test_tick_loop.rb

@@ -0,0 +1,59 @@
+require "test/unit"
+require "eventmachine"
+
+class TestEmTickLoop < Test::Unit::TestCase
+  def test_em_tick_loop
+    i = 0
+    EM.tick_loop { i += 1; EM.stop if i == 10 }
+    EM.run { EM.add_timer(1) { EM.stop } }
+    assert_equal i, 10
+  end
+  
+  def test_tick_loop_on_stop
+    t = nil
+    tick_loop = EM.tick_loop { :stop }
+    tick_loop.on_stop { t = true }
+    EM.run { EM.next_tick { EM.stop } }
+    assert t
+  end
+  
+  def test_start_twice
+    i = 0
+    s = 0
+    tick_loop = EM.tick_loop { i += 1; :stop }
+    tick_loop.on_stop { s += 1; EM.stop }
+    EM.run { EM.next_tick { EM.stop } }
+    assert_equal 1, i
+    assert_equal 1, s
+    tick_loop.start
+    EM.run { EM.next_tick { EM.stop } }
+    assert_equal 2, i
+    assert_equal 1, s # stop callbacks are only called once
+  end
+  
+  def test_stop
+    i, s = 0, 0
+    tick_loop = EM.tick_loop { i += 1 }
+    tick_loop.on_stop { s += 1 }
+    EM.run { EM.next_tick { tick_loop.stop; EM.next_tick { EM.stop } } }
+    assert tick_loop.stopped?
+    assert_equal 1, i
+    assert_equal 1, s
+  end
+  
+  def test_immediate_stops
+    s = 0
+    tick_loop = EM::TickLoop.new { }
+    tick_loop.on_stop { s += 1 }
+    tick_loop.on_stop { s += 1 }
+    assert_equal 2, s
+  end
+  
+  def test_stopped
+    tick_loop = EM::TickLoop.new { }
+    assert tick_loop.stopped?
+    tick_loop.start
+    assert !tick_loop.stopped?
+  end
+    
+end

data/tests/test_timers.rb

@@ -0,0 +1,160 @@
+# $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 TestTimers < Test::Unit::TestCase
+
+  def test_timer_with_block
+    x = false
+    EventMachine.run {
+      EventMachine::Timer.new(0.25) {
+        x = true
+        EventMachine.stop
+      }
+    }
+    assert x
+  end
+
+  def test_timer_with_proc
+    x = false
+    EventMachine.run {
+      EventMachine::Timer.new(0.25, proc {
+        x = true
+        EventMachine.stop
+      })
+    }
+    assert x
+  end
+
+  def test_timer_cancel
+    x = true
+    EventMachine.run {
+      timer = EventMachine::Timer.new(0.25, proc { x = false })
+      timer.cancel
+      EventMachine::Timer.new(0.5, proc {EventMachine.stop})
+    }
+    assert x
+  end
+
+  def test_periodic_timer
+    x = 0
+    EventMachine.run {
+      EventMachine::PeriodicTimer.new(0.1) do
+        x += 1
+        EventMachine.stop if x == 4
+      end
+    }
+    assert( x == 4 )
+  end
+
+  def test_add_periodic_timer
+    x = 0
+    EM.run {
+      t = EM.add_periodic_timer(0.1) do
+        x += 1
+        EM.stop  if x == 4
+      end
+      assert t.respond_to?(:cancel)
+    }
+  end
+
+  def test_periodic_timer_cancel
+    x = 0
+    EventMachine.run {
+      pt = EventMachine::PeriodicTimer.new(0.25, proc { x += 1 })
+      pt.cancel
+      EventMachine::Timer.new(0.5) {EventMachine.stop}
+    }
+    assert( x == 0 )
+  end
+
+  def test_add_periodic_timer_cancel
+    x = 0
+    EventMachine.run {
+      pt = EM.add_periodic_timer(0.1) { x += 1 }
+      EM.cancel_timer(pt)
+      EM.add_timer(0.2) { EM.stop }
+    }
+    assert( x == 0 )
+  end
+
+  def test_periodic_timer_self_cancel
+    x = 0
+    EventMachine.run {
+      pt = EventMachine::PeriodicTimer.new(0.1) {
+        x += 1
+        if x == 4
+          pt.cancel
+          EventMachine.stop
+        end
+      }
+    }
+    assert( x == 4 )
+  end
+
+
+  # This test is only applicable to compiled versions of the reactor.
+  # Pure ruby and java versions have no built-in limit on the number of outstanding timers.
+  #
+  def test_timer_change_max_outstanding
+    defaults = EM.get_max_timers
+    EM.set_max_timers(100)
+
+    one_hundred_one_timers = proc { 101.times { EM.add_timer(5) {} } }
+
+    EM.run {
+      if EM.library_type == :pure_ruby
+        one_hundred_one_timers.call
+      elsif EM.library_type == :java
+        one_hundred_one_timers.call
+      else
+        begin
+          assert_raises( RuntimeError ) {
+            one_hundred_one_timers.call
+          }
+        rescue Object
+          p $!
+          assert(false, $!.message)
+        end
+      end
+      EM.stop
+    }
+
+    EM.set_max_timers( 101 )
+
+    EM.run {
+      one_hundred_one_timers.call
+      EM.stop
+    }
+  ensure
+    EM.set_max_timers(defaults)
+  end
+
+end

data/tests/test_ud.rb

@@ -0,0 +1,36 @@
+# $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 TestUserDefinedEvents < Test::Unit::TestCase
+
+  def test_a
+  end
+
+end