stompserver 0.9.1

data/test/test_frame_journal.rb

@@ -0,0 +1,14 @@
+require 'frame_journal'
+require 'test/unit' unless defined? $ZENTEST and $ZENTEST
+require 'tesly'
+
+# how would you test this? It is pretty simple code
+# based almost completely on Madeleine...
+class TestFrameJournal < Test::Unit::TestCase
+  def setup
+  end  
+  
+  def test_something
+    
+  end
+end

data/test/test_queue_manager.rb

@@ -0,0 +1,83 @@
+require 'queue_manager'
+require 'test/unit' unless defined? $ZENTEST and $ZENTEST
+require 'tesly'
+
+class TestQueues < Test::Unit::TestCase
+  
+  class UserMock
+    attr_accessor :data
+    def initialize ; @data = '' ; end
+    def send_data(data); @data += data ; end
+  end
+  
+  class MessageMock
+    attr_accessor :headers, :data, :command
+    def initialize(dest, msg, id=1)  
+      @headers = { 'destination' => dest, 'message-id' => id }
+      @data = msg
+    end
+    def to_s ; @data ; end
+  end
+
+  FileUtils.rm Dir.glob(".test_queue/*") rescue nil
+  @@journal = FrameJournal.new(".test_queue")
+  
+  def setup
+    #needs a journal
+    @@journal.clear
+    @t = QueueManager.new(@@journal)
+  end
+
+  def test_subscribe
+    u = UserMock.new
+    t = 'foo'
+    @t.subscribe(t, u)
+    
+    m1 = MessageMock.new('foo', 'foomsg')
+    m2 = MessageMock.new('bar', 'barmsg')
+    @t.sendmsg(m1)
+    assert_equal(m1.data, u.data)
+    
+    u.data = ''
+    @t.sendmsg(m2)
+    assert_equal('', u.data)
+  end
+
+  def test_subscribe2
+    t = 'sub2'
+    m1 = MessageMock.new(t, 'sub2msg')
+    @t.sendmsg(m1)
+    
+    u = UserMock.new
+    @t.subscribe(t, u)
+    
+    assert_equal(m1.data, u.data)
+  end
+
+  def test_unsubscribe
+    u = UserMock.new
+    t = 'foo'
+    @t.subscribe(t, u)
+    
+    m1 = MessageMock.new('foo', 'foomsg')
+    @t.sendmsg(m1)
+    assert_equal(m1.data, u.data)
+
+    @t.unsubscribe(t,u)
+    u.data = ''
+    @t.sendmsg(m1)
+    assert_equal('', u.data)        
+  end
+
+  def test_sendmsg(msg)
+    u = UserMock.new
+    t = 'foo'
+    @t.subscribe(t, u)
+    
+    m1 = MessageMock.new('foo', 'foomsg')
+    @t.sendmsg(m1)
+    assert_equal(m1.data, u.data)
+    assert_equal('MESSAGE', m1.command)
+  end
+
+end

data/test/test_stomp_frame.rb

@@ -0,0 +1,121 @@
+require 'stomp_frame'
+require 'test/unit' unless defined? $ZENTEST and $ZENTEST
+require 'tesly'
+
+class TestStompFrame < Test::Unit::TestCase
+  def setup
+    @sfr = StompFrameRecognizer.new
+  end
+  
+  def test_simpleframe
+    @sfr << <<FRAME
+COMMAND
+name:value
+foo:bar
+
+message body
+\000
+FRAME
+    assert_equal(1, @sfr.frames.size)
+    f = @sfr.frames.shift
+    assert_equal(0, @sfr.frames.size)
+    assert_equal("COMMAND", f.command)
+    assert_equal("value", f.headers["name"])
+    assert_equal("bar", f.headers["foo"])
+    assert_equal("message body\n", f.body)
+  end
+  
+  def test_doubleframe
+    @sfr << <<FRAME
+COMMAND
+name:value
+foo:bar
+
+message body
+\000
+
+COMMAND2
+name2:value2
+foo2:bar2
+
+message body 2
+\000
+FRAME
+    assert_equal(2, @sfr.frames.size)
+    f = @sfr.frames.shift
+    assert_equal(1, @sfr.frames.size)
+    assert_equal("COMMAND", f.command)
+    assert_equal("value", f.headers["name"])
+    assert_equal("bar", f.headers["foo"])
+    assert_equal("message body\n", f.body)
+    
+    # check second frame
+    f = @sfr.frames.shift
+    assert_equal(0, @sfr.frames.size)
+    assert_equal("COMMAND2", f.command)
+    assert_equal("value2", f.headers["name2"])
+    assert_equal("bar2", f.headers["foo2"])
+    assert_equal("message body 2\n", f.body)
+  end
+  
+    def test_partialframe
+    @sfr << <<FRAME
+COMMAND
+name:value
+foo:bar
+
+message body
+\000
+
+COMMAND2
+name2:value2
+foo2:bar2
+
+message body 2
+FRAME
+    assert_equal(1, @sfr.frames.size)
+    f = @sfr.frames.shift
+    assert_equal(0, @sfr.frames.size)
+    assert_equal("COMMAND", f.command)
+    assert_equal("value", f.headers["name"])
+    assert_equal("bar", f.headers["foo"])
+    assert_equal("message body\n", f.body)    
+  end
+
+  def test_partialframe2
+    @sfr << <<FRAME
+COMMAND
+name:value
+foo:bar
+FRAME
+    assert_equal(0, @sfr.frames.size)
+  end
+  
+  def test_headless_frame
+    @sfr << <<FRAME
+COMMAND
+
+message body\000
+FRAME
+    assert_equal(1, @sfr.frames.size)
+    f = @sfr.frames.shift
+    assert_equal(0, @sfr.frames.size)
+    assert_equal("COMMAND", f.command)
+    assert_equal("message body", f.body)    
+  end
+
+  def test_destination_cache
+    @sfr << <<FRAME
+MESSAGE
+destination: /queue/foo
+
+message body\000
+FRAME
+    assert_equal(1, @sfr.frames.size)
+    f = @sfr.frames.shift
+    assert_equal(0, @sfr.frames.size)
+    assert_equal("MESSAGE", f.command)
+    assert_equal("message body", f.body)
+    assert_equal('/queue/foo', f.dest)    
+  end  
+end

