Apache
/
activemq-artemis
mirror of https://github.com/apache/activemq-artemis.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix openwire tests - ProducerFlowControlSendFailTest

This commit is contained in:
Howard Gao 2015-08-07 13:40:51 +08:00 committed by Clebert Suconic
parent 4ed21a44c5
commit 5230237a7c
1 changed files with 35 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
tests/activemq5-unit-tests/src/main/java/org/apache/activemq/broker/artemiswrapper/ArtemisBrokerWrapper.java
View File

@ -32,6 +32,7 @@ import org.apache.activemq.artemis.core.config.impl.SecurityConfiguration;
import org.apache.activemq.artemis.core.registry.JndiBindingRegistry; import org.apache.activemq.artemis.core.registry.JndiBindingRegistry;
import org.apache.activemq.artemis.core.remoting.impl.netty.TransportConstants; import org.apache.activemq.artemis.core.remoting.impl.netty.TransportConstants;
import org.apache.activemq.artemis.core.security.Role; import org.apache.activemq.artemis.core.security.Role;
import org.apache.activemq.artemis.core.settings.impl.AddressFullMessagePolicy;
import org.apache.activemq.artemis.core.settings.impl.AddressSettings; import org.apache.activemq.artemis.core.settings.impl.AddressSettings;
import org.apache.activemq.artemis.core.settings.impl.SlowConsumerPolicy; import org.apache.activemq.artemis.core.settings.impl.SlowConsumerPolicy;
import org.apache.activemq.artemis.jms.client.ActiveMQDestination; import org.apache.activemq.artemis.jms.client.ActiveMQDestination;
@ -196,12 +197,12 @@ public class ArtemisBrokerWrapper extends ArtemisBrokerBase
private void translatePolicyMap(Configuration serverConfig, PolicyMap policyMap) private void translatePolicyMap(Configuration serverConfig, PolicyMap policyMap)
{ {
List allEntries = policyMap.getAllEntries(); List allEntries = policyMap.getAllEntries();
Map<String, AddressSettings> settingsMap = serverConfig.getAddressesSettings();
for (Object o : allEntries) for (Object o : allEntries)
{ {
PolicyEntry entry = (PolicyEntry)o; PolicyEntry entry = (PolicyEntry)o;
org.apache.activemq.command.ActiveMQDestination targetDest = entry.getDestination(); org.apache.activemq.command.ActiveMQDestination targetDest = entry.getDestination();
String match = getCorePattern(targetDest); String match = getCorePattern(targetDest);
Map<String, AddressSettings> settingsMap = serverConfig.getAddressesSettings();
AddressSettings settings = settingsMap.get(match); AddressSettings settings = settingsMap.get(match);
if (settings == null) if (settings == null)
{ {
@ -216,6 +217,25 @@ public class ArtemisBrokerWrapper extends ArtemisBrokerBase
settings.setSlowConsumerPolicy(SlowConsumerPolicy.NOTIFY); settings.setSlowConsumerPolicy(SlowConsumerPolicy.NOTIFY);
} }
} }
PolicyEntry defaultEntry = policyMap.getDefaultEntry();
if (defaultEntry != null)
{
AddressSettings defSettings = settingsMap.get("#");
if (defSettings == null)
{
defSettings = new AddressSettings();
settingsMap.put("#", defSettings);
}
if (defaultEntry.isProducerFlowControl())
{
defSettings.setMaxSizeBytes(1).setAddressFullMessagePolicy(AddressFullMessagePolicy.BLOCK);
if (bservice.getSystemUsage().isSendFailIfNoSpace())
{
defSettings.setAddressFullMessagePolicy(AddressFullMessagePolicy.FAIL);
}
}
}
} }
private String getCorePattern(org.apache.activemq.command.ActiveMQDestination dest) private String getCorePattern(org.apache.activemq.command.ActiveMQDestination dest)
@ -236,11 +256,22 @@ public class ArtemisBrokerWrapper extends ArtemisBrokerBase
@Override @Override
public void stop() throws Exception public void stop() throws Exception
{
try
{ {
server.stop(); server.stop();
testQueues.clear(); testQueues.clear();
stopped = true; stopped = true;
} }
catch (Throwable t)
{
//ignore
}
finally
{
server = null;
}
}
public void makeSureQueueExists(String qname) throws Exception public void makeSureQueueExists(String qname) throws Exception
{ {