Author: clebert.suconic(a)jboss.com Date: 2009-11-29 22:31:27 -0500 (Sun, 29 Nov 2009) New Revision: 8456 Added: trunk/tests/src/org/hornetq/tests/stress/client/ trunk/tests/src/org/hornetq/tests/stress/client/SendStressTest.java Log: Adding test to replicate issue described on http://www.jboss.org/index.html?module=bb&op=viewtopic&t=164372 Added: trunk/tests/src/org/hornetq/tests/stress/client/SendStressTest.java =================================================================== --- trunk/tests/src/org/hornetq/tests/stress/client/SendStressTest.java (rev 0) +++ trunk/tests/src/org/hornetq/tests/stress/client/SendStressTest.java 2009-11-30 03:31:27 UTC (rev 8456) @@ -0,0 +1,150 @@ +/* + * Copyright 2009 Red Hat, Inc. + * Red Hat 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.hornetq.tests.stress.client; + +import junit.framework.TestSuite; + +import org.hornetq.core.client.ClientConsumer; +import org.hornetq.core.client.ClientMessage; +import org.hornetq.core.client.ClientProducer; +import org.hornetq.core.client.ClientSession; +import org.hornetq.core.client.ClientSessionFactory; +import org.hornetq.core.server.HornetQServer; +import org.hornetq.tests.util.ServiceTestBase; + +/** + * A SendStressTest + * + * @author <mailto:clebert.suconic@jboss.org">Clebert Suconic</a> + * + * + */ +public class SendStressTest extends ServiceTestBase +{ + + // Constants ----------------------------------------------------- + + // Attributes ---------------------------------------------------- + + // Static -------------------------------------------------------- + + // Constructors -------------------------------------------------- + + // Public -------------------------------------------------------- + + // Remove this method to re-enable those tests + public static TestSuite suite() + { + return new TestSuite(); + } + + public void testStressSendNetty() throws Exception + { + doTestStressSend(true); + } + + public void testStressSendInVM() throws Exception + { + doTestStressSend(false); + } + + public void doTestStressSend(final boolean netty) throws Exception + { + HornetQServer server = createServer(false, netty); + + ClientSessionFactory sf = createFactory(netty); + + ClientSession session = null; + + final int batchSize = 2000; + + final int numberOfMessages = 100000; + + try + { + server.start(); + + session = sf.createSession(false, false); + + session.createQueue("address", "queue"); + + ClientProducer producer = session.createProducer("address"); + + ClientMessage message = session.createClientMessage(false); + + message.getBodyBuffer().writeBytes(new byte[1024]); + + for (int i = 0; i < numberOfMessages; i++) + { + producer.send(message); + if (i % batchSize == 0) + { + System.out.println("Sent " + i); + session.commit(); + } + } + + session.commit(); + + session.close(); + + session = sf.createSession(false, false); + + ClientConsumer consumer = session.createConsumer("queue"); + + session.start(); + + for (int i = 0; i < numberOfMessages; i++) + { + ClientMessage msg = consumer.receive(5000); + assertNotNull(msg); + msg.acknowledge(); + + if (i % batchSize == 0) + { + System.out.println("Consumed " + i); + session.commit(); + } + } + + session.commit(); + } + finally + { + if (session != null) + { + try + { + sf.close(); + session.close(); + } + catch (Exception e) + { + e.printStackTrace(); + } + } + server.stop(); + } + + } + + // Package protected --------------------------------------------- + + // Protected ----------------------------------------------------- + + // Private ------------------------------------------------------- + + // Inner classes ------------------------------------------------- + +}