gmcmillan-xmpp4r 0.5

data/test/bytestreams/tc_ibb.rb

@@ -0,0 +1,188 @@
+#!/usr/bin/env ruby
+
+$:.unshift File::dirname(__FILE__) + '/../../lib'
+
+require 'test/unit'
+require File::dirname(__FILE__) + '/../lib/clienttester'
+
+require 'xmpp4r'
+require 'xmpp4r/bytestreams'
+require 'xmpp4r/semaphore'
+include Jabber
+
+class IBBTest < Test::Unit::TestCase
+  include ClientTester
+
+  def create_buffer(size)
+    ([nil] * size).collect { rand(256).chr }.join
+  end
+
+  def test_ibb_target2initiator
+    target = Bytestreams::IBBTarget.new(@server, '1', nil, '1@a.com/1')
+    initiator = Bytestreams::IBBInitiator.new(@client, '1', nil, '1@a.com/1')
+
+    buffer = create_buffer(9999)
+
+    Thread.new do
+      target.accept
+      target.write(buffer)
+      Thread.pass
+      target.close
+    end
+
+    target.accept_wait
+
+    initiator.open
+
+    received = ''
+    while buf = initiator.read
+      received += buf
+    end
+
+    initiator.close
+
+    assert_equal(buffer, received)
+  end
+
+  def test_ibb_initiator2target
+    target = Bytestreams::IBBTarget.new(@server, '1', nil, '1@a.com/1')
+    initiator = Bytestreams::IBBInitiator.new(@client, '1', nil, '1@a.com/1')
+
+    buffer = create_buffer(9999)
+
+    Thread.new do
+      target.accept_wait
+      initiator.open
+      initiator.write(buffer)
+      initiator.close
+    end
+
+
+    target.accept
+
+    received = ''
+    while buf = target.read
+      received += buf
+    end
+
+    target.close
+
+    assert_equal(buffer, received)
+  end
+
+  def test_ibb_pingpong
+    ignored_stanzas = 0
+    wait = Semaphore.new
+    @server.add_message_callback { ignored_stanzas += 1; wait.run }
+    @server.add_iq_callback { ignored_stanzas += 1; wait.run }
+
+
+    target = Bytestreams::IBBTarget.new(@server, '1', nil, '1@a.com/1')
+    initiator = Bytestreams::IBBInitiator.new(@client, '1', nil, '1@a.com/1')
+
+    Thread.new do
+      target.accept
+
+      while buf = target.read
+        target.write(buf)
+        target.flush
+      end
+
+      target.close
+    end
+
+
+    assert_equal(0, ignored_stanzas)
+    @client.send("<iq from='1@a.com/1' type='set'>
+                    <close xmlns='http://jabber.org/protocol/ibb' sid='another session id'/>
+                  </iq>")
+    wait.wait
+    assert_equal(1, ignored_stanzas)
+
+
+    target.accept_wait
+    initiator.open
+
+
+    assert_equal(1, ignored_stanzas)
+    @client.send("<message from='1@a.com/1' type='error'>
+                    <data xmlns='http://jabber.org/protocol/ibb' sid='another session id' seq='0'/>
+                  </message>")
+    wait.wait
+    assert_equal(2, ignored_stanzas)
+    @client.send("<iq from='1@a.com/1' type='set'>
+                    <close xmlns='http://jabber.org/protocol/ibb' sid='another session id'/>
+                  </iq>")
+    wait.wait
+    assert_equal(3, ignored_stanzas)
+
+
+    5.times do
+      buf = create_buffer(100)
+      initiator.write(buf)
+      initiator.flush
+
+      bufr = ''
+      begin
+        bufr += initiator.read
+      end while bufr.size < buf.size
+      assert_equal(buf, bufr)
+    end
+
+    initiator.close
+
+
+    assert_equal(3, ignored_stanzas)
+  end
+
+  def test_ibb_error
+    target = Bytestreams::IBBTarget.new(@server, '1', nil, '1@a.com/1')
+    initiator = Bytestreams::IBBInitiator.new(@client, '1', nil, '1@a.com/1')
+
+    Thread.new do
+      target.accept
+
+      @server.send("<message from='1@a.com/1' type='error'>
+                      <data xmlns='http://jabber.org/protocol/ibb' sid='#{target.instance_variable_get(:@session_id)}' seq='0'/>
+                    </message>")
+    end
+
+
+    target.accept_wait
+    initiator.open
+
+    assert_nil(initiator.read)
+
+    initiator.close
+  end
+
+  def test_ibb_inactive
+    target = Bytestreams::IBBTarget.new(@server, '1', nil, '1@a.com/1')
+    initiator = Bytestreams::IBBInitiator.new(@client, '1', nil, '1@a.com/1')
+
+    assert_nil(target.read)
+    assert_nil(initiator.read)
+
+    assert_raise(RuntimeError) {
+      target.write('a' * target.block_size)
+    }
+    assert_raise(RuntimeError) {
+      initiator.write('a' * initiator.block_size)
+    }
+  end
+
+  def test_ibb_queueitem
+    i1 = Bytestreams::IBBQueueItem.new(:close)
+    assert_equal(:close, i1.type)
+    assert_nil(i1.seq)
+
+    i2 = Bytestreams::IBBQueueItem.new(:data, 1, ['blah'].pack('m'))
+    assert_equal(:data, i2.type)
+    assert_equal(1, i2.seq)
+    assert_equal('blah', i2.data)
+
+    assert_raise(RuntimeError) {
+      i3 = Bytestreams::IBBQueueItem.new(:invalid)
+    }
+  end
+end

data/test/bytestreams/tc_socks5bytestreams.rb

@@ -0,0 +1,114 @@
+#!/usr/bin/env ruby
+
+$:.unshift File::dirname(__FILE__) + '/../../lib'
+
+require 'test/unit'
+require File::dirname(__FILE__) + '/../lib/clienttester'
+
+require 'xmpp4r'
+require 'xmpp4r/bytestreams'
+include Jabber
+
+class SOCKS5BytestreamsTest < Test::Unit::TestCase
+  include ClientTester
+
+  @@server = Bytestreams::SOCKS5BytestreamsServer.new(65005)
+  @@server.add_address('localhost')
+
+  def create_buffer(size)
+    ([nil] * size).collect { rand(256).chr }.join
+  end
+
+  def test_server2multi
+    target1 = Bytestreams::SOCKS5BytestreamsTarget.new(@server, '1', '1@a.com/1', '1@a.com/2')
+    target2 = Bytestreams::SOCKS5BytestreamsTarget.new(@server, '2', '2@a.com/1', '2@a.com/2')
+    initiator1 = Bytestreams::SOCKS5BytestreamsInitiator.new(@client, '1', '1@a.com/1', '1@a.com/2')
+    initiator2 = Bytestreams::SOCKS5BytestreamsInitiator.new(@client, '2', '2@a.com/1', '2@a.com/2')
+    initiator1.add_streamhost(@@server)
+    initiator2.add_streamhost(@@server)
+
+    buf1 = create_buffer(8192)
+    buf2 = create_buffer(8192)
+
+    Thread.new do
+      target1.accept
+      target1.write(buf1)
+      target1.flush
+      target1.close
+    end
+
+    Thread.new do
+      target2.accept
+      target2.write(buf2)
+      target2.flush
+      target2.close
+    end
+
+    target1.accept_wait
+    target2.accept_wait
+    initiator1.open
+    initiator2.open
+
+    recv1 = ''
+    recv2 = ''
+
+    while buf = initiator2.read(256)
+      recv2 += buf
+    end
+
+    while buf = initiator1.read(256)
+      recv1 += buf
+    end
+
+    initiator1.close
+    initiator2.close
+
+    assert_equal(buf1, recv1)
+    assert_equal(buf2, recv2)
+  end
+
+  def test_pingpong
+    target = Bytestreams::SOCKS5BytestreamsTarget.new(@server, '1', '1@a.com/1', '1@a.com/2')
+    initiator = Bytestreams::SOCKS5BytestreamsInitiator.new(@client, '1', '1@a.com/1', '1@a.com/2')
+    initiator.add_streamhost(@@server)
+
+    Thread.new do
+      target.accept
+
+      while buf = target.read(256)
+        target.write(buf)
+        target.flush
+      end
+
+      target.close
+    end
+    target.accept_wait
+
+    initiator.open
+
+    10.times do
+      buf = create_buffer(8192)
+      initiator.write(buf)
+      initiator.flush
+
+      bufr = ''
+      begin
+        bufr += initiator.read(256)
+      end while bufr.size < buf.size
+      assert_equal(buf, bufr)
+    end
+
+    initiator.close
+  end
+
+  def test_timeout
+    target = Bytestreams::SOCKS5BytestreamsTarget.new(@server, '1', '1@a.com/1', '1@a.com/2')
+
+    assert_nothing_raised do
+      Timeout::timeout(1) do
+        target.connect_timeout = 0.5
+        assert target.accept == false
+      end
+    end
+  end
+end

data/test/caps/tc_helper.rb

@@ -0,0 +1,158 @@
+# coding: utf-8
+
+$:.unshift File::dirname(__FILE__) + '/../../lib'
+
+require 'test/unit'
+require File::dirname(__FILE__) + '/../lib/clienttester'
+
+require 'xmpp4r/discovery'
+require 'xmpp4r/caps/helper/helper'
+require 'xmpp4r/dataforms'
+include Jabber
+
+class Caps::HelperTest < Test::Unit::TestCase
+  include ClientTester
+
+  ##
+  # Walk through the client/ server conversation defined
+  # in http://www.xmpp.org/extensions/xep-0115.html#usecases
+  # and assert conformance.
+  def test_caps_reply
+
+    # This will be invoked by 'wait_state' below...
+    state { |presence|
+      assert_kind_of(Jabber::Presence, presence)
+      c = presence.first_element('c')
+      assert_kind_of(Jabber::Caps::C, c)
+
+      # see http://www.xmpp.org/extensions/xep-0115.html#ver
+      assert_equal('SrFo9ar2CCk2EnOH4q4QANeuxLQ=', c.ver)
+
+      # version 1.5 of xep 0115 indicates that the <c /> stanzq MUST feature a 'hash' attribute
+      assert_equal('sha-1', c.hash)
+
+      assert_equal("http://home.gna.org/xmpp4r/##{Jabber::XMPP4R_VERSION}", c.node)
+    }
+
+    # Construct Caps::Helper which will send a <presence>
+    # stanza (with embedded <c/> advert) to the 'server'
+    h = Caps::Helper.new(@client, identities, features)
+
+    # The 'server' will receive the <presence> stanza and
+    # yield it to the 'state' block above, where an <iq> query
+    # will be sent back to the 'client,' to discover its capabilities.
+    # Wait here until the block has been executed.
+    wait_state
+
+    # The Caps::Helper will process the <iq> query from the 'server'
+    # and reply with an <iq> result providing the details of its
+    # identities and features.
+
+    state { |iq|
+      assert_kind_of(Jabber::Iq, iq)
+      assert_equal(:result, iq.type)
+      assert_equal('1867999907', iq.id)
+      assert_equal('new-big-computer.local', iq.to.to_s)
+
+      assert_kind_of(Jabber::Discovery::IqQueryDiscoInfo, iq.query)
+      assert_equal(iq.query.identities.size, identities.size)
+      get_category_iname_type = lambda{ |i| [i.category,i.iname,i.type] }
+      assert_equal(iq.query.identities.map(&get_category_iname_type).sort!, identities.map(&get_category_iname_type).sort!)
+
+      assert_equal(3, iq.query.features.size)
+      get_var = lambda { |f| f.var }
+      assert_equal(iq.query.features.sort, features.map(&get_var).sort)
+    }
+
+    send(iq_discovering_capabilities)
+
+    # The 'server' will receive the <iq> result from the
+    # 'client' and yield it to the block above. Wait here
+    # until that block exits.
+    wait_state
+  end
+
+  def test_custom_node
+    client_id='http://new-big-computer.local/client#321'
+
+    state { |presence|
+      c = presence.first_element('c')
+      assert_kind_of(Jabber::Caps::C, c)
+      assert_equal(client_id, c.node)
+      assert_equal('SrFo9ar2CCk2EnOH4q4QANeuxLQ=', c.ver)
+    }
+    h = Caps::Helper.new(@client, identities, features, client_id)
+    wait_state
+  end
+
+  def identities
+    [Jabber::Discovery::Identity.new('client', 'Exodus 0.9.1', 'pc')]
+  end
+
+  def features
+    [Jabber::Discovery::Feature.new("http://jabber.org/protocol/disco#info"),
+     Jabber::Discovery::Feature.new("http://jabber.org/protocol/disco#items"),
+     Jabber::Discovery::Feature.new("http://jabber.org/protocol/muc")]
+  end
+
+  def iq_discovering_capabilities
+    "<iq from='new-big-computer.local' type='get' to='matt@new-big-computer.local/capable_client' id='1867999907' xmlns='jabber:client'><query xmlns='http://jabber.org/protocol/disco#info'/></iq>"
+  end
+
+  ##
+  # http://www.xmpp.org/extensions/xep-0115.html#ver-gen-complex
+  def test_caps_complex
+    form = Dataforms::XData.new(:result)
+    form.add(Dataforms::XDataField.new('FORM_TYPE', :hidden)).value = 'urn:xmpp:dataforms:softwareinfo'
+    form.add(Dataforms::XDataField.new('ip_version')).values = ['ipv4', 'ipv6']
+    form.add(Dataforms::XDataField.new('software')).value = 'Psi' # re-ordered
+    form.add(Dataforms::XDataField.new('software_version')).value = '0.11'
+    form.add(Dataforms::XDataField.new('os')).value = 'Mac'
+    form.add(Dataforms::XDataField.new('os_version')).value = '10.5.1'
+    ver = Caps::generate_ver([Discovery::Identity.new('client', 'Psi 0.9.1', 'pc').set_xml_lang('en'),
+                              Discovery::Identity.new('client', 'Ψ 0.9.1', 'pc').set_xml_lang('el')],
+                             [Discovery::Feature.new('http://jabber.org/protocol/muc'), # re-ordered
+                              Discovery::Feature.new('http://jabber.org/protocol/disco#info'),
+                              Discovery::Feature.new('http://jabber.org/protocol/disco#items')],
+                             [form])
+    assert_equal('8lu+88MRxmKM7yO3MEzY7YmTsWs=', ver)
+  end
+
+  ##
+  # http://www.xmpp.org/extensions/xep-0115.html#ver-gen-complex
+  def test_caps_complex_imported
+    query = IqQuery::import(REXML::Document.new(<<END).root)
+<query xmlns='http://jabber.org/protocol/disco#info'
+         node='http://psi-im.org#8lu+88MRxmKM7yO3MEzY7YmTsWs='>
+    <identity xml:lang='en' category='client' name='Psi 0.9.1' type='pc'/>
+    <identity xml:lang='el' category='client' name='Ψ 0.9.1' type='pc'/>
+    <feature var='http://jabber.org/protocol/disco#info'/>
+    <feature var='http://jabber.org/protocol/disco#items'/>
+    <feature var='http://jabber.org/protocol/muc'/>
+    <x xmlns='jabber:x:data' type='result'>
+      <field var='FORM_TYPE' type='hidden'>
+        <value>urn:xmpp:dataforms:softwareinfo</value>
+      </field>
+      <field var='ip_version'>
+        <value>ipv4</value>
+        <value>ipv6</value>
+      </field>
+      <field var='os'>
+        <value>Mac</value>
+      </field>
+      <field var='os_version'>
+        <value>10.5.1</value>
+      </field>
+      <field var='software'>
+        <value>Psi</value>
+      </field>
+      <field var='software_version'>
+        <value>0.11</value>
+      </field>
+    </x>
+  </query>
+END
+    assert_equal('8lu+88MRxmKM7yO3MEzY7YmTsWs=',
+                 Caps::generate_ver_from_discoinfo(query))
+  end
+end