protected void preRouteMuleMessage(final DefaultMuleMessage message) throws Exception {
super.preRouteMuleMessage(message);
final SocketAddress clientAddress = socket.getRemoteSocketAddress();
if (clientAddress != null) {
message.setProperty(MuleProperties.MULE_REMOTE_CLIENT_ADDRESS, clientAddress.toString());
}
}
public void testDelete() throws Exception {
MessagePropertiesTransformer t = new MessagePropertiesTransformer();
t.setDeleteProperties(Collections.singletonList("badProperty"));
t.setMuleContext(muleContext);
DefaultMuleMessage msg = new DefaultMuleMessage("message", (Map) null);
msg.setProperty("badProperty", "badValue");
DefaultMuleMessage transformed = (DefaultMuleMessage) t.transform(msg, null);
assertSame(msg, transformed);
assertEquals(msg.getUniqueId(), transformed.getUniqueId());
assertEquals(msg.getPayload(), transformed.getPayload());
assertEquals(msg.getPropertyNames(), transformed.getPropertyNames());
assertFalse(transformed.getPropertyNames().contains("badValue"));
}
public void testRenameProperties() throws Exception {
MessagePropertiesTransformer t = new MessagePropertiesTransformer();
Map add = new HashMap();
add.put("Foo", "Baz");
t.setRenameProperties(add);
t.setMuleContext(muleContext);
DefaultMuleMessage msg = new DefaultMuleMessage("message", (Map) null);
msg.setProperty("Foo", "Bar");
DefaultMuleMessage transformed = (DefaultMuleMessage) t.transform(msg, null);
assertSame(msg, transformed);
assertEquals(msg.getUniqueId(), transformed.getUniqueId());
assertEquals(msg.getPayload(), transformed.getPayload());
assertEquals(msg.getPropertyNames(), transformed.getPropertyNames());
assertEquals("Bar", transformed.getProperty("Baz"));
}
public void testOverwriteFalsePreservesOriginal() throws Exception {
MessagePropertiesTransformer t = new MessagePropertiesTransformer();
Map add = new HashMap();
add.put("addedProperty", "overwrittenValue");
t.setAddProperties(add);
t.setOverwrite(false);
t.setMuleContext(muleContext);
DefaultMuleMessage msg = new DefaultMuleMessage("message", (Map) null);
msg.setProperty("addedProperty", "originalValue");
DefaultMuleMessage transformed = (DefaultMuleMessage) t.transform(msg, null);
assertSame(msg, transformed);
assertEquals(msg.getUniqueId(), transformed.getUniqueId());
assertEquals(msg.getPayload(), transformed.getPayload());
assertEquals(msg.getPropertyNames(), transformed.getPropertyNames());
assertEquals("originalValue", transformed.getProperty("addedProperty"));
}