vines 0.1.0

data/lib/vines/version.rb

@@ -0,0 +1,5 @@
+# encoding: UTF-8
+
+module Vines
+  VERSION = '0.1.0'
+end

data/lib/vines/xmpp_server.rb

@@ -0,0 +1,25 @@
+# encoding: UTF-8
+
+module Vines
+
+  # The main starting point for the XMPP server process. Starts the
+  # EventMachine processing loop and registers the XMPP protocol handler
+  # with the ports defined in the server configuration file.
+  class XmppServer
+    include Vines::Log
+
+    def initialize(config)
+      @config = config
+    end
+
+    def start
+      log.info('XMPP server started')
+      at_exit { log.fatal('XMPP server stopped') }
+      EM.epoll
+      EM.kqueue
+      EM.run do
+        @config.ports.each {|port| port.start }
+      end
+    end
+  end
+end

data/test/config_test.rb

@@ -0,0 +1,396 @@
+# encoding: UTF-8
+
+require 'vines'
+require 'test/unit'
+
+class ConfigTest < Test::Unit::TestCase
+  def test_missing_host
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        # missing hosts
+      end
+    end
+  end
+
+  def test_duplicate_host
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        host 'wonderland.lit' do
+          storage 'fs' do
+            dir '.'
+          end
+        end
+        host 'wonderland.lit' do
+          storage 'fs' do
+            dir '.'
+          end
+        end
+      end
+    end
+  end
+
+  def test_duplicate_host_in_one_call
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        host 'wonderland.lit', 'wonderland.lit' do
+          storage 'fs' do
+            dir '.'
+          end
+        end
+      end
+    end
+  end
+
+  def test_missing_storage
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        host 'wonderland.lit' do
+          # missing storage
+        end
+      end
+    end
+  end
+
+  def test_bad_storage
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        host 'wonderland.lit' do
+          storage 'bogus' do
+            # no bogus storage implementation
+          end
+        end
+      end
+    end
+  end
+
+  def test_duplicate_storage
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        host 'wonderland.lit' do
+          storage('fs') { dir '.' }
+          storage('fs') { dir '.' }
+        end
+      end
+    end
+  end
+
+  def test_good_storage
+    assert_nothing_raised do
+      Vines::Config.new do
+        host 'wonderland.lit' do
+          storage 'fs' do
+            dir '.'
+          end
+        end
+      end
+    end
+  end
+
+  def test_ldap_added_to_storage
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+        # added after storage
+        ldap 'ldap.wonderland.lit', 1636 do
+          tls true
+          dn 'cn=Directory Manager'
+          password 'secr3t'
+          basedn 'dc=wonderland,dc=lit'
+          object_class 'person'
+          user_attr 'uid'
+          name_attr 'cn'
+        end
+      end
+
+      host 'verona.lit' do
+        ldap 'ldap.verona.lit', 1636 do
+          tls true
+          dn 'cn=Directory Manager'
+          password 'secr3t'
+          basedn 'dc=wonderland,dc=lit'
+          object_class 'person'
+          user_attr 'uid'
+          name_attr 'cn'
+        end
+        # added before storage
+        storage(:fs) { dir '.' }
+      end
+    end
+    %w[wonderland.lit verona.lit].each do |domain|
+      assert_not_nil config.vhosts[domain].ldap
+      assert config.vhosts[domain].ldap?
+    end
+  end
+
+  def test_configure
+    config = Vines::Config.configure do
+      host 'wonderland.lit' do
+        storage :fs do
+          dir '.'
+        end
+      end
+    end
+    assert_not_nil config
+    assert_same config, Vines::Config.instance
+  end
+
+  def test_vhost
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+    end
+    assert_equal ['wonderland.lit'], config.vhosts.keys
+    assert config.vhost?('wonderland.lit')
+    assert !config.vhost?('bogus')
+  end
+
+  def test_port_lookup
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      client
+    end
+    assert_not_nil config[:client]
+    assert_raise(ArgumentError) { config[:server] }
+    assert_raise(ArgumentError) { config[:bogus] }
+  end
+
+  def test_duplicate_client
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        client
+        client
+      end
+    end
+  end
+
+  def test_duplicate_server
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        server
+        server
+      end
+    end
+  end
+
+  def test_duplicate_http
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        http
+        http
+      end
+    end
+  end
+
+  def test_duplicate_component
+    assert_raise(RuntimeError) do
+      Vines::Config.new do
+        component
+        component
+      end
+    end
+  end
+
+  def test_default_client
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      client
+    end
+    port = config.ports.first
+    assert_not_nil port
+    assert_equal Vines::Config::ClientPort, port.class
+    assert_equal '0.0.0.0', port.host
+    assert_equal 5222, port.port
+    assert_equal 131_072, port.max_stanza_size
+    assert_equal 5, port.max_resources_per_account
+    assert_equal Vines::Stream::Client, port.stream
+    assert_same config, port.config 
+    assert_equal 1, config.ports.size
+  end
+
+  def test_configured_client
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      client '0.0.0.1', 42 do
+        max_stanza_size 60_000
+        max_resources_per_account 1
+      end
+    end
+    port = config.ports.first
+    assert_not_nil port
+    assert_equal Vines::Config::ClientPort, port.class
+    assert_equal '0.0.0.1', port.host
+    assert_equal 42, port.port
+    assert_equal 60_000, port.max_stanza_size
+    assert_equal 1, port.max_resources_per_account
+    assert_equal Vines::Stream::Client, port.stream
+    assert_same config, port.config 
+    assert_equal 1, config.ports.size
+  end
+
+  def test_max_stanza_size
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      client do
+        max_stanza_size 0
+      end
+    end
+    assert_equal 10_000, config.ports.first.max_stanza_size
+  end
+
+  def test_default_server
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      server
+    end
+    port = config.ports.first
+    assert_not_nil port
+    assert !config.s2s?('verona.lit')
+    assert_equal Vines::Config::ServerPort, port.class
+    assert_equal '0.0.0.0', port.host
+    assert_equal 5269, port.port
+    assert_equal 131_072, port.max_stanza_size
+    assert_equal Vines::Stream::Server, port.stream
+    assert_same config, port.config 
+    assert_equal 1, config.ports.size
+  end
+
+  def test_configured_server
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      server '0.0.0.1', 42 do
+        max_stanza_size 60_000
+        hosts ['verona.lit', 'denmark.lit']
+      end
+    end
+    port = config.ports.first
+    assert_not_nil port
+    assert config.s2s?('verona.lit')
+    assert config.s2s?('denmark.lit')
+    assert !config.s2s?('bogus')
+    assert_equal Vines::Config::ServerPort, port.class
+    assert_equal '0.0.0.1', port.host
+    assert_equal 42, port.port
+    assert_equal 60_000, port.max_stanza_size
+    assert_equal Vines::Stream::Server, port.stream
+    assert_same config, port.config 
+    assert_equal 1, config.ports.size
+  end
+
+  def test_default_http
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      http
+    end
+    port = config.ports.first
+    assert_not_nil port
+    assert_equal Vines::Config::HttpPort, port.class
+    assert_equal '0.0.0.0', port.host
+    assert_equal 5280, port.port
+    assert_equal 131_072, port.max_stanza_size
+    assert_equal Vines::Stream::Http, port.stream
+    assert_same config, port.config 
+    assert_equal 1, config.ports.size
+  end
+
+  def test_configured_http
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      http '0.0.0.1', 42 do
+        max_stanza_size 60_000
+      end
+    end
+    port = config.ports.first
+    assert_not_nil port
+    assert_equal Vines::Config::HttpPort, port.class
+    assert_equal '0.0.0.1', port.host
+    assert_equal 42, port.port
+    assert_equal 60_000, port.max_stanza_size
+    assert_equal Vines::Stream::Http, port.stream
+    assert_same config, port.config 
+    assert_equal 1, config.ports.size
+  end
+
+  def test_default_component
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      component
+    end
+    port = config.ports.first
+    assert_not_nil port
+    assert port.components.empty?
+    assert_nil port.password('bogus')
+    assert_equal Vines::Config::ComponentPort, port.class
+    assert_equal '0.0.0.0', port.host
+    assert_equal 5347, port.port
+    assert_equal 131_072, port.max_stanza_size
+    assert_equal Vines::Stream::Component, port.stream
+    assert_same config, port.config 
+    assert_equal 1, config.ports.size
+  end
+
+  def test_configured_component
+    config = Vines::Config.new do
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+      component '0.0.0.1', 42 do
+        max_stanza_size 60_000
+        components 'tea.wonderland.lit'  => 'secr3t',
+                   'cake.wonderland.lit' => 'passw0rd'
+      end
+    end
+    port = config.ports.first
+    assert_not_nil port
+    assert_equal 2, port.components.size
+    assert_equal 'secr3t', port.password('tea.wonderland.lit')
+    assert_equal 'passw0rd', port.password('cake.wonderland.lit')
+    assert_nil port.password('bogus')
+    assert_equal Vines::Config::ComponentPort, port.class
+    assert_equal '0.0.0.1', port.host
+    assert_equal 42, port.port
+    assert_equal 60_000, port.max_stanza_size
+    assert_equal Vines::Stream::Component, port.stream
+    assert_same config, port.config 
+    assert_equal 1, config.ports.size
+  end
+
+  def test_invalid_log_level
+    assert_raise(RuntimeError) do
+      config = Vines::Config.new do
+        log 'bogus'
+        host 'wonderland.lit' do
+          storage(:fs) { dir '.' }
+        end
+      end
+    end
+  end
+
+  def test_valid_log_level
+    config = Vines::Config.new do
+      log :error
+      host 'wonderland.lit' do
+        storage(:fs) { dir '.' }
+      end
+    end
+    assert_equal Logger::ERROR, Class.new.extend(Vines::Log).log.level
+  end
+end

