libc-eventmachine 0.12.5.42

data/tests/test_sasl.rb

@@ -0,0 +1,74 @@
+# $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,243 @@
+# $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,80 @@
+# $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
+
+

data/tests/test_smtpclient.rb

@@ -0,0 +1,83 @@
+# $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 TestSmtpClient < Test::Unit::TestCase
+
+  Localhost = "127.0.0.1"
+  Localport = 9801
+
+  def setup
+  end
+
+  def teardown
+  end
+
+  def test_a
+    # No real tests until we have a server implementation to test against.
+    # This is what the call looks like, though:
+    err = nil
+    EM.run {
+      d = EM::Protocols::SmtpClient.send :domain=>"example.com",
+      :host=>Localhost,
+      :port=>Localport, # optional, defaults 25
+      :starttls=>true,
+      :from=>"sender@example.com",
+      :to=> ["to_1@example.com", "to_2@example.com"],
+      :header=> {"Subject" => "This is a subject line"},
+      :body=> "This is the body of the email",
+      :verbose=>true
+      d.errback {|e|
+        err = e
+        EM.stop
+      }
+    }
+    assert(err)
+  end
+
+  def test_content
+    err = nil
+    EM.run {
+      d = EM::Protocols::SmtpClient.send :domain=>"example.com",
+      :host=>Localhost,
+      :port=>Localport, # optional, defaults 25
+      :starttls=>true,
+      :from=>"sender@example.com",
+      :to=> ["to_1@example.com", "to_2@example.com"],
+      :content => ["Subject: xxx\r\n\r\ndata\r\n.\r\n"],
+      :verbose=>true
+      d.errback {|e|
+        err = e
+        EM.stop
+      }
+    }
+    assert(err)
+  end
+
+end

data/tests/test_smtpserver.rb

@@ -0,0 +1,93 @@
+# $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 TestSmtpServer < Test::Unit::TestCase
+
+	# Don't test on port 25. It requires superuser and there's probably
+	# a mail server already running there anyway.
+	Localhost = "127.0.0.1"
+	Localport = 25001
+
+	# This class is an example of what you need to write in order
+	# to implement a mail server. You override the methods you are
+	# interested in. Some, but not all, of these are illustrated here.
+	#
+	class Mailserver < EM::Protocols::SmtpServer
+
+		attr_reader :my_msg_body, :my_sender, :my_recipients
+
+		def initialize *args
+			super
+		end
+		def receive_sender sender
+			@my_sender = sender
+			#p sender
+			true
+		end
+		def receive_recipient rcpt
+			@my_recipients ||= []
+			@my_recipients << rcpt
+			true
+		end
+		def receive_data_chunk c
+			@my_msg_body = c.last
+		end
+		def connection_ended
+			EM.stop
+		end
+	end
+
+
+	def setup
+	end
+
+	def teardown
+	end
+
+	def test_mail
+		c = nil
+		EM.run {
+			EM.start_server( Localhost, Localport, Mailserver ) {|conn| c = conn}
+			EM::Timer.new(2) {EM.stop} # prevent hanging the test suite in case of error
+			EM::Protocols::SmtpClient.send :host=>Localhost,
+				:port=>Localport,
+				:domain=>"bogus",
+				:from=>"me@example.com",
+				:to=>"you@example.com",
+				:header=> {"Subject"=>"Email subject line", "Reply-to"=>"me@example.com"},
+				:body=>"Not much of interest here."
+
+		}
+		assert_equal( "Not much of interest here.", c.my_msg_body )
+		assert_equal( "<me@example.com>", c.my_sender )
+		assert_equal( ["<you@example.com>"], c.my_recipients )
+	end
+end
+