sprsquish-blather 0.1

data/spec/blather/core/roster_item_spec.rb

@@ -0,0 +1,80 @@
+require File.join(File.dirname(__FILE__), *%w[.. .. spec_helper])
+
+describe 'Blather::RosterItem' do
+  it 'initializes with JID' do
+    jid = JID.new(jid)
+    i = RosterItem.new jid
+    i.jid.must_equal jid
+  end
+
+  it 'initializes with an Iq::RosterItem' do
+    jid = 'n@d/r'
+    i = RosterItem.new Stanza::Iq::Roster::RosterItem.new(jid)
+    i.jid.must_equal JID.new(jid).stripped
+  end
+
+  it 'has a JID setter that strips the JID' do
+    jid = JID.new('n@d/r')
+    i = RosterItem.new nil
+    i.jid = jid
+    i.jid.must_equal jid.stripped
+  end
+
+  it 'has a subscription setter that forces a symbol' do
+    i = RosterItem.new nil
+    i.subscription = 'remove'
+    i.subscription.must_equal :remove
+  end
+
+  it 'forces the type of subscription' do
+    proc { RosterItem.new(nil).subscription = 'foo' }.must_raise Blather::ArgumentError
+  end
+
+  it 'returns :none if the subscription field is blank' do
+    RosterItem.new(nil).subscription.must_equal :none
+  end
+
+  it 'ensure #ask is a symbol' do
+    i = RosterItem.new(nil)
+    i.ask = 'subscribe'
+    i.ask.must_equal :subscribe
+  end
+
+  it 'forces #ask to be :subscribe or nothing at all' do
+    proc { RosterItem.new(nil).ask = 'foo' }.must_raise Blather::ArgumentError
+  end
+
+  it 'generates a stanza with #to_stanza' do
+    jid = JID.new('n@d/r')
+    i = RosterItem.new jid
+    s = i.to_stanza
+    s.must_be_kind_of Stanza::Iq::Roster
+    s.items.first.jid.must_equal jid.stripped
+  end
+
+  it 'returns status based on priority' do
+    setup_item_with_presences
+    @i.status.must_equal @p2
+  end
+
+  it 'returns status based on resource' do
+    setup_item_with_presences
+    @i.status('a').must_equal @p
+  end
+
+  def setup_item_with_presences
+    @jid = JID.new('n@d/r')
+    @i = RosterItem.new @jid
+
+    @p = Stanza::Presence::Status.new(:away)
+    @p.from = 'n@d/a'
+    @p.priority = 0
+
+    @p2 = Stanza::Presence::Status.new(:dnd)
+    @p2.from = 'n@d/b'
+    @p2.priority = -1
+
+    @i.status = @p
+    @i.status = @p2
+  end
+end

data/spec/blather/core/roster_spec.rb

