1 files changed, 119 insertions, 0 deletions
diff --git a/qpid/java/qpid-perftests-systests/src/main/java/org/apache/qpid/systest/disttest/clientonly/MessageProviderTest.java b/qpid/java/qpid-perftests-systests/src/main/java/org/apache/qpid/systest/disttest/clientonly/MessageProviderTest.java
new file mode 100644
index 0000000000..dcbff6518b
--- /dev/null
+++ b/qpid/java/qpid-perftests-systests/src/main/java/org/apache/qpid/systest/disttest/clientonly/MessageProviderTest.java
@@ -0,0 +1,119 @@
+/*
+ * 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.systest.disttest.clientonly;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import javax.jms.DeliveryMode;
+import javax.jms.JMSException;
+import javax.jms.Message;
+import javax.jms.MessageConsumer;
+import javax.jms.Session;
+import javax.jms.TextMessage;
+
+import org.apache.qpid.disttest.client.MessageProvider;
+import org.apache.qpid.disttest.client.property.PropertyValue;
+import org.apache.qpid.disttest.client.property.SimplePropertyValue;
+import org.apache.qpid.disttest.message.CreateMessageProviderCommand;
+import org.apache.qpid.disttest.message.CreateProducerCommand;
+import org.apache.qpid.systest.disttest.DistributedTestSystemTestBase;
+import org.apache.qpid.systest.disttest.clientonly.ProducerParticipantTest.TestClientJmsDelegate;
+
+public class MessageProviderTest extends DistributedTestSystemTestBase
+{
+    private MessageConsumer _consumer;
+    private Session _session;
+    private TestClientJmsDelegate _delegate;
+
+    @Override
+    protected void setUp() throws Exception
+    {
+        super.setUp();
+        _session = _connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
+        _consumer = _session.createConsumer(getTestQueue());
+        _delegate = new TestClientJmsDelegate(getContext());
+    }
+
+    public void testMessageSize() throws Exception
+    {
+        runSizeTest(0);
+        runSizeTest(5);
+        runSizeTest(512);
+    }
+
+    public void runSizeTest(int size) throws Exception
+    {
+        CreateProducerCommand command = new CreateProducerCommand();
+        command.setMessageSize(size);
+        MessageProvider messageProvider = new MessageProvider(null);
+        Message message = messageProvider.nextMessage(_session, command);
+        assertNotNull("Message is not generated", message);
+        assertTrue("Wrong message type", message instanceof TextMessage);
+        TextMessage textMessage = (TextMessage)message;
+        String text = textMessage.getText();
+        assertNotNull("Message payload is not generated", text);
+        assertEquals("Message payload size is incorrect", size, text.length());
+    }
+
+    public void testCreateMessageProviderAndSendMessage() throws Exception
+    {
+        final CreateMessageProviderCommand messageProviderCommand = new CreateMessageProviderCommand();
+        messageProviderCommand.setProviderName("test1");
+        Map<String, PropertyValue> messageProperties = new HashMap<String, PropertyValue>();
+        messageProperties.put("test", new SimplePropertyValue("testValue"));
+        messageProperties.put("priority", new SimplePropertyValue(new Integer(9)));
+        messageProviderCommand.setMessageProperties(messageProperties);
+        _delegate.createMessageProvider(messageProviderCommand);
+
+        final CreateProducerCommand producerCommand = new CreateProducerCommand();
+        producerCommand.setNumberOfMessages(1);
+        producerCommand.setDeliveryMode(DeliveryMode.PERSISTENT);
+        producerCommand.setPriority(6);
+        producerCommand.setParticipantName("test");
+        producerCommand.setMessageSize(10);
+        producerCommand.setSessionName("testSession");
+        producerCommand.setDestinationName(getTestQueueName());
+        producerCommand.setMessageProviderName(messageProviderCommand.getProviderName());
+
+        Session session = _connection.createSession(true, Session.SESSION_TRANSACTED);
+        _delegate.addConnection("name-does-not-matter", _connection);
+        _delegate.addSession(producerCommand.getSessionName(), session);
+        _delegate.createProducer(producerCommand);
+
+        Message message = _delegate.sendNextMessage(producerCommand);
+        session.commit();
+        assertMessage(message);
+
+        _connection.start();
+        Message receivedMessage = _consumer.receive(1000l);
+        assertMessage(receivedMessage);
+     }
+
+    protected void assertMessage(Message message) throws JMSException
+    {
+        assertNotNull("Message should not be null", message);
+        assertEquals("Unexpected test property", "testValue", message.getStringProperty("test"));
+        assertEquals("Unexpected priority property", 9, message.getJMSPriority());
+        assertTrue("Unexpected message type", message instanceof TextMessage);
+        String text = ((TextMessage)message).getText();
+        assertNotNull("Message text should not be null", text);
+        assertNotNull("Unexpected message size ", text.length());
+    }
+}