mirror of https://github.com/apache/activemq.git
Merge pull request #759 from mattrpav/AMQ-8413b
[AMQ-8413] NetworkConnector remote user and pass unit test
This commit is contained in:
Christopher L. Shannon
GitHub
commit
fccb40f518
|
|
@ -0,0 +1,131 @@
|
|||
/**
|
||||
* 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.activemq.network;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
import java.net.URI;
|
||||
|
||||
import javax.jms.Connection;
|
||||
import javax.jms.JMSException;
|
||||
import javax.jms.Message;
|
||||
import javax.jms.MessageConsumer;
|
||||
import javax.jms.MessageProducer;
|
||||
import javax.jms.Session;
|
||||
import javax.jms.TextMessage;
|
||||
|
||||
import org.apache.activemq.ActiveMQConnectionFactory;
|
||||
import org.junit.Test;
|
||||
|
||||
public class NetworkRemoteUserPasswordTest extends BaseNetworkTest {
|
||||
|
||||
@Override
|
||||
protected void doSetUp(boolean deleteAllMessages) throws Exception {
|
||||
remoteBroker = createRemoteBroker();
|
||||
remoteBroker.setDeleteAllMessagesOnStartup(deleteAllMessages);
|
||||
remoteBroker.start();
|
||||
remoteBroker.waitUntilStarted();
|
||||
localBroker = createLocalBroker();
|
||||
localBroker.setDeleteAllMessagesOnStartup(deleteAllMessages);
|
||||
localBroker.start();
|
||||
localBroker.waitUntilStarted();
|
||||
URI localURI = localBroker.getVmConnectorURI();
|
||||
ActiveMQConnectionFactory fac = new ActiveMQConnectionFactory(localURI);
|
||||
fac.setAlwaysSyncSend(true);
|
||||
fac.setDispatchAsync(false);
|
||||
localConnection = fac.createConnection("userA", "passwordA");
|
||||
localConnection.setClientID("clientId");
|
||||
localConnection.start();
|
||||
URI remoteURI = remoteBroker.getVmConnectorURI();
|
||||
fac = new ActiveMQConnectionFactory(remoteURI);
|
||||
remoteConnection = fac.createConnection("userB", "passwordB");
|
||||
remoteConnection.setClientID("clientId");
|
||||
remoteConnection.start();
|
||||
localSession = localConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
|
||||
remoteSession = remoteConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected String getRemoteBrokerURI() {
|
||||
return "org/apache/activemq/network/remoteBroker-authn.xml";
|
||||
}
|
||||
|
||||
@Override
|
||||
protected String getLocalBrokerURI() {
|
||||
return "org/apache/activemq/network/localBroker-remote-userName.xml";
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testNetworkRemoteUserPassword() throws JMSException {
|
||||
|
||||
// Across the network
|
||||
sendTextMessage(localConnection, "include.test.foo", "This network message uses remoteUserName");
|
||||
verifyTextMessage(remoteConnection, "include.test.foo", "This network message uses remoteUserName", null, null, false);
|
||||
}
|
||||
|
||||
protected void sendTextMessage(Connection connection, String queueName, String textBody) throws JMSException {
|
||||
Session tmpSession = null;
|
||||
MessageProducer tmpProducer = null;
|
||||
try {
|
||||
tmpSession = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
|
||||
MessageProducer tmpLocalProducer = tmpSession.createProducer(tmpSession.createQueue(queueName));
|
||||
tmpLocalProducer.send(tmpSession.createTextMessage(textBody));
|
||||
} finally {
|
||||
if(tmpProducer != null) { tmpProducer.close(); }
|
||||
if(tmpSession != null) { tmpSession.close(); }
|
||||
}
|
||||
}
|
||||
|
||||
protected void verifyTextMessage(Connection connection, String queueName, String body, String property, String value, boolean propertyExists) throws JMSException {
|
||||
int maxLoops = 50;
|
||||
int loopCount = 0;
|
||||
boolean found = false;
|
||||
|
||||
Session tmpSession = null;
|
||||
MessageConsumer tmpConsumer = null;
|
||||
|
||||
try {
|
||||
tmpSession = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
|
||||
tmpConsumer = tmpSession.createConsumer(tmpSession.createQueue(queueName));
|
||||
do {
|
||||
Message tmpMessage = tmpConsumer.receive(100l);
|
||||
if(tmpMessage != null) {
|
||||
assert(TextMessage.class.isAssignableFrom(tmpMessage.getClass()));
|
||||
TextMessage tmpTextMessage = TextMessage.class.cast(tmpMessage);
|
||||
assertEquals(body, tmpTextMessage.getText());
|
||||
|
||||
if(propertyExists) {
|
||||
assertNotNull(tmpTextMessage.getStringProperty(property));
|
||||
assertEquals(value, tmpTextMessage.getStringProperty(property));
|
||||
} else if(property != null) {
|
||||
assertNull(tmpTextMessage.getStringProperty(property));
|
||||
}
|
||||
|
||||
found = true;
|
||||
}
|
||||
loopCount++;
|
||||
} while(!found && loopCount < maxLoops);
|
||||
assertTrue(found);
|
||||
} finally {
|
||||
if(tmpConsumer != null) { tmpConsumer.close(); }
|
||||
if(tmpSession != null) { tmpSession.close(); }
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,60 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!--
|
||||
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.
|
||||
-->
|
||||
<beans
|
||||
xmlns="http://www.springframework.org/schema/beans"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-2.0.xsd
|
||||
http://activemq.apache.org/schema/core http://activemq.apache.org/schema/core/activemq-core.xsd
|
||||
http://camel.apache.org/schema/spring http://camel.apache.org/schema/spring/camel-spring.xsd">
|
||||
<bean class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer"/>
|
||||
|
||||
<broker brokerName="localBroker" start="false" persistent="true" useShutdownHook="false" monitorConnectionSplits="true" xmlns="http://activemq.apache.org/schema/core">
|
||||
<networkConnectors>
|
||||
<networkConnector uri="static:(tcp://localhost:61617)"
|
||||
dynamicOnly = "false"
|
||||
conduitSubscriptions = "true"
|
||||
decreaseNetworkConsumerPriority = "false"
|
||||
name="networkConnector"
|
||||
userName="userA"
|
||||
password="passwordA"
|
||||
remoteUserName="userB"
|
||||
remotePassword="passwordB">
|
||||
<dynamicallyIncludedDestinations>
|
||||
<queue physicalName="include.test.foo"/>
|
||||
<topic physicalName="include.test.bar"/>
|
||||
</dynamicallyIncludedDestinations>
|
||||
<excludedDestinations>
|
||||
<queue physicalName="exclude.test.foo"/>
|
||||
<topic physicalName="exclude.test.bar"/>
|
||||
</excludedDestinations>
|
||||
</networkConnector>
|
||||
</networkConnectors>
|
||||
<plugins>
|
||||
<simpleAuthenticationPlugin>
|
||||
<users>
|
||||
<authenticationUser username="userA" password="passwordA" groups="groupA"/>
|
||||
<authenticationUser username="userB" password="passwordB" groups="groupB"/>
|
||||
</users>
|
||||
</simpleAuthenticationPlugin>
|
||||
</plugins>
|
||||
<transportConnectors>
|
||||
<transportConnector uri="tcp://localhost:61616"/>
|
||||
</transportConnectors>
|
||||
</broker>
|
||||
</beans>
|
||||
|
||||
Loading…
Reference in New Issue