@@ -0,0 +1,79 @@
+require File.join(File.dirname(__FILE__), *%w[.. .. spec_helper])
+
+describe 'Blather::Roster' do
+  before do
+    @stream = mock()
+    @stream.stubs(:send_data)
+
+    @stanza = mock()
+    items = []; 4.times { |n| items << JID.new("n@d/#{n}r") }
+    @stanza.stubs(:items).returns(items)
+
+    @roster = Roster.new(@stream, @stanza)
+  end
+
+  it 'initializes with items' do
+    @roster.items.map { |_,i| i.jid.to_s }.must_equal(@stanza.items.map { |i| i.stripped.to_s }.uniq)
+  end
+
+  it 'processes @stanzas with remove requests' do
+    s = @roster['n@d/0r']
+    s.subscription = :remove
+    proc { @roster.process(s.to_stanza) }.must_change('@roster.items', :length, :by => -1)
+  end
+
+  it 'processes @stanzas with add requests' do
+    s = Stanza::Iq::Roster::RosterItem.new('a@b/c').to_stanza
+    proc { @roster.process(s) }.must_change('@roster.items', :length, :by => 1)
+  end
+
+  it 'allows a jid to be pushed' do
+    jid = 'a@b/c'
+    proc { @roster.push(jid) }.must_change('@roster.items', :length, :by => 1)
+    @roster[jid].wont_be_nil
+  end
+
+  it 'allows an item to be pushed' do
+    jid = 'a@b/c'
+    item = RosterItem.new(JID.new(jid))
+    proc { @roster.push(item) }.must_change('@roster.items', :length, :by => 1)
+    @roster[jid].wont_be_nil
+  end
+
+  it 'sends a @roster addition over the wire' do
+    stream = mock()
+    stream.expects(:send_data)
+    roster = Roster.new stream, @stanza
+    roster.push('a@b/c')
+  end
+
+  it 'removes a JID' do
+    proc { @roster.delete 'n@d' }.must_change('@roster.items', :length, :by => -1)
+  end
+
+  it 'sends a @roster removal over the wire' do
+    stream = mock(:send_data => nil)
+    roster = Roster.new stream, @stanza
+    roster.delete('a@b/c')
+  end
+
+  it 'returns an item through []' do
+    item = @roster['n@d']
+    item.must_be_kind_of RosterItem
+    item.jid.must_equal JID.new('n@d')
+  end
+
+  it 'responds to #each' do
+    @roster.must_respond_to :each
+  end
+
+  it 'cycles through the items using #each' do
+    @roster.map { |i| i }.sort.must_equal(@roster.items.sort)
+  end
+
+  it 'returns a duplicate of items through #items' do
+    items = @roster.items
+    items.delete 'n@d'
+    items.wont_equal @roster.items
+  end
+end

data/spec/blather/core/stanza_spec.rb

@@ -0,0 +1,95 @@
+require File.join(File.dirname(__FILE__), *%w[.. .. spec_helper])
+
+describe 'Blather::Stanza' do
+  it 'provides .next_id helper for generating new IDs' do
+    proc { Stanza.next_id }.must_change 'Stanza', :next_id
+  end
+
+  it 'can import a node' do
+    s = Stanza.import XMPPNode.new('foo')
+    s.element_name.must_equal 'foo'
+  end
+
+  it 'sets the ID when created' do
+    Stanza.new('message').id.wont_be_nil
+  end
+
+  it 'sets the document when created' do
+    Stanza.new('message').doc.wont_be_nil
+  end
+
+  it 'provides an #error? helper' do
+    s = Stanza.new('message')
+    s.error?.must_equal false
+    s.type = :error
+    s.error?.must_equal true
+  end
+
+  it 'will generate a reply' do
+    s = Stanza.new('message')
+    s.from = f = JID.new('n@d/r')
+    s.to = t = JID.new('d@n/r')
+
+    r = s.reply
+    r.object_id.wont_equal s.object_id
+    r.from.must_equal t
+    r.to.must_equal f
+  end
+
+  it 'convert to a reply' do
+    s = Stanza.new('message')
+    s.from = f = JID.new('n@d/r')
+    s.to = t = JID.new('d@n/r')
+
+    r = s.reply!
+    r.object_id.must_equal s.object_id
+    r.from.must_equal t
+    r.to.must_equal f
+  end
+
+  it 'provides "attr_accessor" for id' do
+    s = Stanza.new('message')
+    s.id.wont_be_nil
+    s['id'].wont_be_nil
+
+    s.id = nil
+    s.id.must_be_nil
+    s['id'].must_be_nil
+  end
+
+  it 'provides "attr_accessor" for to' do
+    s = Stanza.new('message')
+    s.to.must_be_nil
+    s['to'].must_be_nil
+
+    s.to = JID.new('n@d/r')
+    s.to.wont_be_nil
+    s.to.must_be_kind_of JID
+
+    s['to'].wont_be_nil
+    s['to'].must_equal 'n@d/r'
+  end
+
+  it 'provides "attr_accessor" for from' do
+    s = Stanza.new('message')
+    s.from.must_be_nil
+    s['from'].must_be_nil
+
+    s.from = JID.new('n@d/r')
+    s.from.wont_be_nil
+    s.from.must_be_kind_of JID
+
+    s['from'].wont_be_nil
+    s['from'].must_equal 'n@d/r'
+  end
+
+  it 'provides "attr_accessor" for type' do
+    s = Stanza.new('message')
+    s.type.must_be_nil
+    s['type'].must_be_nil
+
+    s.type = 'testing'
+    s.type.wont_be_nil
+    s['type'].wont_be_nil
+  end
+end

