blather 0.1

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