2 files changed, 157 insertions, 0 deletions
diff --git a/spec/unit/util/queue.rb b/spec/unit/util/queue.rb
new file mode 100755
index 000000000..525e6239f
--- /dev/null
+++ b/spec/unit/util/queue.rb
@@ -0,0 +1,95 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../spec_helper'
+require 'puppet/util/queue'
+require 'spec/mocks'
+
+def make_test_client_class(n)
+    c = Class.new do
+        class <<self
+            attr_accessor :name
+            def to_s
+                name
+            end
+        end
+    end
+    c.name = n
+    c
+end
+
+mod = Puppet::Util::Queue
+client_classes = { :default => make_test_client_class('Bogus::Default'), :setup => make_test_client_class('Bogus::Setup') }
+mod.register_queue_type(client_classes[:default], :default)
+mod.register_queue_type(client_classes[:setup], :setup)
+
+describe Puppet::Util::Queue do
+    before :each do
+        @class = Class.new do
+            extend mod
+            self.queue_type_default = :default
+        end
+    end
+
+    context 'when determining a type name from a class' do
+        it 'should handle a simple one-word class name' do
+            mod.queue_type_from_class(make_test_client_class('Foo')).should == :foo
+        end
+
+        it 'should handle a simple two-word class name' do
+            mod.queue_type_from_class(make_test_client_class('FooBar')).should == :foo_bar
+        end
+
+        it 'should handle a two-part class name with one terminating word' do
+            mod.queue_type_from_class(make_test_client_class('Foo::Bar')).should == :bar
+        end
+
+        it 'should handle a two-part class name with two terminating words' do
+            mod.queue_type_from_class(make_test_client_class('Foo::BarBah')).should == :bar_bah
+        end
+    end
+
+    context 'when registering a queue client class' do
+        c = make_test_client_class('Foo::Bogus')
+        it 'uses the proper default name logic when type is unspecified' do
+            mod.register_queue_type(c)
+            mod.queue_type_to_class(:bogus).should == c
+        end
+
+        it 'uses an explicit type name when provided' do
+            mod.register_queue_type(c, :aardvark)
+            mod.queue_type_to_class(:aardvark).should == c
+        end
+
+        it 'throws an exception when type names conflict' do
+            mod.register_queue_type( make_test_client_class('Conflict') )
+            lambda { mod.register_queue_type( c, :conflict) }.should raise_error
+        end
+
+        it 'handle multiple, non-conflicting registrations' do
+            a = make_test_client_class('TestA')
+            b = make_test_client_class('TestB')
+            mod.register_queue_type(a)
+            mod.register_queue_type(b)
+            mod.queue_type_to_class(:test_a).should == a
+            mod.queue_type_to_class(:test_b).should == b
+        end
+
+        it 'throws an exception when type name is unknown' do
+            lambda { mod.queue_type_to_class(:nope) }.should raise_error
+        end
+    end
+
+    context 'when determining client type' do
+        it 'returns client class based on queue_type_default' do
+            Puppet.settings.stubs(:value).returns(nil)
+            @class.client_class.should == client_classes[:default]
+            @class.client.class.should == client_classes[:default]
+        end
+
+        it 'prefers settings variable for client class when specified' do
+            Puppet.settings.stubs(:value).with(:queue_client).returns(:setup)
+            @class.client_class.should == client_classes[:setup]
+            @class.client.class.should == client_classes[:setup]
+        end
+    end
+end
diff --git a/spec/unit/util/queue/stomp.rb b/spec/unit/util/queue/stomp.rb
new file mode 100755
index 000000000..c4d8b7672
--- /dev/null
+++ b/spec/unit/util/queue/stomp.rb
@@ -0,0 +1,62 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../../spec_helper'
+require 'puppet/util/queue'
+
+describe Puppet::Util::Queue do
+    it 'should load :stomp client appropriately' do
+        Puppet.settings.stubs(:value).returns 'faux_queue_source'
+        Puppet::Util::Queue.queue_type_to_class(:stomp).name.should == 'Puppet::Util::Queue::Stomp'
+    end
+end
+
+describe 'Puppet::Util::Queue::Stomp' do
+    before :all do
+        class Stomp::Client
+            include Mocha::Standalone
+            attr_accessor :queue_source
+
+            def send(q, m)
+                'To %s: %s' % [q, m]
+            end
+
+            def subscribe(q)
+                yield(stub(:body => 'subscribe: %s' % q))
+            end
+
+            def initialize(s)
+                self.queue_source = s
+            end
+        end
+    end
+
+    before :each do
+        Puppet.settings.stubs(:value).returns 'faux_queue_source'
+    end
+
+    it 'should make send function like core Ruby instead of stomp client send method' do
+        o = Puppet::Util::Queue::Stomp.new
+        o.expects(:pants).with('foo').once
+        o.send(:pants, 'foo')
+    end
+
+    it 'should be registered with Puppet::Util::Queue as :stomp type' do
+        Puppet::Util::Queue.queue_type_to_class(:stomp).should == Puppet::Util::Queue::Stomp
+    end
+
+    it 'should initialize using Puppet[:queue_source] for configuration' do
+        o = Puppet::Util::Queue::Stomp.new
+        o.stomp_client.queue_source.should == 'faux_queue_source'
+    end
+
+    it 'should transform the simple queue name to "/queue/<queue_name>"' do
+        Puppet::Util::Queue::Stomp.new.stompify_target('blah').should == '/queue/blah'
+    end
+
+    it 'should transform the queue name properly and pass along to superclass for send and subscribe' do
+        o = Puppet::Util::Queue::Stomp.new
+        o.send_message('fooqueue', 'Smite!').should == 'To /queue/fooqueue: Smite!'
+        o.subscribe('moodew') {|obj| obj}.should == 'subscribe: /queue/moodew'
+    end
+end
+