| # |
| # 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 |
| |
| class SelectorTests (Base): |
| """ |
| Tests for the selector filter registered for AMQP 1.0 under the |
| apache namespace. |
| """ |
| def setup_connection(self): |
| return Connection.establish(self.broker, **self.connection_options()) |
| |
| def setup_session(self): |
| return self.conn.session() |
| |
| def basic_selection_test(self, node): |
| properties = [(1, 'red','dog'), (2, 'black', 'cat'), (3, 'red', 'squirrel'), (4, 'grey', 'squirrel')] |
| msgs = [Message(content="%s.%s" % (colour, creature), properties={'sequence':sequence,'colour':colour}) for sequence, colour, creature in properties] |
| |
| snd = self.ssn.sender(node) |
| rcv = self.ssn.receiver("%s; {link:{selector:\"colour IN ('red', 'grey') AND (sequence > 3 OR sequence = 1)\"}}" % snd.target) |
| |
| for m in msgs: snd.send(m) |
| |
| for expected in ["red.dog", "grey.squirrel"]: |
| msg = rcv.fetch(0) |
| assert msg.content == expected |
| self.ssn.acknowledge(msg) |
| |
| def test_topic(self): |
| self.basic_selection_test(self.config.defines.get("topic_name", "amq.fanout")) |
| |
| def test_queue(self): |
| self.basic_selection_test("#") |
| |
| def test_special_fields(self): |
| msgs = [Message(content=i, id=i, correlation_id=i, priority=p) for p, i in enumerate(['a', 'b', 'c', 'd'], 1)] |
| |
| snd = self.ssn.sender("#") |
| rcv_1 = self.ssn.receiver("%s; {link:{selector:\"amqp.message_id = 'c'\"}}" % snd.target) |
| rcv_2 = self.ssn.receiver("%s; {link:{selector:\"amqp.correlation_id = 'b'\"}}" % snd.target) |
| rcv_3 = self.ssn.receiver("%s; {link:{selector:\"amqp.priority = 1\"}}" % snd.target) |
| |
| for m in msgs: snd.send(m) |
| |
| msg = rcv_1.fetch(0) |
| assert msg.content == 'c', msg |
| self.ssn.acknowledge(msg) |
| |
| msg = rcv_2.fetch(0) |
| assert msg.content == 'b', msg |
| self.ssn.acknowledge(msg) |
| |
| msg = rcv_3.fetch(0) |
| assert msg.content == 'a', msg |
| self.ssn.acknowledge(msg) |
| |
| rcv_4 = self.ssn.receiver(snd.target) |
| msg = rcv_4.fetch(0) |
| assert msg.content == 'd' |
| self.ssn.acknowledge(msg) |