MattHulse-eventmachine 0.0.1

data/tests/test_queue.rb

@@ -0,0 +1,44 @@
+$:.unshift "../lib"
+require 'eventmachine'
+require 'test/unit'
+
+class TestEventMachineQueue < Test::Unit::TestCase
+  def test_queue_push
+    s = 0
+    EM.run do
+      q = EM::Queue.new
+      q.push(1)
+      EM.next_tick { s = q.size; EM.stop }
+    end
+    assert_equal 1, s
+  end
+
+  def test_queue_pop
+    x,y,z = nil
+    EM.run do
+      q = EM::Queue.new
+      q.push(1,2,3)
+      q.pop { |v| x = v }
+      q.pop { |v| y = v }
+      q.pop { |v| z = v; EM.stop }
+    end
+    assert_equal 1, x
+    assert_equal 2, y
+    assert_equal 3, z
+  end
+
+  def test_queue_reactor_thread
+    q = EM::Queue.new
+
+    Thread.new { q.push(1,2,3) }.join
+    assert q.empty?
+    EM.run { EM.next_tick { EM.stop } }
+    assert_equal 3, q.size
+
+    x = nil
+    Thread.new { q.pop { |v| x = v } }.join
+    assert_equal nil, x
+    EM.run { EM.next_tick { EM.stop } }
+    assert_equal 1, x
+  end
+end

data/tests/test_running.rb

@@ -0,0 +1,42 @@
+# $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 TestRunning < Test::Unit::TestCase
+  def test_running
+    assert_equal( false, EM::reactor_running? )
+    r = false
+    EM.run {
+      r = EM::reactor_running?
+      EM.stop
+    }
+    assert_equal( true, r )
+  end
+end
+

data/tests/test_sasl.rb

@@ -0,0 +1,72 @@
+# $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 TestSASL < Test::Unit::TestCase
+
+  # SASL authentication is usually done with UNIX-domain sockets, but
+  # we'll use TCP so this test will work on Windows. As far as the
+  # protocol handlers are concerned, there's no difference.
+
+  Host,Port = "127.0.0.1",9560
+  TestUser,TestPsw = "someone", "password"
+
+  class SaslServer < EM::Connection
+    include EM::Protocols::SASLauth
+    def validate usr, psw, sys, realm
+      usr == TestUser and psw == TestPsw
+    end
+  end
+
+  class SaslClient < EM::Connection
+    include EM::Protocols::SASLauthclient
+  end
+
+  def test_sasl
+    resp = nil
+    EM.run {
+      EM.start_server( Host, Port, SaslServer )
+
+      c = EM.connect( Host, Port, SaslClient )
+      d = c.validate?( TestUser, TestPsw )
+      d.timeout 2
+      d.callback {
+        resp = true
+        EM.stop
+      }
+      d.errback {
+        resp = false
+        EM.stop
+      }
+    }
+    assert_equal( true, resp )
+  end
+
+end

data/tests/test_send_file.rb

