eventmachine 1.2.0.dev.2-x64-mingw32

data/tests/test_ssl_verify.rb

@@ -0,0 +1,126 @@
+require 'em_test_helper'
+
+class TestSslVerify < Test::Unit::TestCase
+  def setup
+    $dir = File.dirname(File.expand_path(__FILE__)) + '/'
+    $cert_from_file = File.read($dir+'client.crt')
+  end
+
+  module ClientNoCert
+    def connection_completed
+      start_tls()
+    end
+
+    def ssl_handshake_completed
+      $client_handshake_completed = true
+      close_connection
+    end
+
+    def unbind
+      EM.stop_event_loop
+    end
+  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
+
+  module FailServerNoPeerCert
+    def post_init
+      start_tls(:verify_peer => true, :fail_if_no_peer_cert => true)
+    end
+
+    def ssl_verify_peer(cert)
+      raise "Verify peer should not get called for a client without a certificate"
+    end
+
+    def ssl_handshake_completed
+      $server_handshake_completed = true
+    end
+  end
+
+  def test_fail_no_peer_cert
+    omit_unless(EM.ssl?)
+    omit_if(rbx?)
+
+    $client_handshake_completed, $server_handshake_completed = false, false
+
+    EM.run {
+      EM.start_server("127.0.0.1", 16784, FailServerNoPeerCert)
+      EM.connect("127.0.0.1", 16784, ClientNoCert)
+    }
+
+    assert(!$client_handshake_completed)
+    assert(!$server_handshake_completed)
+  end
+
+  def test_accept_server
+    omit_unless(EM.ssl?)
+    omit_if(rbx?)
+    $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
+    omit_unless(EM.ssl?)
+    omit_if(rbx?)
+    $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_stomp.rb

@@ -0,0 +1,37 @@
+require 'em_test_helper'
+
+class TestStomp < Test::Unit::TestCase
+  CONTENT_LENGTH_REGEX = /^content-length: (\d+)$/
+
+  def bytesize(str)
+    str = str.to_s
+    size = str.bytesize if str.respond_to?(:bytesize) # bytesize added in 1.9
+    size || str.size
+  end
+
+  def test_content_length_in_bytes
+    connection = Object.new
+    connection.instance_eval do 
+      extend EM::P::Stomp
+
+      def last_sent_content_length
+        @sent && Integer(@sent[CONTENT_LENGTH_REGEX, 1])
+      end
+
+      def send_data(string)
+        @sent = string
+      end
+    end
+
+    queue = "queue"
+    failure_message = "header content-length is not the byte size of last sent body"
+
+    body = "test"
+    connection.send queue, body
+    assert_equal bytesize(body), connection.last_sent_content_length, failure_message
+
+    body = "test\u221A"
+    connection.send queue, body
+    assert_equal bytesize(body), connection.last_sent_content_length, failure_message
+  end
+end

data/tests/test_system.rb

@@ -0,0 +1,46 @@
+# coding: utf-8
+require 'em_test_helper'
+
+class TestSystem < Test::Unit::TestCase
+  def setup
+    @filename = File.expand_path("../я манал dump.txt", __FILE__)
+    @test_data = 'a' * 100
+    File.open(@filename, 'w'){|f| f.write(@test_data)}
+  end
+
+  def test_system
+    omit_if(windows?)
+
+    result = nil
+    status = nil
+    EM.run {
+      EM.system('cat', @filename){|out, state|
+        result = out
+        status = state.exitstatus
+        EM.stop
+      }
+    }
+    assert_equal(0, status)
+    assert_equal(@test_data, result)
+  end
+
+  def test_system_with_string
+    omit_if(windows?)
+
+    result = nil
+    status = nil
+    EM.run {
+      EM.system("cat '#@filename'"){|out, state|
+        result = out
+        status = state.exitstatus
+        EM.stop
+      }
+    }
+    assert_equal(0, status)
+    assert_equal(@test_data, result)
+  end
+
+  def teardown
+    File.unlink(@filename)
+  end
+end

data/tests/test_threaded_resource.rb