data/spec/blather/core/stream_spec.rb

@@ -0,0 +1,263 @@
+require File.join(File.dirname(__FILE__), *%w[.. .. spec_helper])
+
+describe 'Blather::Stream' do
+  class MockStream; include Stream; end
+  def mock_stream(&block)
+    @client = mock()
+    @client.stubs(:jid=)
+    stream = MockStream.new @client, JID.new('n@d/r'), 'pass'
+
+    stream.expects(:send_data).at_least(1).with &block
+    stream
+  end
+
+  it 'can be started' do
+    client = mock()
+    params = [client, 'n@d/r', 'pass', 'host', 1234]
+    EM.expects(:connect).with do |*parms|
+      parms[0] == 'host'  &&
+      parms[1] == 1234    &&
+      parms[3] == client  &&
+      parms[5] == 'pass'  &&
+      parms[4] == JID.new('n@d/r')
+    end
+
+    Stream.start *(params)
+  end
+
+  it 'can figure out the host to use based on the jid' do
+    client = mock()
+    params = [client, 'n@d/r', 'pass', 'd', 5222]
+    EM.expects(:connect).with do |*parms|
+      parms[0] == 'd'     &&
+      parms[1] == 5222    &&
+      parms[3] == client  &&
+      parms[5] == 'pass'  &&
+      parms[4] == JID.new('n@d/r')
+    end
+
+    Stream.start client, 'n@d/r', 'pass'
+  end
+
+  it 'starts the stream once the connection is complete' do
+    s = mock_stream { |d| d =~ /stream:stream/ }
+    s.connection_completed
+  end
+
+  it 'starts TLS when asked' do
+    state = nil
+    @stream = mock_stream do |val|
+      case
+      when state.nil? && val =~ /stream:stream/   then state = :started
+      when state == :started && val =~ /starttls/ then true
+      else false
+      end
+    end
+    @stream.connection_completed
+    @stream.receive_data "<stream:stream><stream:features><starttls xmlns='urn:ietf:params:xml:ns:xmpp-tls' /></stream:features>"
+  end
+
+  it 'connects via SASL MD5 when asked' do
+    Time.any_instance.stubs(:to_f).returns(1.1)
+
+    state = nil
+    client = mock()
+    client.stubs(:jid=)
+    stream = MockStream.new client, JID.new('n@d/r'), 'pass'
+
+    stream.expects(:send_data).times(5).with do |val|
+      case state
+      when nil
+        val.must_match(/stream:stream/)
+        state = :started
+        stream.receive_data "<stream:stream><stream:features><mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'><mechanism>DIGEST-MD5</mechanism></mechanisms></stream:features>"
+        true
+
+      when :started
+        val.must_match(/auth.*DIGEST\-MD5/)
+        state = :auth_sent
+        stream.receive_data "<challenge xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>cmVhbG09InNvbWVyZWFsbSIsbm9uY2U9Ik9BNk1HOXRFUUdtMmhoIixxb3A9ImF1dGgiLGNoYXJzZXQ9dXRmLTgsYWxnb3JpdGhtPW1kNS1zZXNzCg==</challenge>"
+        true
+
+      when :auth_sent
+        val.must_equal('<response xmlns="urn:ietf:params:xml:ns:xmpp-sasl">bm9uY2U9Ik9BNk1HOXRFUUdtMmhoIixjaGFyc2V0PXV0Zi04LHVzZXJuYW1lPSJuIixyZWFsbT0ic29tZXJlYWxtIixjbm9uY2U9Ijc3N2Q0NWJiYmNkZjUwZDQ5YzQyYzcwYWQ3YWNmNWZlIixuYz0wMDAwMDAwMSxxb3A9YXV0aCxkaWdlc3QtdXJpPSJ4bXBwL2Qi</response>')
+        state = :response1_sent
+        stream.receive_data "<challenge xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>cnNwYXV0aD1lYTQwZjYwMzM1YzQyN2I1NTI3Yjg0ZGJhYmNkZmZmZAo=</challenge>"
+        true
+
+      when :response1_sent
+        val.must_equal('<response xmlns="urn:ietf:params:xml:ns:xmpp-sasl"/>')
+        state = :response2_sent
+        stream.receive_data "<success xmlns='urn:ietf:params:xml:ns:xmpp-sasl'/>"
+        true
+
+      when :response2_sent
+        val.must_match(/stream:stream/)
+        state = :complete
+        true
+
+      else
+        false
+
+      end
+    end
+    stream.connection_completed
+  end
+
+  it 'will connect via SSL PLAIN when asked' do
+    state = nil
+    client = mock()
+    client.stubs(:jid=)
+    stream = MockStream.new client, JID.new('n@d/r'), 'pass'
+
+    stream.expects(:send_data).times(3).with do |val|
+      case state
+      when nil
+        val.must_match(/stream:stream/)
+        state = :started
+        stream.receive_data "<stream:stream><stream:features><mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'><mechanism>PLAIN</mechanism></mechanisms></stream:features>"
+        true
+
+      when :started
+        val.must_equal('<auth xmlns="urn:ietf:params:xml:ns:xmpp-sasl" mechanism="PLAIN">bkBkAG4AcGFzcw==</auth>')
+        state = :auth_sent
+        stream.receive_data "<success xmlns='urn:ietf:params:xml:ns:xmpp-sasl'/>"
+        true
+
+      when :auth_sent
+        val.must_match(/stream:stream/)
+        state = :complete
+        true
+
+      else
+        false
+
+      end
+    end
+    stream.connection_completed
+  end
+
+  it 'will connect via SSL ANONYMOUS when asked' do
+    state = nil
+    client = mock()
+    client.stubs(:jid=)
+    stream = MockStream.new client, JID.new('n@d/r'), 'pass'
+
+    stream.expects(:send_data).times(3).with do |val|
+      case state
+      when nil
+        val.must_match(/stream:stream/)
+        state = :started
+        stream.receive_data "<stream:stream><stream:features><mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'><mechanism>ANONYMOUS</mechanism></mechanisms></stream:features>"
+        true
+
+      when :started
+        val.must_equal('<auth xmlns="urn:ietf:params:xml:ns:xmpp-sasl" mechanism="ANONYMOUS">bg==</auth>')
+        state = :auth_sent
+        stream.receive_data "<success xmlns='urn:ietf:params:xml:ns:xmpp-sasl'/>"
+        true
+
+      when :auth_sent
+        val.must_match(/stream:stream/)
+        state = :complete
+        true
+
+      else
+        false
+
+      end
+    end
+    stream.connection_completed
+  end
+
+  it 'will bind to a resource set by the server' do
+    state = nil
+    class Client; attr_accessor :jid; end
+    client = Client.new
+
+    jid = JID.new('n@d')
+    stream = MockStream.new client, jid, 'pass'
+
+    stream.expects(:send_data).times(2).with do |val|
+      case state
+      when nil
+        val.must_match(/stream:stream/)
+        state = :started
+        stream.receive_data "<stream:stream><stream:features><bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'/></stream:features>"
+        true
+
+      when :started
+        val.must_match(%r{<bind xmlns="urn:ietf:params:xml:ns:xmpp-bind"\s*/>})
+        val =~ %r{<iq[^>]+id="([^"]+)"}
+        state = :complete
+        stream.receive_data "<iq type='result' id='#{$1}'><bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'><jid>#{jid}/server_resource</jid></bind></iq>"
+        client.jid.must_equal JID.new('n@d/server_resource')
+        true
+
+      else
+        false
+
+      end
+    end
+    stream.connection_completed
+  end
+
+  it 'will bind to a resource set by the client' do
+    state = nil
+    class Client; attr_accessor :jid; end
+    client = Client.new
+
+    jid = JID.new('n@d/r')
+    stream = MockStream.new client, jid, 'pass'
+
+    stream.expects(:send_data).times(2).with do |val|
+      case state
+      when nil
+        val.must_match(/stream:stream/)
+        state = :started
+        stream.receive_data "<stream:stream><stream:features><bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'/></stream:features>"
+        true
+
+      when :started
+        val.must_match(%r{<bind xmlns="urn:ietf:params:xml:ns:xmpp-bind"><resource>r</resource></bind>})
+        val =~ %r{<iq[^>]+id="([^"]+)"}
+        state = :complete
+        stream.receive_data "<iq type='result' id='#{$1}'><bind xmlns='urn:ietf:params:xml:ns:xmpp-bind'><jid>#{jid}</jid></bind></iq>"
+        client.jid.must_equal JID.new('n@d/r')
+        true
+
+      else
+        false
+
+      end
+    end
+    stream.connection_completed
+  end
+
+  it 'will establish a session if requested' do
+    state = nil
+    client = mock()
+    client.stubs(:jid=)
+    stream = MockStream.new client, JID.new('n@d/r'), 'pass'
+
+    stream.expects(:send_data).times(2).with do |val|
+      case state
+      when nil
+        val.must_match(/stream:stream/)
+        state = :started
+        stream.receive_data "<stream:stream><stream:features><session xmlns='urn:ietf:params:xml:ns:xmpp-session'/></stream:features>"
+        true
+
+      when :started
+        val.must_match('<iq id="[^"]+" type="set" to="d"><session xmlns="urn:ietf:params:xml:ns:xmpp-session"/></iq>')
+        state = :completed
+        true
+
+      else
+        false
+
+      end
+    end
+    stream.connection_completed
+  end
+end