data/test/test_stomp_server.rb

@@ -0,0 +1,241 @@
+require 'stomp_server'
+require 'frame_journal'
+require 'fileutils'
+require 'test/unit' unless defined? $ZENTEST and $ZENTEST
+require 'tesly'
+
+#$DEBUG = true
+
+class TestStompServer < Test::Unit::TestCase
+
+  # Is it a mock? it is what we are testing, but of 
+  # course I am really testing the module, so I say
+  # yes it is a mock :-)
+  class MockStompServer
+    include StompServer
+    attr_accessor :sent, :connected
+    
+    def initialize
+      @sent = ''
+      @connected = true
+    end
+    
+    def send_data(data)
+      @sent += data
+    end
+    
+    def close_connection_after_writing
+      @connected = false
+    end
+    alias close_connection close_connection_after_writing
+    
+    def stomp(cmd, headers={}, body='', flush_prev = true)
+      @sent = '' if flush_prev
+      sf = StompFrame.new(cmd, headers, body)
+      receive_data(sf.to_s)
+    end
+    
+    def do_connect(flush = true)
+      stomp('CONNECT')
+      @sent = '' if flush
+    end
+    
+    def self.make_client(start_connection=true, flush=true)
+      ss = MockStompServer.new
+      ss.post_init
+      ss.do_connect(flush) if start_connection
+      ss
+    end    
+    
+  end
+
+  def assert_stomp_error(ss=@ss, match = nil)
+    assert_match(/ERROR/, ss.sent)
+    assert_match(match, ss.sent) if match
+    assert(!ss.connected)
+  end
+  
+  FileUtils.rm Dir.glob(".test_journal/*") rescue nil
+  @@journal = FrameJournal.new(".test_journal")
+  StompServer.setup(@@journal)
+  
+  def setup
+    # be sure and delete anything in our journal directory
+    @@journal.clear
+    @ss = MockStompServer.make_client
+  end  
+  
+  def test_version
+    assert(StompServer.const_defined?(:VERSION))
+  end
+  
+  def test_invalid_command
+    assert_nothing_raised do
+      @ss.stomp('INVALID')
+    end
+    assert_stomp_error
+  end
+  
+  def test_unconnected_command
+    ss = MockStompServer.make_client(false)
+    assert_nothing_raised do
+      ss.stomp('SEND')
+    end
+    assert_stomp_error(ss)
+  end
+  
+  def test_connect
+    ss = MockStompServer.make_client(false)
+    assert(!ss.connected)    
+    assert_nothing_raised do
+      ss.connect(false)
+    end
+    assert_match(/CONNECTED/, ss.sent)
+    assert(ss.connected)
+  end
+  
+  def test_disconnect
+    assert_nothing_raised do
+      @ss.stomp('DISCONNECT')
+    end
+    assert(!@ss.connected)
+  end
+  
+  def test_receipt
+    assert_nothing_raised do
+      @ss.stomp('SUBSCRIBE', {'receipt' => 'foobar'})
+    end
+    
+    assert_match(/RECEIPT/, @ss.sent)
+    assert_match(/receipt-id:foobar/, @ss.sent)
+    
+    assert(@ss.connected)    
+  end
+  
+  def test_topic
+    assert_equal('', @ss.sent)
+    
+    # setup two clients (@ss and this one)
+    ss2 = MockStompServer.make_client
+    assert_equal('', ss2.sent)
+
+    @ss.stomp('SEND', {'destination' => '/topic/foo'}, 'Hi Pat')
+    @ss.stomp('SEND', {'destination' => '/topic/foo'}, 'Hi Sue')
+    
+    assert_equal('', @ss.sent)
+    assert_equal('', ss2.sent)
+    
+    ss2.stomp("SUBSCRIBE", {'destination' => '/topic/foo'})
+    assert_equal('', ss2.sent)
+    
+    @ss.stomp('SEND', {'destination' => '/topic/foo'}, 'Hi Pat')    
+    assert_match(/Hi Pat/, ss2.sent)
+    assert_equal('', @ss.sent)
+  end
+  
+  def test_bad_topic
+    assert_equal('', @ss.sent)
+    
+    # setup two clients (@ss and this one)
+    ss2 = MockStompServer.make_client
+    assert_equal('', ss2.sent)
+
+    ss2.stomp("SUBSCRIBE", {'destination' => '/badtopic/foo'})
+    assert_equal('', ss2.sent)
+    
+    @ss.stomp('SEND', {'destination' => '/badtopic/foo'}, 'Hi Pat')    
+    assert_match(/Hi Pat/, ss2.sent)
+    assert_equal('', @ss.sent)
+  end
+  
+  def test_multiple_subscriber_topic
+    assert_equal('', @ss.sent)
+    
+    # setup two clients (@ss and this one)
+    ss2 = MockStompServer.make_client
+    assert_equal('', ss2.sent)
+
+    @ss.stomp("SUBSCRIBE", {'destination' => '/topic/foo'})
+    ss2.stomp("SUBSCRIBE", {'destination' => '/topic/foo'})
+    
+    assert_equal('', @ss.sent)
+    assert_equal('', ss2.sent)
+    
+    @ss.stomp('SEND', {'destination' => '/topic/foo'}, 'Hi Pat')
+        
+    assert_match(/Hi Pat/, ss2.sent)
+    assert_match(/Hi Pat/, @ss.sent)
+  end
+ 
+  def test_invalid_transaction
+    @ss.stomp("SEND", {'transaction' => 't'})
+    assert_stomp_error
+  end
+  
+  def test_simple_transaction
+    ss1 = MockStompServer.make_client
+    ss2 = MockStompServer.make_client
+    
+    ss2.stomp("SUBSCRIBE", {"destination" => '/topic/foo'})
+    assert_equal('', ss2.sent)
+        
+    ss1.stomp("BEGIN", {"transaction" => 'simple'})
+    ss1.stomp("SEND", {"transaction" => 'simple', 'destination' => '/topic/foo'}, 'Hi Pat')
+    assert_equal('', ss2.sent)
+    assert_equal('', ss1.sent)
+
+    ss1.stomp("COMMIT", {"transaction" => 'simple'})    
+    assert_equal('', ss1.sent)
+    assert_match(/Hi Pat/, ss2.sent)
+  end
+  
+  def test_simple_transaction2
+    ss1 = MockStompServer.make_client
+    ss2 = MockStompServer.make_client
+    
+    ss2.stomp("BEGIN", {"transaction" => 'simple'})
+    ss2.stomp("SUBSCRIBE", {"transaction" => 'simple', "destination" => '/topic/foo'})
+    assert_equal('', ss2.sent)
+        
+    ss1.stomp("SEND", {'destination' => '/topic/foo'}, 'Hi Pat')
+    assert_equal('', ss2.sent)
+    assert_equal('', ss1.sent)
+
+    ss2.stomp("COMMIT", {"transaction" => 'simple'})    
+    assert_equal('', ss1.sent)
+    assert_equal('', ss2.sent)
+
+    ss1.stomp("SEND", {'destination' => '/topic/foo'}, 'Hi Pat')
+    assert_match(/Hi Pat/, ss2.sent)
+  end
+
+  def test_simple_abort_transaction
+    ss1 = MockStompServer.make_client
+    ss2 = MockStompServer.make_client
+    
+    ss2.stomp("BEGIN", {"transaction" => 'simple'})
+    ss2.stomp("SUBSCRIBE", {"transaction" => 'simple', "destination" => '/topic/foo'})
+    assert_equal('', ss2.sent)
+        
+    ss1.stomp("SEND", {'destination' => '/topic/foo'}, 'Hi Pat')
+    assert_equal('', ss2.sent)
+    assert_equal('', ss1.sent)
+
+    ss2.stomp("ABORT", {"transaction" => 'simple'})    
+    assert_equal('', ss1.sent)
+    assert_equal('', ss2.sent)
+    
+    ss1.stomp("SEND", {'destination' => '/topic/foo'}, 'Hi Pat')
+    assert_match('', ss2.sent)
+  end
+  
+  def test_simple_queue_message
+    ss1 = MockStompServer.make_client
+    ss1.stomp("SEND", {'destination' => '/queue/foo'}, 'Hi Pat')
+    ss1.stomp("DISCONNECT")
+
+    ss2 = MockStompServer.make_client
+    ss2.stomp("SUBSCRIBE", {"destination" => '/queue/foo'})
+    assert_match(/Hi Pat/, ss2.sent)
+  end
+end

