ln-xmpp4r 0.5

data/test/pubsub/tc_nodeconfig.rb

@@ -0,0 +1,61 @@
+#!/usr/bin/ruby
+
+$:.unshift File::dirname(__FILE__) + '/../../lib'
+
+require 'test/unit'
+require File::dirname(__FILE__) + '/../lib/clienttester'
+
+require 'xmpp4r'
+require 'xmpp4r/pubsub/children/node_config'
+require 'xmpp4r/dataforms'
+include Jabber
+
+# Jabber.debug = true
+
+class PubSub::NodeConfigTest < Test::Unit::TestCase
+  include ClientTester
+
+  def test_create()
+    config = PubSub::NodeConfig.new()
+    assert_nil(config.form)
+    assert_nil(config.node)
+    assert_equal({}, config.options)
+  end
+
+  def test_create_with_options
+    options = {'pubsub#access_model'=>'open'}
+
+    config = PubSub::NodeConfig.new(nil, options)
+    assert_kind_of(Jabber::Dataforms::XData, config.form)
+    assert_equal(options, config.options)
+    assert_equal(:submit, config.form.type)
+    assert_equal('http://jabber.org/protocol/pubsub#node_config', config.form.field('FORM_TYPE').values.first)
+  end
+
+  def test_create_with_options_and_node
+    node = 'mynode'
+    options = {'pubsub#access_model'=>'open'}
+
+    config = PubSub::NodeConfig.new(node, options)
+    assert_equal(node, config.node)
+    assert_kind_of(Jabber::Dataforms::XData, config.form)
+    assert_equal(options, config.options)
+    assert_equal(:submit, config.form.type)
+    assert_equal('http://jabber.org/protocol/pubsub#node_config', config.form.field('FORM_TYPE').values.first)
+  end
+
+  def test_set_options
+    options = {'pubsub#access_model'=>'open'}
+    config = PubSub::NodeConfig.new()
+    config.options = options
+    assert_kind_of(Jabber::Dataforms::XData, config.form)
+    assert_equal(options, config.options)
+  end
+
+  def test_create_with_array_in_options
+    options = {'pubsub#collection'=>['parent1','parent2']}
+    config = PubSub::OwnerNodeConfig.new(nil, options)
+
+    assert_equal(options, config.options)
+  end
+end

data/test/pubsub/tc_subscriptionconfig.rb

@@ -0,0 +1,41 @@
+#!/usr/bin/ruby
+
+$:.unshift File::dirname(__FILE__) + '/../../lib'
+
+require 'test/unit'
+require File::dirname(__FILE__) + '/../lib/clienttester'
+
+require 'xmpp4r'
+require 'xmpp4r/pubsub/children/subscription_config'
+require 'xmpp4r/dataforms'
+include Jabber
+
+# Jabber.debug = true
+
+class PubSub::SubscriptionConfigTest < Test::Unit::TestCase
+  include ClientTester
+
+  def test_create()
+    config = PubSub::SubscriptionConfig.new()
+    assert_nil(config.form)
+    assert_nil(config.node)
+    assert_equal({}, config.options)
+  end
+
+  def test_create_with_options
+    node = 'mynode'
+    jid = 'test@test.com'
+    options = {'pubsub#access_model'=>'open'}
+    subid = '004-yyy'
+
+    config = PubSub::SubscriptionConfig.new(node, jid, options, subid)
+    assert_equal(node, config.node)
+    assert_equal(subid, config.subid)
+    assert_kind_of(Jabber::JID, config.jid)
+    assert_equal(Jabber::JID.new(jid), config.jid)
+    assert_kind_of(Jabber::Dataforms::XData, config.form)
+    assert_equal(options, config.options)
+    assert_equal(:submit, config.form.type)
+    assert_equal('http://jabber.org/protocol/pubsub#subscribe_options', config.form.field('FORM_TYPE').values.first)
+  end
+end

data/test/reliable/tc_disconnect_cleanup.rb

@@ -0,0 +1,334 @@
+#!/usr/bin/ruby
+
+$:.unshift "#{File.dirname(__FILE__)}/../../lib"
+
+require 'test/unit'
+require 'xmpp4r'
+
+# Jabber::debug = true
+# Jabber::warnings = true
+
+class ClientDisconnectCleanupTest < Test::Unit::TestCase
+  class ControledClient < Jabber::Client
+    attr_accessor :tcpserver, :socket_override
+    def connect
+      begin
+        @port = 1024 + rand(32768 - 1024)
+        @tcpserver = TCPServer.new("127.0.0.1", @port)
+      rescue Errno::EADDRINUSE, Errno::EACCES
+        @tcpserver.close rescue nil
+        retry
+      end
+      super("127.0.0.1", @port)
+    end
+    def start
+      @socket = socket_override
+      super
+    end
+  end
+
+  def test_regular_stream_end
+    rd, wr = IO.pipe
+    rd.instance_eval do
+      def write(*args)
+      end
+      def flush
+      end
+    end
+    client = ControledClient.new("test@localhost")
+    @exceptions_caught = 0
+    client.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+    
+    Thread.new do
+      client.socket_override = rd
+      client.instance_eval{ @keepalive_interval = 0.1 }
+      client.connect
+      client.auth_nonsasl("password", false)
+    end
+    sleep(0.1)
+    assert client.is_connected?
+    
+    wr.write('<stream:stream xmlns:stream="http://etherx.jabber.org/streams">')
+    wr.write("</stream:stream>")
+    sleep(0.2)
+    
+    assert !client.is_connected?
+    assert client.instance_eval{ @parser_thread.nil? || !@parser_thread.alive? }
+    assert client.instance_eval{ @keepaliveThread.nil? || !@keepaliveThread.alive? }
+    assert @exceptions_caught > 0
+  end
+
+  def test_error_on_send
+    rd, wr = IO.pipe
+    rd.instance_eval do
+      def write(*args)
+      end
+      def flush
+      end
+    end
+    client = ControledClient.new("test@localhost")
+    @exceptions_caught = 0
+    client.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+    
+    Thread.new do
+      client.socket_override = rd
+      client.connect
+      client.auth_nonsasl("password", false)
+    end
+    sleep(0.1)
+    assert client.is_connected?
+
+    rd.instance_eval do
+      def write(*args)
+        raise "No writting for you, you disconnect now"
+      end
+    end
+    wr.write(%Q{<stream:stream from='localhost' id="acecf234be084aecdc16509077573c7d7200912f" version='1.0'  xmlns:stream="http://etherx.jabber.org/streams" xmlns="jabber:client"><stream:features><auth xmlns='http://jabber.org/features/iq-auth'/></stream:features> })
+    sleep(0.1)
+    
+    assert !client.is_connected?
+    assert client.instance_eval{ @parser_thread.nil? || !@parser_thread.alive? }
+    assert client.instance_eval{ @keepaliveThread.nil? || !@keepaliveThread.alive? }   
+    assert @exceptions_caught > 0
+  end
+
+  def test_client_disconnect
+    rd, wr = IO.pipe
+    rd.instance_eval do
+      def write(*args)
+      end
+      def flush
+      end
+    end
+    client = ControledClient.new("test@localhost")
+    @exceptions_caught = 0
+    client.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+    
+    Thread.new do
+      client.socket_override = rd
+      client.connect
+      client.auth_nonsasl("password", false)
+    end
+    wr.write(%Q{<stream:stream from='localhost' id="acecf234be084aecdc16509077573c7d7200912f" version='1.0'  xmlns:stream="http://etherx.jabber.org/streams" xmlns="jabber:client"><stream:features><auth xmlns='http://jabber.org/features/iq-auth'/></stream:features> })
+    sleep(0.1)
+    assert client.is_connected?
+    assert client.instance_eval{ @parser_thread.alive? }
+    assert client.instance_eval{ @keepaliveThread.alive? }
+
+    wr.close
+    sleep(0.1)
+    
+    assert !client.is_connected?
+    assert client.instance_eval{ @parser_thread.nil? || !@parser_thread.alive? }
+    assert client.instance_eval{ @keepaliveThread.nil? || !@keepaliveThread.alive? }   
+    assert @exceptions_caught > 0
+  end
+
+end
+
+class ConnectionDisconnectCleanupTest < Test::Unit::TestCase
+  class PipeConnection < Jabber::Connection
+    attr_accessor :socket_override
+    def connect
+      begin
+        @port = 1024 + rand(32768 - 1024)
+        tcpsocket = TCPServer.new("127.0.0.1", @port)
+      rescue Errno::EADDRINUSE, Errno::EACCES
+        tcpsocket.close rescue nil
+        retry
+      end
+      super("127.0.0.1", @port)
+    end
+    def accept_features
+      #do nothing
+    end
+    def start
+      @socket = self.socket_override
+      super
+    end
+  end
+  
+  def test_cleanup_when_disconnected_during_keepalive
+    rd, wr = IO.pipe
+    conn = PipeConnection.new
+    @exceptions_caught = 0
+    conn.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+    
+    Thread.new do
+      conn.socket_override = rd
+      #this will raise exception in keepalive thread, when attempts to send blank space shortly after connect
+      conn.instance_eval{ @keepalive_interval = 0.1 }
+      conn.connect
+    end
+    
+    sleep(0.2)
+    
+    assert rd.closed?
+    assert !conn.is_connected?
+    assert !conn.instance_eval{ @parser_thread }.alive?
+    assert !conn.instance_eval{ @keepaliveThread }.alive?
+    assert @exceptions_caught > 0
+  end
+  
+  def test_cleanup_after_stream_close
+    rd, wr = IO.pipe
+    conn = PipeConnection.new
+    @exceptions_caught = 0
+    conn.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+    
+    Thread.new do
+      conn.socket_override = rd
+      conn.connect
+    end
+    
+    wr.write("<hi/>")    
+    wr.close
+    sleep(0.1)
+    
+    assert rd.closed?
+    assert !conn.is_connected?
+    assert !conn.instance_eval{ @parser_thread }.alive?
+    assert !conn.instance_eval{ @keepaliveThread }.alive?
+    assert @exceptions_caught > 0
+  end
+  
+  def test_cleanup_after_stream_end
+    rd, wr = IO.pipe    
+    conn = PipeConnection.new
+    @exceptions_caught = 0
+    conn.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+    
+    Thread.new do
+      conn.socket_override = rd
+      conn.connect
+    end
+    
+    wr.write('<stream:stream xmlns:stream="http://etherx.jabber.org/streams">')
+    wr.write("</stream:stream>")
+    # wr.close
+    sleep(0.1)
+    
+    assert rd.closed?
+    assert !conn.is_connected?
+    assert !conn.instance_eval{ @parser_thread }.alive?
+    assert !conn.instance_eval{ @keepaliveThread }.alive?
+    assert @exceptions_caught > 0
+  end
+  
+end
+
+class StreamDisconnectCleanupTest < Test::Unit::TestCase
+  
+  def test_cleanup_when_errors_on_send
+    rd, wr = IO.pipe
+    
+    stream = Jabber::Stream.new
+    @exceptions_caught = 0
+    stream.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+    
+    assert !stream.is_connected?
+
+    stream.start(rd)
+    wr.write("<hi/>")
+    assert stream.is_connected?
+    #should raise error trying to write to stream that can't be written to, and catch should close it.
+    begin
+      stream.send("<hi/>")
+    rescue
+    end
+    sleep(0.1)
+    
+    assert rd.closed?
+    assert !stream.is_connected?
+    assert !stream.instance_eval{ @parser_thread }.alive?
+    assert @exceptions_caught > 0
+  end
+  
+  def test_cleanup_after_stream_close
+    rd, wr = IO.pipe
+    
+    stream = Jabber::Stream.new
+    @exceptions_caught = 0
+    stream.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+
+    assert !stream.is_connected?
+
+    stream.start(rd)
+    wr.write("<hi/>")
+    assert stream.is_connected?
+
+    wr.close
+    sleep(0.1)
+    
+    assert rd.closed?
+    assert !stream.is_connected?
+    assert !stream.instance_eval{ @parser_thread }.alive?
+    assert @exceptions_caught > 0
+  end
+
+  def test_cleanup_after_stream_end
+    rd, wr = IO.pipe
+    
+    stream = Jabber::Stream.new
+    @exceptions_caught = 0
+    stream.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+    
+    assert !stream.is_connected?
+
+    stream.start(rd)
+    wr.write('<stream:stream xmlns:stream="http://etherx.jabber.org/streams">')
+    wr.write("</stream:stream>")
+    assert stream.is_connected?
+
+    # wr.close
+    sleep(0.1)
+    
+    assert rd.closed?
+    assert !stream.is_connected?
+    assert !stream.instance_eval{ @parser_thread }.alive?
+    assert @exceptions_caught > 0
+  end
+
+  def test_cleanup_after_parse_failure
+    rd, wr = IO.pipe
+    
+    stream = Jabber::Stream.new
+    @exceptions_caught = 0
+    stream.on_exception do |e, connection, where_failed|
+      @exceptions_caught += 1
+    end
+    
+    assert !stream.is_connected?
+
+    stream.start(rd)
+    wr.write('<this is bad xml>>')
+    wr.close
+    assert stream.is_connected?
+
+    sleep(0.1)
+    
+    assert rd.closed?
+    assert !stream.is_connected?
+    assert !stream.instance_eval{ @parser_thread }.alive?
+    assert @exceptions_caught > 0
+  end
+  
+end

data/test/reliable/tc_disconnect_exception.rb

@@ -0,0 +1,37 @@
+#!/usr/bin/ruby
+
+$:.unshift "#{File.dirname(__FILE__)}/../../lib"
+
+require 'test/unit'
+require 'xmpp4r'
+
+class DisconnectExceptionTest < Test::Unit::TestCase
+  class Listener
+    def receive(element)
+    end
+  end
+  
+  def test_streamparser
+    rd, wr = IO.pipe
+    listener = Listener.new
+    exception_raised = nil
+    
+    Thread.new do
+      begin
+        parser = Jabber::StreamParser.new(rd, listener)
+        parser.parse
+      rescue => e
+        exception_raised = e
+      end
+    end
+    
+    wr.write("<hi/>")
+    wr.close
+    sleep(0.1)
+    
+    assert exception_raised
+    assert exception_raised.is_a?(Jabber::ServerDisconnected), "Expected a Jabber::ServerDisconnected but got #{exception_raised}"
+    # puts exception_raised.inspect
+  end
+  
+end