data/spec/blather/core/xmpp_node_spec.rb

@@ -0,0 +1,130 @@
+require File.join(File.dirname(__FILE__), *%w[.. .. spec_helper])
+
+describe 'Blather::XMPPNode' do
+  it 'generates a new node' do
+    n = XMPPNode.new 'foo'
+    n.element_name.must_equal 'foo'
+  end
+
+  it 'generates a node based on the current name' do
+    class Foo < XMPPNode; end
+    Foo.name = 'foo'
+    Foo.new.element_name.must_equal 'foo'
+  end
+
+  it 'sets the namespace on creation' do
+    class Foo < XMPPNode; end
+    Foo.xmlns = 'foo'
+    Foo.new('foo').xmlns.must_equal 'foo'
+  end
+
+  it 'registers sub classes' do
+    class Foo < XMPPNode; register 'foo', 'foo:bar'; end
+    Foo.name.must_equal 'foo'
+    Foo.xmlns.must_equal 'foo:bar'
+    XMPPNode.class_from_registration('foo', 'foo:bar').must_equal Foo
+  end
+
+  it 'imports another node' do
+    class Foo < XMPPNode; register 'foo', 'foo:bar'; end
+    n = XMPPNode.new('foo')
+    n.xmlns = 'foo:bar'
+    XMPPNode.import(n).must_be_kind_of Foo
+  end
+
+  it 'can convert itself into a stanza' do
+    class Foo < XMPPNode; register 'foo'; end
+    n = XMPPNode.new('foo')
+    n.to_stanza.must_be_kind_of Foo
+  end
+
+  it 'provides "attr_accessor" for xmlns' do
+    n = XMPPNode.new('foo')
+    n.xmlns.must_be_nil
+    n['xmlns'].must_be_nil
+
+    n.xmlns = 'foo:bar'
+    n.xmlns.must_equal 'foo:bar'
+    n['xmlns'].must_equal 'foo:bar'
+  end
+
+  it 'will remove a child element' do
+    n = XMPPNode.new 'foo'
+    n << XMPPNode.new('bar')
+    n << XMPPNode.new('bar')
+
+    n.find('bar').size.must_equal 2
+    n.remove_child 'bar'
+    n.find('bar').size.must_equal 1
+  end
+
+  it 'will remove a child with a specific xmlns' do
+    n = XMPPNode.new 'foo'
+    n << XMPPNode.new('bar')
+    c = XMPPNode.new('bar')
+    c.xmlns = 'foo:bar'
+    n << c
+
+    n.find('bar').size.must_equal 2
+    n.remove_child 'bar', 'foo:bar'
+    n.find('bar').size.must_equal 1
+    n.find('bar').first.xmlns.must_be_nil
+  end
+
+  it 'will remove all child elements' do
+    n = XMPPNode.new 'foo'
+    n << XMPPNode.new('bar')
+    n << XMPPNode.new('bar')
+
+    n.find('bar').size.must_equal 2
+    n.remove_children 'bar'
+    n.find('bar').size.must_equal 0
+  end
+
+  it 'provides a helper to grab content from a child' do
+    n = XMPPNode.new 'foo'
+    n << XMPPNode.new('bar', 'baz')
+    n.content_from(:bar).must_equal 'baz'
+  end
+
+  it 'provides a copy mechanism' do
+    n = XMPPNode.new 'foo'
+    n2 = n.copy
+    n2.object_id.wont_equal n.object_id
+    n2.element_name.must_equal n.element_name
+  end
+
+  it 'provides an inhert mechanism' do
+    n = XMPPNode.new 'foo'
+    n2 = XMPPNode.new 'foo', 'bar'
+    n2['foo'] = 'bar'
+
+    n.inherit(n2)
+    n['foo'].must_equal 'bar'
+    n.content.must_equal 'bar'
+  end
+
+  it 'provides a mechanism to inherit attrs' do
+    n = XMPPNode.new 'foo'
+    n2 = XMPPNode.new 'foo'
+    n2['foo'] = 'bar'
+
+    n.inherit_attrs(n2.attributes)
+    n['foo'].must_equal 'bar'
+  end
+
+  it 'cuts line breaks out of #to_s' do
+    n = XMPPNode.new 'foo'
+    n << XMPPNode.new('bar', 'baz')
+    n.to_s.scan(">\n<").size.must_equal 0
+  end
+
+  it 'overrides #find to find without xpath' do
+    n = XMPPNode.new 'foo'
+    n << XMPPNode.new('bar', 'baz')
+    n.find('bar').must_be_kind_of Array
+
+    XML::Document.new.root = n
+    n.find('bar').must_be_kind_of XML::XPath::Object
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,49 @@
+require File.join(File.dirname(__FILE__), *%w[.. lib blather])
+require 'rubygems'
+require 'minitest/spec'
+require 'mocha'
+
+module MiniTest
+  if MINI_DIR =~ %r{^./}
+    require 'pathname'
+    path = Pathname.new(MINI_DIR).realpath
+#    remove_const 'MINI_DIR'
+#    const_set 'MINI_DIR', path.to_s
+  end
+
+  module Assertions
+    def assert_change(obj, method, args = {}, msg = nil)
+      msg ||= proc {
+        m = "Expected #{obj}.#{method} to change"
+        m << " by #{mu_pp args[:by]}" if args[:by]
+        m << (args[:from] ? " from #{mu_pp args[:from]}" : '') + " to #{mu_pp args[:to]}" if args[:to]
+        m
+      }.call
+
+      init_val = eval(obj).__send__(method)
+      yield
+      new_val = eval(obj).__send__(method)
+
+      assert_equal(args[:by], (new_val - init_val), msg) if args[:by]
+      assert_equal([args[:from], args[:to]], [(init_val if args[:from]), new_val], msg) if args[:to]
+      refute_equal(init_val, new_val, msg) if args.empty?
+    end
+  end
+end
+
+class Object
+  def must_change *args, &block
+    return MiniTest::Spec.current.assert_change(*args, &self)     if Proc === self
+    return MiniTest::Spec.current.assert_change(args.first, self) if args.size == 1
+    return MiniTest::Spec.current.assert_change(self, *args)
+  end
+end
+
+require 'mocha/expectation_error'
+
+include Blather
+include MiniTest
+
+LOG.level = Logger::INFO
+
+Unit.autorun