diff options
Diffstat (limited to 'qpid/java/broker/src/test/java/org/apache/qpid/server/virtualhost/VirtualHostImplTest.java')
| -rw-r--r-- | qpid/java/broker/src/test/java/org/apache/qpid/server/virtualhost/VirtualHostImplTest.java | 214 |
1 files changed, 214 insertions, 0 deletions
diff --git a/qpid/java/broker/src/test/java/org/apache/qpid/server/virtualhost/VirtualHostImplTest.java b/qpid/java/broker/src/test/java/org/apache/qpid/server/virtualhost/VirtualHostImplTest.java new file mode 100644 index 0000000000..c87e5a1648 --- /dev/null +++ b/qpid/java/broker/src/test/java/org/apache/qpid/server/virtualhost/VirtualHostImplTest.java @@ -0,0 +1,214 @@ +/* + * + * 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.virtualhost; + +import java.io.BufferedWriter; +import java.io.File; +import java.io.FileWriter; +import java.io.IOException; + +import org.apache.commons.configuration.ConfigurationException; +import org.apache.commons.configuration.XMLConfiguration; +import org.apache.qpid.server.configuration.ServerConfiguration; +import org.apache.qpid.server.exchange.Exchange; +import org.apache.qpid.server.queue.AMQQueue; +import org.apache.qpid.server.registry.ApplicationRegistry; +import org.apache.qpid.server.store.TestableMemoryMessageStore; +import org.apache.qpid.server.util.TestApplicationRegistry; +import org.apache.qpid.test.utils.QpidTestCase; + +public class VirtualHostImplTest extends QpidTestCase +{ + private ServerConfiguration _configuration; + private ApplicationRegistry _registry; + + @Override + public void tearDown() throws Exception + { + super.tearDown(); + + ApplicationRegistry.remove(); + } + + /** + * Tests that custom routing keys for the queue specified in the configuration + * file are correctly bound to the exchange (in addition to the queue name) + */ + public void testSpecifyingCustomBindings() throws Exception + { + customBindingTestImpl(new String[]{"custom1","custom2"}); + } + + /** + * Tests that a queue specified in the configuration file to be bound to a + * specified(non-default) direct exchange is a correctly bound to the exchange + * and the default exchange using the queue name. + */ + public void testQueueSpecifiedInConfigurationIsBoundToDefaultExchange() throws Exception + { + customBindingTestImpl(new String[0]); + } + + private void customBindingTestImpl(final String[] routingKeys) throws Exception + { + String exchangeName = getName() +".direct"; + String vhostName = getName(); + String queueName = getName(); + + File config = writeConfigFile(vhostName, queueName, exchangeName, false, routingKeys); + VirtualHost vhost = createVirtualHost(vhostName, config); + assertNotNull("virtualhost should exist", vhost); + + AMQQueue queue = vhost.getQueueRegistry().getQueue(queueName); + assertNotNull("queue should exist", queue); + + Exchange defaultExch = vhost.getExchangeRegistry().getDefaultExchange(); + assertTrue("queue should have been bound to default exchange with its name", defaultExch.isBound(queueName, queue)); + + Exchange exch = vhost.getExchangeRegistry().getExchange(exchangeName); + assertTrue("queue should have been bound to " + exchangeName + " with its name", exch.isBound(queueName, queue)); + + for(String key: routingKeys) + { + assertTrue("queue should have been bound to " + exchangeName + " with key " + key, exch.isBound(key, queue)); + } + } + + /** + * Tests that specifying custom routing keys for a queue in the configuration file results in failure + * to create the vhost (since this is illegal, only queue names are used with the default exchange) + */ + public void testSpecifyingCustomBindingForDefaultExchangeThrowsException() throws Exception + { + File config = writeConfigFile(getName(), getName(), null, false, new String[]{"custom-binding"}); + + try + { + createVirtualHost(getName(), config); + fail("virtualhost creation should have failed due to illegal configuration"); + } + catch (ConfigurationException e) + { + //expected + } + } + + /** + * Tests that specifying an unknown exchange to bind the queue to results in failure to create the vhost + */ + public void testSpecifyingUnknownExchangeThrowsException() throws Exception + { + File config = writeConfigFile(getName(), getName(), "made-up-exchange", true, new String[0]); + + try + { + createVirtualHost(getName(), config); + fail("virtualhost creation should have failed due to illegal configuration"); + } + catch (ConfigurationException e) + { + //expected + } + } + + private VirtualHost createVirtualHost(String vhostName, File config) throws Exception + { + _configuration = new ServerConfiguration(new XMLConfiguration(config)); + + _registry = new TestApplicationRegistry(_configuration); + ApplicationRegistry.initialise(_registry); + + return _registry.getVirtualHostRegistry().getVirtualHost(vhostName); + } + + /** + * Create a configuration file for testing virtualhost creation + * + * @param vhostName name of the virtualhost + * @param queueName name of the queue + * @param exchangeName name of a direct exchange to declare (unless dontDeclare = true) and bind the queue to (null = none) + * @param dontDeclare if true then dont declare the exchange, even if its name is non-null + * @param routingKeys routingKeys to bind the queue with (empty array = none) + * @return + */ + private File writeConfigFile(String vhostName, String queueName, String exchangeName, boolean dontDeclare, String[] routingKeys) + { + File tmpFile = null; + try + { + tmpFile = File.createTempFile(getName(), ".tmp"); + tmpFile.deleteOnExit(); + + FileWriter fstream = new FileWriter(tmpFile); + BufferedWriter writer = new BufferedWriter(fstream); + + //extra outer tag to please Commons Configuration + writer.write("<configuration>"); + + writer.write("<virtualhosts>"); + writer.write(" <default>" + vhostName + "</default>"); + writer.write(" <virtualhost>"); + writer.write(" <store>"); + writer.write(" <class>" + TestableMemoryMessageStore.class.getName() + "</class>"); + writer.write(" </store>"); + writer.write(" <name>" + vhostName + "</name>"); + writer.write(" <" + vhostName + ">"); + if(exchangeName != null && !dontDeclare) + { + writer.write(" <exchanges>"); + writer.write(" <exchange>"); + writer.write(" <type>direct</type>"); + writer.write(" <name>" + exchangeName + "</name>"); + writer.write(" </exchange>"); + writer.write(" </exchanges>"); + } + writer.write(" <queues>"); + writer.write(" <queue>"); + writer.write(" <name>" + queueName + "</name>"); + writer.write(" <" + queueName + ">"); + if(exchangeName != null) + { + writer.write(" <exchange>" + exchangeName + "</exchange>"); + } + for(String routingKey: routingKeys) + { + writer.write(" <routingKey>" + routingKey + "</routingKey>"); + } + writer.write(" </" + queueName + ">"); + writer.write(" </queue>"); + writer.write(" </queues>"); + writer.write(" </" + vhostName + ">"); + writer.write(" </virtualhost>"); + writer.write("</virtualhosts>"); + + writer.write("</configuration>"); + + writer.flush(); + writer.close(); + } + catch (IOException e) + { + fail("Unable to create virtualhost configuration"); + } + + return tmpFile; + } +} |