| /* |
| * 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. |
| * |
| * |
| */ |
| package org.apache.qpid.server.exchange; |
| |
| import junit.framework.TestCase; |
| import junit.framework.Assert; |
| import org.apache.qpid.server.queue.*; |
| import org.apache.qpid.server.virtualhost.VirtualHost; |
| import org.apache.qpid.server.registry.ApplicationRegistry; |
| import org.apache.qpid.server.txn.NonTransactionalContext; |
| import org.apache.qpid.server.txn.TransactionalContext; |
| import org.apache.qpid.server.store.MessageStore; |
| import org.apache.qpid.server.store.MemoryMessageStore; |
| import org.apache.qpid.server.store.StoreContext; |
| import org.apache.qpid.server.RequiredDeliveryException; |
| import org.apache.qpid.server.protocol.InternalTestProtocolSession; |
| import org.apache.qpid.AMQException; |
| import org.apache.qpid.framing.AMQShortString; |
| import org.apache.qpid.framing.ContentHeaderBody; |
| import org.apache.qpid.framing.abstraction.MessagePublishInfo; |
| |
| import java.util.LinkedList; |
| |
| public class DestWildExchangeTest extends TestCase |
| { |
| |
| TopicExchange _exchange; |
| |
| VirtualHost _vhost; |
| MessageStore _store; |
| StoreContext _context; |
| |
| InternalTestProtocolSession _protocolSession; |
| |
| |
| public void setUp() throws AMQException |
| { |
| _exchange = new TopicExchange(); |
| _vhost = ApplicationRegistry.getInstance().getVirtualHostRegistry().getVirtualHosts().iterator().next(); |
| _store = new MemoryMessageStore(); |
| _context = new StoreContext(); |
| _protocolSession = new InternalTestProtocolSession(); |
| } |
| |
| public void tearDown() |
| { |
| ApplicationRegistry.remove(1); |
| } |
| |
| |
| public void testNoRoute() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a*#b"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.*.#.b"), queue, null); |
| |
| |
| MessagePublishInfo info = new PublishInfo(new AMQShortString("a.b")); |
| |
| IncomingMessage message = new IncomingMessage(0L, info, null, _protocolSession); |
| |
| _exchange.route(message); |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| } |
| |
| public void testDirectMatch() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("ab"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.b"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a.b"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| |
| message = createMessage("a.c"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has no route and should fail to be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| } |
| |
| |
| public void testStarMatch() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a*"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.*"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a.b"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| |
| message = createMessage("a.c"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| |
| message = createMessage("a"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has no route and should fail to be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| } |
| |
| public void testHashMatch() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a#"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.#"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a.b.c"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| |
| message = createMessage("a.b"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| |
| message = createMessage("a.c"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| message = createMessage("a"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| |
| message = createMessage("b"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has no route and should fail to be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| } |
| |
| |
| public void testMidHash() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.*.#.b"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a.c.d.b"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has no route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| message = createMessage("a.c.b"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has no route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| } |
| |
| public void testMatchafterHash() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a#"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.*.#.b.c"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a.c.b.b"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has route and should not be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| |
| message = createMessage("a.a.b.c"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has no route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| message = createMessage("a.b.c.b"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has route and should not be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| message = createMessage("a.b.c.b.c"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has no route and should be routed"); |
| |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| } |
| |
| |
| public void testHashAfterHash() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a#"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.*.#.b.c.#.d"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a.c.b.b.c"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has route and should not be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| |
| message = createMessage("a.a.b.c.d"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has no route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| } |
| |
| public void testHashHash() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a#"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.#.*.#.d"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a.c.b.b.c"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has route and should not be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| message = createMessage("a.a.b.c.d"); |
| |
| try |
| { |
| routeMessage(message); |
| } |
| catch (AMQException nre) |
| { |
| fail("Message has no route and should be routed"); |
| } |
| |
| Assert.assertEquals(1, queue.getMessageCount()); |
| |
| Assert.assertEquals("Wrong message recevied", (Object) message.getMessageId(), queue.getMessagesOnTheQueue().get(0).getMessage().getMessageId()); |
| |
| queue.deleteMessageFromTop(_context); |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| } |
| |
| public void testSubMatchFails() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.b.c.d"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a.b.c"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has route and should not be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| } |
| |
| private void routeMessage(final IncomingMessage message) |
| throws AMQException |
| { |
| _exchange.route(message); |
| message.routingComplete(_store, new MessageHandleFactory()); |
| message.deliverToQueues(); |
| } |
| |
| public void testMoreRouting() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.b"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a.b.c"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has route and should not be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| } |
| |
| public void testMoreQueue() throws AMQException |
| { |
| AMQQueue queue = AMQQueueFactory.createAMQQueueImpl(new AMQShortString("a"), false, null, false, _vhost, null); |
| _exchange.registerQueue(new AMQShortString("a.b"), queue, null); |
| |
| |
| IncomingMessage message = createMessage("a"); |
| |
| try |
| { |
| routeMessage(message); |
| fail("Message has route and should not be routed"); |
| } |
| catch (AMQException nre) |
| { |
| } |
| |
| Assert.assertEquals(0, queue.getMessageCount()); |
| |
| } |
| |
| private IncomingMessage createMessage(String s) throws AMQException |
| { |
| MessagePublishInfo info = new PublishInfo(new AMQShortString(s)); |
| |
| TransactionalContext trancontext = new NonTransactionalContext(_store, _context, null, |
| new LinkedList<RequiredDeliveryException>() |
| ); |
| |
| IncomingMessage message = new IncomingMessage(0L, info, trancontext,_protocolSession); |
| message.setContentHeaderBody( new ContentHeaderBody()); |
| |
| |
| return message; |
| } |
| |
| |
| class PublishInfo implements MessagePublishInfo |
| { |
| AMQShortString _routingkey; |
| |
| PublishInfo(AMQShortString routingkey) |
| { |
| _routingkey = routingkey; |
| } |
| |
| public AMQShortString getExchange() |
| { |
| return null; |
| } |
| |
| public void setExchange(AMQShortString exchange) |
| { |
| |
| } |
| |
| public boolean isImmediate() |
| { |
| return false; |
| } |
| |
| public boolean isMandatory() |
| { |
| return true; |
| } |
| |
| public AMQShortString getRoutingKey() |
| { |
| return _routingkey; |
| } |
| } |
| } |