data/test/test_topic_manager.rb

@@ -0,0 +1,68 @@
+require 'topic_manager'
+require 'test/unit' unless defined? $ZENTEST and $ZENTEST
+require 'tesly_reporter'
+
+class TestTopics < Test::Unit::TestCase
+
+  class UserMock
+    attr_accessor :data
+    def initialize ; @data = '' ; end
+    def send_data(data); @data += data ; end
+  end
+  
+  class MessageMock
+    attr_accessor :headers, :data, :command
+    def initialize(dest, msg)  
+      @headers = { 'destination' => dest }
+      @data = msg
+    end
+    def to_s ; @data ; end
+  end
+  
+  def setup
+    @t = TopicManager.new
+  end
+    
+  def test_subscribe
+    u = UserMock.new
+    t = 'foo'
+    @t.subscribe(t, u)
+    
+    m1 = MessageMock.new('foo', 'foomsg')
+    m2 = MessageMock.new('bar', 'barmsg')
+    @t.sendmsg(m1)
+    assert_equal(m1.data, u.data)
+    
+    u.data = ''
+    @t.sendmsg(m2)
+    assert_equal('', u.data)
+  end
+
+  def test_unsubscribe
+    u = UserMock.new
+    t = 'foo'
+    @t.subscribe(t, u)
+    
+    m1 = MessageMock.new('foo', 'foomsg')
+    @t.sendmsg(m1)
+    assert_equal(m1.data, u.data)
+
+    @t.unsubscribe(t,u)
+    u.data = ''
+    @t.sendmsg(m1)
+    assert_equal('', u.data)        
+  end
+
+  def test_sendmsg(msg)
+    u = UserMock.new
+    t = 'foo'
+    @t.subscribe(t, u)
+    
+    m1 = MessageMock.new('foo', 'foomsg')
+    @t.sendmsg(m1)
+    assert_equal(m1.data, u.data)
+    assert_equal('MESSAGE', m1.command)
+  end
+
+end
+

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.0
+specification_version: 1
+name: stompserver
+version: !ruby/object:Gem::Version 
+  version: 0.9.1
+date: 2006-10-16 00:00:00 -04:00
+summary: A very light messaging server
+require_paths: 
+- lib
+- test
+email: phurley@gmail.com
+homepage: "    by Patrick Hurley"
+rubyforge_project: stompserver
+description: "Don't want to install a JVM, but still want to use messaging? Me too, so I threw together this little server. All the hard work was done by Francis Cianfrocca (big thank you) in his event machine gem (which is required by this server).  == FEATURES/PROBLEMS:  Handles basic message queue processing Does not support any server to server messaging (although you could write a client to do this) Server Id is not being well initialized    Quite a bit of polish is still required to make into a daemon/service  and add command line handling. And oh yeah, I need to write some docs (see the tests for now)"
+autorequire: 
+default_executable: 
+bindir: bin
+has_rdoc: true
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+- Patrick Hurley
+files: 
+- History.txt
+- Manifest.txt
+- README.txt
+- Rakefile
+- setup.rb
+- bin/stompserver
+- lib/frame_journal.rb
+- lib/queue_manager.rb
+- lib/stomp_frame.rb
+- lib/stomp_server.rb
+- lib/topic_manager.rb
+- test/test_frame_journal.rb
+- test/test_queue_manager.rb
+- test/test_stomp_frame.rb
+- test/test_stomp_server.rb
+- test/test_topic_manager.rb
+test_files: []
+
+rdoc_options: []
+
+extra_rdoc_files: []
+
+executables: 
+- stompserver
+extensions: []
+
+requirements: []
+
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: eventmachine
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Version::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 5.0.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: madeleine
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Version::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.7.3
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: hoe
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Version::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.1.1
+    version: