Make Handler configurable, add tests

lib/slanger/config.rb

@@ -13,7 +13,8 @@ def [](key)
     def options
       @options ||= {
         api_host: '0.0.0.0', api_port: '4567', websocket_host: '0.0.0.0',
-        websocket_port: '8080', debug: false, redis_address: 'redis://0.0.0.0:6379/0'
+        websocket_port: '8080', debug: false, redis_address: 'redis://0.0.0.0:6379/0',
+        socket_handler: Slanger::Handler
       }
     end
 

lib/slanger/web_socket_server.rb

@@ -9,7 +9,7 @@ def run
           # Keep track of handler instance in instance of EM::Connection to ensure a unique handler instance is used per connection.
           ws.class_eval    { attr_accessor :connection_handler }
           # Delegate connection management to handler instance.
-          ws.onopen        { ws.connection_handler = Slanger::Handler.new ws }
+          ws.onopen        { ws.connection_handler = Slanger::Config.socket_handler.new ws }
           ws.onmessage     { |msg| ws.connection_handler.onmessage msg }
           ws.onclose       { ws.connection_handler.onclose }
         end

spec/integration/replaced_handler_spec.rb

@@ -0,0 +1,66 @@
+require 'bundler/setup'
+
+require 'active_support/json'
+require 'active_support/core_ext/hash'
+require 'eventmachine'
+require 'em-http-request'
+require 'pusher'
+require 'thin'
+require './spec/spec_helper'
+
+describe 'replacable handler' do
+  let(:errback) { Proc.new { fail 'cannot connect to slanger. your box might be too slow. try increasing sleep value in the before block' } }
+
+  before(:each) do
+    # Fork service. Our integration tests MUST block the main thread because we want to wait for i/o to finish.
+    @server_pid = EM.fork_reactor do
+      require File.expand_path(File.dirname(__FILE__) + '/../../slanger.rb')
+
+      class ReplacedHandler < Slanger::Handler
+        def authenticate
+          super
+          @socket.send(payload nil, 'pusher:info', { message: "Welcome!" })
+        end
+      end
+
+      Thin::Logging.silent = true
+
+      Slanger::Config.load host:           '0.0.0.0',
+                           api_port:       '4567',
+                           websocket_port: '8080',
+                           app_key:        '765ec374ae0a69f4ce44',
+                           secret:         'your-pusher-secret',
+                           socket_handler: ReplacedHandler
+
+      Slanger::Service.run
+    end
+    # Give Slanger a chance to start
+    sleep 0.6
+  end
+
+  after(:each) do
+    # Ensure Slanger is properly stopped. No orphaned processes allowed!
+    Process.kill 'SIGKILL', @server_pid
+    Process.wait @server_pid
+  end
+
+  before :all do
+    Pusher.tap do |p|
+      p.host   = '0.0.0.0'
+      p.port   = 4567
+      p.app_id = 'your-pusher-app-id'
+      p.secret = 'your-pusher-secret'
+      p.key    = '765ec374ae0a69f4ce44'
+    end
+  end
+
+  it 'immediately disconnects' do
+    msgs = em_stream do |websocket, messages|
+      if messages.length == 2
+        EM.stop
+      end
+    end
+
+    msgs.last.should == {"channel"=>nil, "event"=>"pusher:info", "data"=>{"message"=>"Welcome!"}}
+  end
+end