data/test/error_test.rb

@@ -0,0 +1,59 @@
+# encoding: UTF-8
+
+require 'vines'
+require 'test/unit'
+
+class ErrorTest < Test::Unit::TestCase
+  def test_sasl_error_without_text
+    expected = %q{<failure xmlns="urn:ietf:params:xml:ns:xmpp-sasl"><temporary-auth-failure/></failure>}
+    assert_equal expected, Vines::SaslErrors::TemporaryAuthFailure.new.to_xml
+  end
+
+  def test_sasl_error_with_text
+    text = %q{<text xml:lang="en">busted</text>}
+    expected = %q{<failure xmlns="urn:ietf:params:xml:ns:xmpp-sasl"><temporary-auth-failure/>%s</failure>} % text
+    assert_equal expected, Vines::SaslErrors::TemporaryAuthFailure.new('busted').to_xml
+  end
+
+  def test_stream_error_without_text
+    expected = %q{<stream:error><internal-server-error xmlns="urn:ietf:params:xml:ns:xmpp-streams"/></stream:error>}
+    assert_equal expected, Vines::StreamErrors::InternalServerError.new.to_xml
+  end
+
+  def test_stream_error_with_text
+    text = %q{<text xmlns="urn:ietf:params:xml:ns:xmpp-streams" xml:lang="en">busted</text>}
+    expected = %q{<stream:error><internal-server-error xmlns="urn:ietf:params:xml:ns:xmpp-streams"/>%s</stream:error>} % text
+    assert_equal expected, Vines::StreamErrors::InternalServerError.new('busted').to_xml
+  end
+
+  def test_stanza_error_with_bad_type
+    node = node('<message/>')
+    assert_raises(RuntimeError) { Vines::StanzaErrors::BadRequest.new(node, 'bogus') }
+  end
+
+  def test_stanza_error_with_bad_stanza
+    node = node('<bogus/>')
+    assert_raises(RuntimeError) { Vines::StanzaErrors::BadRequest.new(node, 'modify') }
+  end
+
+  def test_stanza_error_without_text
+    error = %q{<error type="modify"><bad-request xmlns="urn:ietf:params:xml:ns:xmpp-stanzas"/></error>}
+    expected = %q{<message from="hatter@wonderland.lit" to="alice@wonderland.lit" type="error">%s</message>} % error
+    node = node(%Q{<message from="alice@wonderland.lit" to="hatter@wonderland.lit"/>})
+    assert_equal expected, Vines::StanzaErrors::BadRequest.new(node, 'modify').to_xml
+  end
+
+  def test_stanza_error_with_text
+    text = %q{<text xmlns="urn:ietf:params:xml:ns:xmpp-stanzas" xml:lang="en">busted</text>}
+    error = %q{<error type="modify"><bad-request xmlns="urn:ietf:params:xml:ns:xmpp-stanzas"/>%s</error>} % text
+    expected = %q{<message id="42" type="error">%s</message>} % error
+    node = node(%Q{<message id="42"/>})
+    assert_equal expected, Vines::StanzaErrors::BadRequest.new(node, 'modify', 'busted').to_xml
+  end
+
+  private
+
+  def node(xml)
+    Nokogiri::XML(xml).root
+  end
+end