@@ -0,0 +1,242 @@
+# $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 'socket'
+require 'test/unit'
+
+class TestSendFile < Test::Unit::TestCase
+
+  module TestModule
+    def post_init
+      send_file_data TestFilename
+      close_connection_after_writing
+    end
+  end
+
+  module TestClient
+    def data_to(&blk)
+      @data_to = blk
+    end
+    
+    def receive_data(data)
+      @data_to.call(data) if @data_to
+    end
+    
+    def unbind
+      EM.stop
+    end
+  end
+
+  TestHost = "0.0.0.0"
+  TestPort = 9055
+  TestFilename = "./xxxxxx"
+
+  def setup
+  end
+
+  def teardown
+    File.unlink( TestFilename ) if File.exist?( TestFilename )
+  end
+
+  def test_send_file
+    File.open( TestFilename, "w" ) {|f|
+      f << ("A" * 5000)
+    }
+
+    data = ''
+
+    EM.run {
+      EM.start_server TestHost, TestPort, TestModule
+      EM.add_timer(2) {EM.stop} # avoid hanging in case of error
+
+      EM.connect TestHost, TestPort, TestClient do |c|
+        c.data_to { |d| data << d }
+      end
+    }
+
+    assert_equal( "A" * 5000, data )
+    File.unlink TestFilename
+  end
+
+  # EventMachine::Connection#send_file_data has a strict upper limit on the filesize it will work with.
+  def test_send_large_file
+    File.open( TestFilename, "w" ) {|f|
+      f << ("A" * 1000000)
+    }
+
+    data = ''
+
+    ex_class = RUBY_PLATFORM == 'java' ? NativeException : RuntimeError
+    assert_raises( ex_class ) {
+      EM.run {
+        EM.start_server TestHost, TestPort, TestModule
+        EM.add_timer(2) {EM.stop} # avoid hanging in case of error
+        EM.connect TestHost, TestPort, TestClient do |c|
+          c.data_to { |d| data << d }
+        end
+      }
+    }
+
+    File.unlink TestFilename
+  end
+
+
+  module StreamTestModule
+    def post_init
+      EM::Deferrable.future( stream_file_data(TestFilename)) {
+        close_connection_after_writing
+      }
+    end
+  end
+
+  module ChunkStreamTestModule
+    def post_init
+      EM::Deferrable.future( stream_file_data(TestFilename, :http_chunks=>true)) {
+        close_connection_after_writing
+      }
+    end
+  end
+
+  def test_stream_file_data
+    File.open( TestFilename, "w" ) {|f|
+      f << ("A" * 1000)
+    }
+
+    data = ''
+
+    EM.run {
+      EM.start_server TestHost, TestPort, StreamTestModule
+      EM.add_timer(2) {EM.stop} # avoid hanging in case of error
+      EM.connect TestHost, TestPort, TestClient do |c|
+        c.data_to { |d| data << d }
+      end
+    }
+
+    assert_equal( "A" * 1000, data )
+
+    File.unlink TestFilename
+  end
+
+  def test_stream_chunked_file_data
+    File.open( TestFilename, "w" ) {|f|
+      f << ("A" * 1000)
+    }
+
+    data = ''
+
+    EM.run {
+      EM.start_server TestHost, TestPort, ChunkStreamTestModule
+      EM.add_timer(2) {EM.stop} # avoid hanging in case of error
+      EM.connect TestHost, TestPort, TestClient do |c|
+        c.data_to { |d| data << d }
+      end
+    }
+
+    assert_equal( "3e8\r\n#{"A" * 1000}\r\n0\r\n\r\n", data )
+
+    File.unlink TestFilename
+  end
+
+  module BadFileTestModule
+    def post_init
+      de = stream_file_data( TestFilename+"..." )
+      de.errback {|msg|
+        send_data msg
+        close_connection_after_writing
+      }
+    end
+  end
+  def test_stream_bad_file
+    data = ''
+    EM.run {
+      EM.start_server TestHost, TestPort, BadFileTestModule
+      EM.add_timer(2) {EM.stop} # avoid hanging in case of error
+      EM.connect TestHost, TestPort, TestClient do |c|
+        c.data_to { |d| data << d }
+      end
+    }
+
+    assert_equal( "file not found", data )
+  end
+
+  def test_stream_large_file_data
+    begin
+      require 'fastfilereaderext'
+    rescue LoadError
+      return
+    end
+    File.open( TestFilename, "w" ) {|f|
+      f << ("A" * 10000)
+    }
+
+    data = ''
+
+    EM.run {
+      EM.start_server TestHost, TestPort, StreamTestModule
+      EM.add_timer(2) {EM.stop} # avoid hanging in case of error
+      EM.connect TestHost, TestPort, TestClient do |c|
+        c.data_to { |d| data << d }
+      end
+    }
+
+    assert_equal( "A" * 10000, data )
+
+    File.unlink TestFilename
+  end
+
+  def test_stream_large_chunked_file_data
+    begin
+      require 'fastfilereaderext'
+    rescue LoadError
+      return
+    end
+    File.open( TestFilename, "w" ) {|f|
+      f << ("A" * 100000)
+    }
+
+    data = ''
+
+    EM.run {
+      EM.start_server TestHost, TestPort, ChunkStreamTestModule
+      EM.add_timer(2) {EM.stop} # avoid hanging in case of error
+      EM.connect TestHost, TestPort, TestClient do |c|
+        c.data_to { |d| data << d }
+      end
+    }
+
+    expected = [
+      "4000\r\n#{"A" * 16384}\r\n" * 6,
+      "6a0\r\n#{"A" * 0x6a0}\r\n",
+      "0\r\n\r\n"
+    ].join
+    assert_equal( expected, data )
+
+    File.unlink TestFilename
+  end
+
+end

data/tests/test_servers.rb

@@ -0,0 +1,76 @@
+# $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 'socket'
+require 'test/unit'
+
+class TestServers < Test::Unit::TestCase
+
+  Host = "127.0.0.1"
+  Port = 9555
+
+  module NetstatHelper
+    GlobalUdp4Rexp = /udp.*\s+(?:\*|(?:0\.){3}0)[:.](\d+)\s/i
+    GlobalTcp4Rexp = /tcp.*\s+(?:\*|(?:0\.){3}0)[:.](\d+)\s/i
+    LocalUdpRexp = /udp.*\s+(?:127\.0\.0\.1|::1)[:.](\d+)\s/i
+    LocalTcpRexp = /tcp.*\s+(?:127\.0\.0\.1|::1)[:.](\d+)\s/i
+    def grep_netstat(pattern)
+      `netstat -an`.scan(/^.*$/).grep(pattern)
+    end
+  end
+  include NetstatHelper
+
+  class TestStopServer < EM::Connection
+    def initialize *args
+      super
+    end
+    def post_init
+      # TODO,sucks that this isn't OOPy enough.
+      EM.stop_server @server_instance
+    end
+  end
+
+  def run_test_stop_server
+    EM.run {
+      sig = EM.start_server(Host, Port)
+      assert(grep_netstat(LocalTcpRexp).grep(%r(#{Port})).size >= 1, "Server didn't start")
+      EM.stop_server sig
+      # Give the server some time to shutdown.
+      EM.add_timer(0.1) {
+        assert(grep_netstat(LocalTcpRexp).grep(%r(#{Port})).empty?, "Servers didn't stop")
+        EM.stop
+      }
+    }
+  end
+  def test_stop_server
+    assert(grep_netstat(LocalTcpRexp).grep(Port).empty?, "Port already in use")
+    5.times {run_test_stop_server}
+    assert(grep_netstat(LocalTcpRexp).grep(%r(#{Port})).empty?, "Servers didn't stop")
+  end
+
+end