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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
require 'stomp_server/stomp_id'
require 'stomp_server/stomp_frame'
require 'stomp_server/queue_manager'
require 'stomp_server/queue'
require 'stomp_server/queue/file_queue'
require 'stomp_server/queue/memory_queue'
require 'stomp_server/queue/dbm_queue'
require 'test/unit'
require 'fileutils'
class TestQueues < Test::Unit::TestCase
class MockQueueManager < StompServer::QueueManager
def initialize(qstore)
@qstore = qstore
@queue_stats = Hash.new
@queues = Hash.new { Array.new }
@pending = Hash.new { Array.new }
end
end
class UserMock
attr_accessor :data
def initialize ; @data = '' ; end
def stomp_send_data(data); @data += data.to_s ; end
def connected?;true;end
end
class MessageMock
attr_accessor :headers, :data, :command, :body
def initialize(dest, msg, id=1)
@body = msg
@headers = {
'message-id' => id,
'destination' => dest,
'content-length' => msg.size.to_s
}
@frame = StompServer::StompFrame.new('MESSAGE', headers, body)
@data = @frame.to_s
end
def to_s ; @data.to_s ; end
end
def teardown
FileUtils.rm_rf(".queue_test")
end
def setup
FileUtils.rm_rf(".queue_test") if File.directory?('.queue_test')
@@qstore = StompServer::FileQueue.new(".queue_test")
@@qstore.checkpoint_interval=0
@t = MockQueueManager.new(@@qstore)
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
# 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
def test_queued_sendmsg
t = 'foo'
m1 = MessageMock.new('foo', 'foomsg')
@t.sendmsg(m1)
u = UserMock.new
@t.subscribe(t, u)
assert_equal(m1.data, u.data)
assert_equal('MESSAGE', m1.command)
u2 = UserMock.new
@t.subscribe(t, u2)
assert_equal('', u2.data)
end
end
|
