1 22 package org.objectweb.petals.jbi.messaging.types; 23 24 import java.io.ByteArrayInputStream ; 25 26 import javax.jbi.messaging.NormalizedMessage; 27 import javax.jbi.messaging.MessageExchange.Role; 28 import javax.xml.transform.stream.StreamSource ; 29 30 import junit.framework.TestCase; 31 32 import org.objectweb.petals.jbi.messaging.MessageExchangeImpl; 33 import org.objectweb.petals.jbi.messaging.NormalizedMessageImpl; 34 35 40 public class InOnlyImplTest extends TestCase { 41 42 private InOnlyImpl inOnlyImpl; 43 44 private MessageExchangeImpl messageExchangeImpl; 45 46 private NormalizedMessageImpl normalizedMessageImpl; 47 48 public void setUp() throws Exception { 49 messageExchangeImpl = new MessageExchangeImpl(null); 50 51 normalizedMessageImpl = new NormalizedMessageImpl(); 52 53 StreamSource source = new StreamSource (); 54 byte[] msgByte = "test message".getBytes(); 55 ByteArrayInputStream in = new ByteArrayInputStream (msgByte); 56 source.setInputStream(in); 57 58 normalizedMessageImpl.setContent(source); 59 60 } 61 62 public void testGetInMessage() throws Exception { 63 normalizedMessageImpl.setProperty("test", "valuein"); 64 messageExchangeImpl.setPattern(MessageExchangeImpl.IN_OUT_PATTERN); 65 messageExchangeImpl.setMessage(normalizedMessageImpl, "in"); 66 inOnlyImpl = new InOnlyImpl(messageExchangeImpl, Role.CONSUMER); 67 68 NormalizedMessage normalizedMessage = inOnlyImpl.getInMessage(); 69 assertEquals(normalizedMessage.getProperty("test"), "valuein"); 70 } 71 72 public void testSetInMessage() throws Exception { 73 normalizedMessageImpl.setProperty("test", "valuein"); 74 messageExchangeImpl.setPattern(MessageExchangeImpl.IN_OUT_PATTERN); 75 messageExchangeImpl.setRole(Role.CONSUMER); 76 inOnlyImpl = new InOnlyImpl(messageExchangeImpl, Role.CONSUMER); 77 inOnlyImpl.setInMessage(normalizedMessageImpl); 78 79 NormalizedMessage normalizedMessage = inOnlyImpl.getInMessage(); 80 assertEquals(normalizedMessage.getProperty("test"), "valuein"); 81 } 82 83 } 84 | Popular Tags |