mirror of https://github.com/apache/activemq.git
Adds a STOMP based test to cover some inactivity monitor usage over all
four transports.
This commit is contained in:
parent
f37b005acf
commit
2d7280f33a
|
@ -0,0 +1,145 @@
|
|||
/**
|
||||
* 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.transport.stomp;
|
||||
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.Socket;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collection;
|
||||
|
||||
import javax.net.SocketFactory;
|
||||
import javax.net.ssl.SSLSocketFactory;
|
||||
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.junit.runners.Parameterized;
|
||||
import org.junit.runners.Parameterized.Parameters;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
/**
|
||||
* Test that the inactivity monitor works as expected.
|
||||
*/
|
||||
@RunWith(Parameterized.class)
|
||||
public class StompInactivityMonitorTest extends StompTestSupport {
|
||||
|
||||
private static final Logger LOG = LoggerFactory.getLogger(StompInactivityMonitorTest.class);
|
||||
|
||||
private final String transportScheme;
|
||||
|
||||
@Parameters(name="{0}")
|
||||
public static Collection<Object[]> data() {
|
||||
return Arrays.asList(new Object[][] {
|
||||
{ "stomp" },
|
||||
{ "stomp+ssl" },
|
||||
{ "stomp+nio" },
|
||||
{ "stomp+nio+ssl" }
|
||||
});
|
||||
}
|
||||
|
||||
public StompInactivityMonitorTest(String transportScheme) {
|
||||
this.transportScheme = transportScheme;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void test() throws Exception {
|
||||
stompConnect();
|
||||
|
||||
String connectFrame = "STOMP\n" +
|
||||
"login:system\n" +
|
||||
"passcode:manager\n" +
|
||||
"accept-version:1.1\n" +
|
||||
"heart-beat:1000,0\n" +
|
||||
"host:localhost\n" +
|
||||
"\n" + Stomp.NULL;
|
||||
|
||||
stompConnection.sendFrame(connectFrame);
|
||||
String response = stompConnection.receiveFrame().trim();
|
||||
LOG.info("Broker sent response: {}", response);
|
||||
|
||||
String messageHead = "SEND\n" +
|
||||
"receipt:1" + "\n" +
|
||||
"destination:/queue/" + getQueueName() +
|
||||
"\n\n" + "AAAA";
|
||||
|
||||
stompConnection.sendFrame(messageHead);
|
||||
|
||||
for (int i = 0; i < 30; i++) {
|
||||
stompConnection.sendFrame("A");
|
||||
Thread.sleep(100);
|
||||
}
|
||||
|
||||
stompConnection.sendFrame(Stomp.NULL);
|
||||
|
||||
response = stompConnection.receiveFrame().trim();
|
||||
assertTrue(response.startsWith("RECEIPT"));
|
||||
}
|
||||
|
||||
@Override
|
||||
protected boolean isUseTcpConnector() {
|
||||
return !transportScheme.contains("nio") && !transportScheme.contains("ssl");
|
||||
}
|
||||
|
||||
@Override
|
||||
protected boolean isUseSslConnector() {
|
||||
return !transportScheme.contains("nio") && transportScheme.contains("ssl");
|
||||
}
|
||||
|
||||
@Override
|
||||
protected boolean isUseNioConnector() {
|
||||
return transportScheme.contains("nio") && !transportScheme.contains("ssl");
|
||||
}
|
||||
|
||||
@Override
|
||||
protected boolean isUseNioPlusSslConnector() {
|
||||
return transportScheme.contains("nio") && transportScheme.contains("ssl");
|
||||
}
|
||||
|
||||
@Override
|
||||
protected Socket createSocket() throws IOException {
|
||||
int port = 0;
|
||||
boolean useSSL = false;
|
||||
|
||||
if (transportScheme.contains("ssl")) {
|
||||
if (transportScheme.contains("nio")) {
|
||||
port = this.nioSslPort;
|
||||
} else {
|
||||
port = this.sslPort;
|
||||
}
|
||||
|
||||
useSSL = true;
|
||||
} else {
|
||||
if (transportScheme.contains("nio")) {
|
||||
port = this.nioPort;
|
||||
} else {
|
||||
port = this.port;
|
||||
}
|
||||
}
|
||||
|
||||
SocketFactory factory = null;
|
||||
|
||||
if (useSSL) {
|
||||
factory = SSLSocketFactory.getDefault();
|
||||
} else {
|
||||
factory = SocketFactory.getDefault();
|
||||
}
|
||||
|
||||
return factory.createSocket("127.0.0.1", port);
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue