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
|
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
#
from qpid.tests.messaging.implementation import *
from qpid.tests.messaging import Base
import math
import random
class LVQTests (Base):
"""
Test last value queue behaviour
"""
def setup_connection(self):
return Connection.establish(self.broker, **self.connection_options())
def setup_session(self):
return self.conn.session()
def test_simple(self):
snd = self.ssn.sender("lvq; {create: sender, delete: sender, node: {x-declare:{arguments:{'qpid.last_value_queue_key':lvq-key}}}}",
durable=self.durable())
snd.send(create_message("a", "a-1"))
snd.send(create_message("b", "b-1"))
snd.send(create_message("a", "a-2"))
snd.send(create_message("a", "a-3"))
snd.send(create_message("c", "c-1"))
snd.send(create_message("c", "c-2"))
rcv = self.ssn.receiver("lvq; {mode: browse}")
assert fetch_all(rcv) == ["b-1", "a-3", "c-2"]
snd.send(create_message("b", "b-2"))
assert fetch_all(rcv) == ["b-2"]
snd.send(create_message("c", "c-3"))
snd.send(create_message("d", "d-1"))
assert fetch_all(rcv) == ["c-3", "d-1"]
snd.send(create_message("b", "b-3"))
assert fetch_all(rcv) == ["b-3"]
rcv.close()
rcv = self.ssn.receiver("lvq; {mode: browse}")
assert (fetch_all(rcv) == ["a-3", "c-3", "d-1", "b-3"])
def check_ring_lvq(self, ring_size, keys, message_count):
address = "lvq; {create: sender, delete: sender, node: {x-declare:{arguments:{'qpid.last_value_queue_key':lvq-key,'qpid.policy_type':'ring','qpid.max_count':%i}}}}" % ring_size
snd = self.ssn.sender(address, durable=self.durable())
counters = {}
for k in keys:
counters[k] = 0
messages = []
for i in range(message_count):
k = random.choice(keys)
counters[k] += 1
messages.append(create_message(k, "%s-%i" % (k, counters[k])))
# make sure we have sent at least one message for every key
for k, v in counters.iteritems():
if v == 0:
counters[k] += 1
messages.append(create_message(k, "%s-%i" % (k, counters[k])))
for m in messages:
snd.send(m)
rcv = self.ssn.receiver("lvq; {mode: browse}")
retrieved = fetch_all_as_tuples(rcv)
print [v for k, v in retrieved]
for k, v in retrieved:
assert v == "%s-%i" % (k, counters[k])
assert len(retrieved) <= ring_size
def test_ring_lvq1(self):
self.check_ring_lvq(25, ["a","b","c","d"], 50)
def test_ring_lvq2(self):
self.check_ring_lvq(5, ["a","b","c","d","e","f","g"], 50)
def test_ring_lvq3(self):
self.check_ring_lvq(49, ["a"], 50)
def create_message(key, content):
msg = Message(content=content, properties={"lvq-key":key})
return msg
def fetch_all(rcv):
content = []
while True:
try:
content.append(rcv.fetch(0).content)
except Empty:
break
return content
def fetch_all_as_tuples(rcv):
content = []
while True:
try:
m = rcv.fetch(0)
k = m.properties["lvq-key"]
content.append((k, m.content))
except Empty:
break
return content
|
