spec/unit/util/queue/stomp.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
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