@@ -0,0 +1,61 @@
+class TestThreadedResource < Test::Unit::TestCase
+  def object
+    @object ||= {}
+  end
+
+  def resource
+    @resource = EM::ThreadedResource.new do
+      object
+    end
+  end
+
+  def teardown
+    resource.shutdown
+  end
+
+  def test_dispatch_completion
+    EM.run do
+      EM.add_timer(3) do
+        EM.stop
+        fail 'Resource dispatch timed out'
+      end
+      completion = resource.dispatch do |o|
+        o[:foo] = :bar
+        :foo
+      end
+      completion.callback do |result|
+        assert_equal :foo, result
+        EM.stop
+      end
+      completion.errback do |error|
+        EM.stop
+        fail "Unexpected error: #{error.message}"
+      end
+    end
+    assert_equal :bar, object[:foo]
+  end
+
+  def test_dispatch_failure
+    completion = resource.dispatch do |o|
+      raise 'boom'
+    end
+    completion.errback do |error|
+      assert_kind_of RuntimeError, error
+      assert_equal 'boom', error.message
+    end
+  end
+
+  def test_dispatch_threading
+    main = Thread.current
+    resource.dispatch do |o|
+      o[:dispatch_thread] = Thread.current
+    end
+    assert_not_equal main, object[:dispatch_thread]
+  end
+
+  def test_shutdown
+    # This test should get improved sometime. The method returning thread is
+    # NOT an api that will be maintained.
+    assert !resource.shutdown.alive?
+  end
+end

data/tests/test_tick_loop.rb

@@ -0,0 +1,59 @@
+require "test/unit"
+require 'em_test_helper'
+
+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,123 @@
+require 'em_test_helper'
+
+class TestTimers < Test::Unit::TestCase
+
+  def test_timer_with_block
+    x = false
+    EM.run {
+      EM::Timer.new(0) {
+        x = true
+        EM.stop
+      }
+    }
+    assert x
+  end
+
+  def test_timer_with_proc
+    x = false
+    EM.run {
+      EM::Timer.new(0, proc {
+        x = true
+        EM.stop
+      })
+    }
+    assert x
+  end
+
+  def test_timer_cancel
+    assert_nothing_raised do
+      EM.run {
+        timer = EM::Timer.new(0.01) { flunk "Timer was not cancelled." }
+        timer.cancel
+
+        EM.add_timer(0.02) { EM.stop }
+      }
+    end
+  end
+
+  def test_periodic_timer
+    x = 0
+    EM.run {
+      EM::PeriodicTimer.new(0.01) do
+        x += 1
+        EM.stop if x == 4
+      end
+    }
+
+    assert_equal 4, x
+  end
+
+  def test_add_periodic_timer
+    x = 0
+    EM.run {
+      t = EM.add_periodic_timer(0.01) do
+        x += 1
+        EM.stop  if x == 4
+      end
+      assert t.respond_to?(:cancel)
+    }
+    assert_equal 4, x
+  end
+
+  def test_periodic_timer_cancel
+    x = 0
+    EM.run {
+      pt = EM::PeriodicTimer.new(0.01) { x += 1 }
+      pt.cancel
+      EM::Timer.new(0.02) { EM.stop }
+    }
+    assert_equal 0, x
+  end
+
+  def test_add_periodic_timer_cancel
+    x = 0
+    EM.run {
+      pt = EM.add_periodic_timer(0.01) { x += 1 }
+      EM.cancel_timer(pt)
+      EM.add_timer(0.02) { EM.stop }
+    }
+    assert_equal 0, x
+  end
+
+  def test_periodic_timer_self_cancel
+    x = 0
+    EM.run {
+      pt = EM::PeriodicTimer.new(0) {
+        x += 1
+        if x == 4
+          pt.cancel
+          EM.stop
+        end
+      }
+    }
+    assert_equal 4, x
+  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.
+  unless [:pure_ruby, :java].include? EM.library_type
+    def test_timer_change_max_outstanding
+      defaults = EM.get_max_timers
+      EM.set_max_timers(100)
+
+      one_hundred_one_timers = lambda do
+        101.times { EM.add_timer(0.01) {} }
+        EM.stop
+      end
+
+      assert_raises(RuntimeError) do
+        EM.run( &one_hundred_one_timers )
+      end
+
+      EM.set_max_timers( 101 )
+
+      assert_nothing_raised do
+        EM.run( &one_hundred_one_timers )
+      end
+    ensure
+      EM.set_max_timers(defaults)
+    end
+  end
+
+end