Source: MessageHeaderTest.java 
/*
* JBoss, Home of Professional Open Source
* Copyright 2005-2008, Red Hat Middleware LLC, and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.test.messaging.jms.message;
import static org.easymock.EasyMock.expect;
import java.io.Serializable;
import java.nio.ByteBuffer;
import java.util.Enumeration;
import java.util.HashSet;
import javax.jms.BytesMessage;
import javax.jms.DeliveryMode;
import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.MapMessage;
import javax.jms.Message;
import javax.jms.MessageFormatException;
import javax.jms.MessageNotWriteableException;
import javax.jms.ObjectMessage;
import javax.jms.StreamMessage;
import javax.jms.TextMessage;
import org.easymock.EasyMock;
import org.jboss.messaging.core.client.ClientMessage;
import org.jboss.messaging.core.client.ClientSession;
import org.jboss.messaging.core.client.impl.ClientMessageImpl;
import org.jboss.messaging.core.remoting.impl.ByteBufferWrapper;
import org.jboss.messaging.jms.client.JBossBytesMessage;
import org.jboss.messaging.jms.client.JBossMapMessage;
import org.jboss.messaging.jms.client.JBossMessage;
import org.jboss.messaging.jms.client.JBossObjectMessage;
import org.jboss.messaging.jms.client.JBossStreamMessage;
import org.jboss.messaging.jms.client.JBossTextMessage;
/**
*
* @author <a href="mailto:ovidiu@feodorov.com">Ovidiu Feodorov</a>
* @author <a href="mailto:tim.fox@jboss.com">Tim Fox</a>
* @version <tt>$Revision$</tt>
*
* $Id$
*/
public class MessageHeaderTest extends MessageHeaderTestBase
{
// Constants -----------------------------------------------------
// Static --------------------------------------------------------
// Attributes ----------------------------------------------------
// Constructors --------------------------------------------------
public MessageHeaderTest(String name)
{
super(name);
}
// Public --------------------------------------------------------
public void testClearMessage() throws Exception
{
queueProducer.setDeliveryMode(DeliveryMode.NON_PERSISTENT);
Message message = queueProducerSession.createTextMessage("some message");
queueProducer.send(message);
message = queueConsumer.receive(1000);
assertNotNull(message);
message.clearProperties();
assertNotNull(message.getJMSDestination());
}
public void testMessageOrderQueue() throws Exception
{
final int NUM_MESSAGES = 10;
queueProducer.setDeliveryMode(DeliveryMode.NON_PERSISTENT);
for (int i = 0; i < NUM_MESSAGES; i++)
{
Message m = queueProducerSession.createMessage();
m.setIntProperty("count", i);
queueProducer.send(m);
}
for (int i = 0; i < NUM_MESSAGES; i++)
{
Message m = queueConsumer.receive(3000);
assertNotNull(m);
int count = m.getIntProperty("count");
assertEquals(i, count);
}
queueProducer.setDeliveryMode(DeliveryMode.PERSISTENT);
for (int i = 0; i < NUM_MESSAGES; i++)
{
Message m = queueProducerSession.createMessage();
m.setIntProperty("count2", i);
queueProducer.send(m);
}
for (int i = 0; i < NUM_MESSAGES; i++)
{
Message m = queueConsumer.receive(3000);
assertNotNull(m);
int count = m.getIntProperty("count2");
assertEquals(i, count);
}
}
public void testMessageOrderTopic() throws Exception
{
final int NUM_MESSAGES = 10;
topicProducer.setDeliveryMode(DeliveryMode.NON_PERSISTENT);
for (int i = 0; i < NUM_MESSAGES; i++)
{
Message m = topicProducerSession.createMessage();
m.setIntProperty("count", i);
topicProducer.send(m);
}
for (int i = 0; i < NUM_MESSAGES; i++)
{
Message m = topicConsumer.receive(3000);
assertNotNull(m);
int count = m.getIntProperty("count");
assertEquals(i, count);
}
topicProducer.setDeliveryMode(DeliveryMode.PERSISTENT);
for (int i = 0; i < NUM_MESSAGES; i++)
{
Message m = topicProducerSession.createMessage();
m.setIntProperty("count2", i);
topicProducer.send(m);
}
for (int i = 0; i < NUM_MESSAGES; i++)
{
Message m = topicConsumer.receive(3000);
assertNotNull(m);
int count = m.getIntProperty("count2");
assertEquals(i, count);
}
}
public void testProperties() throws Exception
{
Message m1 = queueProducerSession.createMessage();
//Some arbitrary values
boolean myBool = true;
byte myByte = 13;
short myShort = 15321;
int myInt = 0x71ab6c80;
long myLong = 0x20bf1e3fb6fa31dfL;
float myFloat = Float.MAX_VALUE - 23465;
double myDouble = Double.MAX_VALUE - 72387633;
String myString = "abcdef&^*&!^ghijkl";
m1.setBooleanProperty("myBool", myBool);
m1.setByteProperty("myByte", myByte);
m1.setShortProperty("myShort", myShort);
m1.setIntProperty("myInt", myInt);
m1.setLongProperty("myLong", myLong);
m1.setFloatProperty("myFloat", myFloat);
m1.setDoubleProperty("myDouble", myDouble);
m1.setStringProperty("myString", myString);
m1.setObjectProperty("myBool", new Boolean(myBool));
m1.setObjectProperty("myByte", new Byte(myByte));
m1.setObjectProperty("myShort", new Short(myShort));
m1.setObjectProperty("myInt", new Integer(myInt));
m1.setObjectProperty("myLong", new Long(myLong));
m1.setObjectProperty("myFloat", new Float(myFloat));
m1.setObjectProperty("myDouble", new Double(myDouble));
m1.setObjectProperty("myString", myString);
try
{
m1.setObjectProperty("myIllegal", new Object());
fail();
}
catch (javax.jms.MessageFormatException e)
{}
queueProducer.send(queue1, m1);
Message m2 = queueConsumer.receive(2000);
assertNotNull(m2);
assertEquals(myBool, m2.getBooleanProperty("myBool"));
assertEquals(myByte, m2.getByteProperty("myByte"));
assertEquals(myShort, m2.getShortProperty("myShort"));
assertEquals(myInt, m2.getIntProperty("myInt"));
assertEquals(myLong, m2.getLongProperty("myLong"));
assertEquals(myFloat, m2.getFloatProperty("myFloat"), 0);
assertEquals(myDouble, m2.getDoubleProperty("myDouble"), 0);
assertEquals(myString, m2.getStringProperty("myString"));
//Properties should now be read-only
try
{
m2.setBooleanProperty("myBool", myBool);
fail();
}
catch (MessageNotWriteableException e) {}
try
{
m2.setByteProperty("myByte", myByte);
fail();
}
catch (MessageNotWriteableException e) {}
try
{
m2.setShortProperty("myShort", myShort);
fail();
}
catch (MessageNotWriteableException e) {}
try
{
m2.setIntProperty("myInt", myInt);
fail();
}
catch (MessageNotWriteableException e) {}
try
{
m2.setLongProperty("myLong", myLong);
fail();
}
catch (MessageNotWriteableException e) {}
try
{
m2.setFloatProperty("myFloat", myFloat);
fail();
}
catch (MessageNotWriteableException e) {}
try
{
m2.setDoubleProperty("myDouble", myDouble);
fail();
}
catch (MessageNotWriteableException e) {}
try
{
m2.setStringProperty("myString", myString);
fail();
}
catch (MessageNotWriteableException e) {}
assertTrue(m2.propertyExists("myBool"));
assertTrue(m2.propertyExists("myByte"));
assertTrue(m2.propertyExists("myShort"));
assertTrue(m2.propertyExists("myInt"));
assertTrue(m2.propertyExists("myLong"));
assertTrue(m2.propertyExists("myFloat"));
assertTrue(m2.propertyExists("myDouble"));
assertTrue(m2.propertyExists("myString"));
assertFalse(m2.propertyExists("sausages"));
HashSet propNames = new HashSet();
Enumeration en = m2.getPropertyNames();
while (en.hasMoreElements())
{
String propName = (String)en.nextElement();
log.info("Prop name is " + propName);
propNames.add(propName);
}
assertEquals(9, propNames.size());
assertTrue(propNames.contains("myBool"));
assertTrue(propNames.contains("myByte"));
assertTrue(propNames.contains("myShort"));
assertTrue(propNames.contains("myInt"));
assertTrue(propNames.contains("myLong"));
assertTrue(propNames.contains("myFloat"));
assertTrue(propNames.contains("myDouble"));
assertTrue(propNames.contains("myString"));
// Check property conversions
//Boolean property can be read as String but not anything else
assertEquals(String.valueOf(myBool), m2.getStringProperty("myBool"));
try
{
m2.getByteProperty("myBool");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getShortProperty("myBool");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getIntProperty("myBool");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getLongProperty("myBool");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getFloatProperty("myBool");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getDoubleProperty("myBool");
fail();
} catch (MessageFormatException e) {}
// byte property can be read as short, int, long or String
assertEquals((short)myByte, m2.getShortProperty("myByte"));
assertEquals((int)myByte, m2.getIntProperty("myByte"));
assertEquals((long)myByte, m2.getLongProperty("myByte"));
assertEquals(String.valueOf(myByte), m2.getStringProperty("myByte"));
try
{
m2.getBooleanProperty("myByte");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getFloatProperty("myByte");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getDoubleProperty("myByte");
fail();
} catch (MessageFormatException e) {}
// short property can be read as int, long or String
assertEquals((int)myShort, m2.getIntProperty("myShort"));
assertEquals((long)myShort, m2.getLongProperty("myShort"));
assertEquals(String.valueOf(myShort), m2.getStringProperty("myShort"));
try
{
m2.getByteProperty("myShort");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getBooleanProperty("myShort");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getFloatProperty("myShort");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getDoubleProperty("myShort");
fail();
} catch (MessageFormatException e) {}
// int property can be read as long or String
assertEquals((long)myInt, m2.getLongProperty("myInt"));
assertEquals(String.valueOf(myInt), m2.getStringProperty("myInt"));
try
{
m2.getShortProperty("myInt");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getByteProperty("myInt");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getBooleanProperty("myInt");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getFloatProperty("myInt");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getDoubleProperty("myInt");
fail();
} catch (MessageFormatException e) {}
// long property can be read as String
assertEquals(String.valueOf(myLong), m2.getStringProperty("myLong"));
try
{
m2.getIntProperty("myLong");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getShortProperty("myLong");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getByteProperty("myLong");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getBooleanProperty("myLong");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getFloatProperty("myLong");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getDoubleProperty("myLong");
fail();
} catch (MessageFormatException e) {}
// float property can be read as double or String
assertEquals(String.valueOf(myFloat), m2.getStringProperty("myFloat"));
assertEquals((double)myFloat, m2.getDoubleProperty("myFloat"), 0);
try
{
m2.getIntProperty("myFloat");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getShortProperty("myFloat");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getLongProperty("myFloat");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getByteProperty("myFloat");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getBooleanProperty("myFloat");
fail();
} catch (MessageFormatException e) {}
// double property can be read as String
assertEquals(String.valueOf(myDouble), m2.getStringProperty("myDouble"));
try
{
m2.getFloatProperty("myDouble");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getIntProperty("myDouble");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getShortProperty("myDouble");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getByteProperty("myDouble");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getBooleanProperty("myDouble");
fail();
} catch (MessageFormatException e) {}
try
{
m2.getFloatProperty("myDouble");
fail();
} catch (MessageFormatException e) {}
log.info("** clearing props");
m2.clearProperties();
log.info("** getting prop names");
Enumeration en2 = m2.getPropertyNames();
log.info("** got prop names");
assertTrue(en2.hasMoreElements());
en2.nextElement();
assertFalse(en2.hasMoreElements());
// Test String -> Numeric and bool conversions
Message m3 = queueProducerSession.createMessage();
m3.setStringProperty("myBool", String.valueOf(myBool));
m3.setStringProperty("myByte", String.valueOf(myByte));
m3.setStringProperty("myShort", String.valueOf(myShort));
m3.setStringProperty("myInt", String.valueOf(myInt));
m3.setStringProperty("myLong", String.valueOf(myLong));
m3.setStringProperty("myFloat", String.valueOf(myFloat));
m3.setStringProperty("myDouble", String.valueOf(myDouble));
m3.setStringProperty("myIllegal", "xyz123");
assertEquals(myBool, m3.getBooleanProperty("myBool"));
assertEquals(myByte, m3.getByteProperty("myByte"));
assertEquals(myShort, m3.getShortProperty("myShort"));
assertEquals(myInt, m3.getIntProperty("myInt"));
assertEquals(myLong, m3.getLongProperty("myLong"));
assertEquals(myFloat, m3.getFloatProperty("myFloat"), 0);
assertEquals(myDouble, m3.getDoubleProperty("myDouble"), 0);
m3.getBooleanProperty("myIllegal");
try
{
m3.getByteProperty("myIllegal");
fail();
}
catch (NumberFormatException e) {}
try
{
m3.getShortProperty("myIllegal");
fail();
}
catch (NumberFormatException e) {}
try
{
m3.getIntProperty("myIllegal");
fail();
}
catch (NumberFormatException e) {}
try
{
m3.getLongProperty("myIllegal");
fail();
}
catch (NumberFormatException e) {}
try
{
m3.getFloatProperty("myIllegal");
fail();
}
catch (NumberFormatException e) {}
try
{
m3.getDoubleProperty("myIllegal");
fail();
}
catch (NumberFormatException e) {}
}
public void testSendReceiveForeignMessage() throws JMSException
{
log.trace("Starting da test");
SimpleJMSMessage foreignMessage = new SimpleJMSMessage();
foreignMessage.setStringProperty("animal", "aardvark");
//foreign messages don't have to be serializable
assertFalse(foreignMessage instanceof Serializable);
log.trace("Sending message");
queueProducer.send(foreignMessage);
log.trace("Sent message");
Message m2 = queueConsumer.receive(3000);
log.trace("The message is " + m2);
assertNotNull(m2);
assertEquals("aardvark", m2.getStringProperty("animal"));
log.trace("Received message");
log.trace("Done that test");
}
public void testCopyOnJBossMessage() throws JMSException
{
ClientSession session = EasyMock.createNiceMock(ClientSession.class);
ByteBufferWrapper body = new ByteBufferWrapper(ByteBuffer.allocate(1024));
ClientMessage clientMessage = new ClientMessageImpl(JBossTextMessage.TYPE, true, 0, System.currentTimeMillis(), (byte)4, body);
expect(session.createClientMessage(EasyMock.anyByte(), EasyMock.anyBoolean(), EasyMock.anyInt(), EasyMock.anyLong(), EasyMock.anyByte())).andReturn(clientMessage);
EasyMock.replay(session);
JBossMessage jbossMessage = new JBossMessage();
configureMessage(jbossMessage);
JBossMessage copy = new JBossMessage(jbossMessage, session);
ensureEquivalent(jbossMessage, copy);
EasyMock.verify(session);
}
public void testCopyOnForeignMessage() throws JMSException
{
ClientSession session = EasyMock.createNiceMock(ClientSession.class);
ByteBufferWrapper body = new ByteBufferWrapper(ByteBuffer.allocate(1024));
ClientMessage clientMessage = new ClientMessageImpl(JBossMessage.TYPE, true, 0, System.currentTimeMillis(), (byte)4, body);
expect(session.createClientMessage(EasyMock.anyByte(), EasyMock.anyBoolean(), EasyMock.anyInt(), EasyMock.anyLong(), EasyMock.anyByte())).andReturn(clientMessage);
EasyMock.replay(session);
Message foreignMessage = new SimpleJMSMessage();
JBossMessage copy = new JBossMessage(foreignMessage, session);
ensureEquivalent(foreignMessage, copy);
EasyMock.verify(session);
}
public void testCopyOnForeignBytesMessage() throws JMSException
{
ClientSession session = EasyMock.createNiceMock(ClientSession.class);
ByteBufferWrapper body = new ByteBufferWrapper(ByteBuffer.allocate(1024));
ClientMessage clientMessage = new ClientMessageImpl(JBossBytesMessage.TYPE, true, 0, System.currentTimeMillis(), (byte)4, body);
expect(session.createClientMessage(EasyMock.anyByte(), EasyMock.anyBoolean(), EasyMock.anyInt(), EasyMock.anyLong(), EasyMock.anyByte())).andReturn(clientMessage);
EasyMock.replay(session);
BytesMessage foreignBytesMessage = new SimpleJMSBytesMessage();
for(int i = 0; i < 20; i++)
{
foreignBytesMessage.writeByte((byte)i);
}
JBossBytesMessage copy = new JBossBytesMessage(foreignBytesMessage, session);
foreignBytesMessage.reset();
copy.reset();
ensureEquivalent(foreignBytesMessage, copy);
EasyMock.verify(session);
}
public void testCopyOnForeignMapMessage() throws JMSException
{
ClientSession session = EasyMock.createNiceMock(ClientSession.class);
ByteBufferWrapper body = new ByteBufferWrapper(ByteBuffer.allocate(1024));
ClientMessage clientMessage = new ClientMessageImpl(JBossMapMessage.TYPE, true, 0, System.currentTimeMillis(), (byte)4, body);
expect(session.createClientMessage(EasyMock.anyByte(), EasyMock.anyBoolean(), EasyMock.anyInt(), EasyMock.anyLong(), EasyMock.anyByte())).andReturn(clientMessage);
EasyMock.replay(session);
MapMessage foreignMapMessage = new SimpleJMSMapMessage();
foreignMapMessage.setInt("int", 1);
foreignMapMessage.setString("string", "test");
JBossMapMessage copy = new JBossMapMessage(foreignMapMessage, session);
ensureEquivalent(foreignMapMessage, copy);
EasyMock.verify(session);
}
public void testCopyOnForeignObjectMessage() throws JMSException
{
ClientSession session = EasyMock.createNiceMock(ClientSession.class);
ByteBufferWrapper body = new ByteBufferWrapper(ByteBuffer.allocate(1024));
ClientMessage clientMessage = new ClientMessageImpl(JBossObjectMessage.TYPE, true, 0, System.currentTimeMillis(), (byte)4, body);
expect(session.createClientMessage(EasyMock.anyByte(), EasyMock.anyBoolean(), EasyMock.anyInt(), EasyMock.anyLong(), EasyMock.anyByte())).andReturn(clientMessage);
EasyMock.replay(session);
ObjectMessage foreignObjectMessage = new SimpleJMSObjectMessage();
JBossObjectMessage copy = new JBossObjectMessage(foreignObjectMessage, session);
ensureEquivalent(foreignObjectMessage, copy);
}
public void testCopyOnForeignStreamMessage() throws JMSException
{
ClientSession session = EasyMock.createNiceMock(ClientSession.class);
ByteBufferWrapper body = new ByteBufferWrapper(ByteBuffer.allocate(1024));
ClientMessage clientMessage = new ClientMessageImpl(JBossStreamMessage.TYPE, true, 0, System.currentTimeMillis(), (byte)4, body);
expect(session.createClientMessage(EasyMock.anyByte(), EasyMock.anyBoolean(), EasyMock.anyInt(), EasyMock.anyLong(), EasyMock.anyByte())).andReturn(clientMessage);
EasyMock.replay(session);
StreamMessage foreignStreamMessage = new SimpleJMSStreamMessage();
foreignStreamMessage.writeByte((byte)1);
foreignStreamMessage.writeByte((byte)2);
foreignStreamMessage.writeByte((byte)3);
JBossStreamMessage copy = new JBossStreamMessage(foreignStreamMessage, session);
ensureEquivalent(foreignStreamMessage, copy);
EasyMock.verify(session);
}
public void testCopyOnForeignTextMessage() throws JMSException
{
ClientSession session = EasyMock.createNiceMock(ClientSession.class);
ByteBufferWrapper body = new ByteBufferWrapper(ByteBuffer.allocate(1024));
ClientMessage clientMessage = new ClientMessageImpl(JBossTextMessage.TYPE, true, 0, System.currentTimeMillis(), (byte)4, body);
expect(session.createClientMessage(EasyMock.anyByte(), EasyMock.anyBoolean(), EasyMock.anyInt(), EasyMock.anyLong(), EasyMock.anyByte())).andReturn(clientMessage);
EasyMock.replay(session);
TextMessage foreignTextMessage = new SimpleJMSTextMessage();
JBossTextMessage copy = new JBossTextMessage(foreignTextMessage, session);
ensureEquivalent(foreignTextMessage, copy);
EasyMock.verify(session);
}
public void testForeignJMSDestination() throws JMSException
{
Message message = queueProducerSession.createMessage();
Destination foreignDestination = new ForeignDestination();
message.setJMSDestination(foreignDestination);
assertSame(foreignDestination, message.getJMSDestination());
queueProducer.send(message);
assertSame(queue1, message.getJMSDestination());
Message receivedMessage = queueConsumer.receive(2000);
ensureEquivalent(receivedMessage, (JBossMessage)message);
}
// Invalid!!
// public void testForeignJMSReplyTo() throws JMSException
// {
// Message msg = queueProducerSession.createTextMessage();
//
// JBossMessage jbossMessage = (JBossMessage)msg;
//
// Destination foreignDestination = new ForeignDestination();
//
// jbossMessage.setJMSReplyTo(foreignDestination);
//
// queueProducer.send(msg);
//
// Message receivedMessage = queueConsumer.receive(2000);
//
// ensureEquivalent(receivedMessage, jbossMessage);
// }
//
// public void testCopyForeignDestinationAndReplyTo() throws JMSException
// {
// Message foreignMessage = new SimpleJMSMessage();
// foreignMessage.setJMSDestination(new ForeignDestination());
// foreignMessage.setJMSReplyTo(new ForeignDestination());
//
// JBossMessage copy = new JBossMessage(foreignMessage);
//
// ensureEquivalent(foreignMessage, copy);
// }
// Package protected ---------------------------------------------
// Protected -----------------------------------------------------
// Private -------------------------------------------------------
// Inner classes -------------------------------------------------
private static class ForeignDestination implements Destination, Serializable
{
private static final long serialVersionUID = 5545509674580823610L;
// A ForeignDestination equals any other ForeignDestination, for simplicity
public boolean equals(Object obj)
{
return obj instanceof ForeignDestination;
}
public int hashCode()
{
return 157;
}
}
}
