@Test
public void testHandleMessage() {
StaticResponseEndpointAdapter endpointAdapter = new StaticResponseEndpointAdapter();
Map<String, Object> header = new HashMap<String, Object>();
header.put("Operation", "UnitTest");
endpointAdapter.setMessageHeader(header);
endpointAdapter.setMessagePayload("<TestMessage>Hello User!</TestMessage>");
Message response =
endpointAdapter.handleMessage(
new DefaultMessage("<TestMessage>Hello World!</TestMessage>"));
Assert.assertEquals(response.getPayload(), "<TestMessage>Hello User!</TestMessage>");
Assert.assertNotNull(response.getHeader("Operation"));
Assert.assertEquals(response.getHeader("Operation"), "UnitTest");
}
@Test
public void testSendBuilderWithObjectMessageInstance() {
final Message message = new DefaultMessage(new Integer(10)).setHeader("operation", "foo");
MockTestDesigner builder =
new MockTestDesigner(applicationContext) {
@Override
public void configure() {
send(messageEndpoint).message(message);
}
};
builder.configure();
TestCase test = builder.getTestCase();
Assert.assertEquals(test.getActionCount(), 1);
Assert.assertEquals(test.getActions().get(0).getClass(), SendMessageAction.class);
SendMessageAction action = ((SendMessageAction) test.getActions().get(0));
Assert.assertEquals(action.getName(), "send");
Assert.assertEquals(action.getEndpoint(), messageEndpoint);
Assert.assertEquals(action.getMessageBuilder().getClass(), StaticMessageContentBuilder.class);
StaticMessageContentBuilder messageBuilder =
(StaticMessageContentBuilder) action.getMessageBuilder();
Assert.assertEquals(messageBuilder.getMessage().getPayload(), 10);
Assert.assertEquals(messageBuilder.getMessageHeaders().size(), 0L);
Assert.assertEquals(
messageBuilder.getMessage().copyHeaders().size(), message.copyHeaders().size());
Assert.assertEquals(
messageBuilder.getMessage().getHeader(MessageHeaders.ID),
message.getHeader(MessageHeaders.ID));
Assert.assertEquals(messageBuilder.getMessage().getHeader("operation"), "foo");
Message constructed =
messageBuilder.buildMessageContent(new TestContext(), MessageType.PLAINTEXT.name());
Assert.assertEquals(constructed.copyHeaders().size(), message.copyHeaders().size());
Assert.assertEquals(constructed.getHeader("operation"), "foo");
Assert.assertEquals(
constructed.getHeader(MessageHeaders.ID), message.getHeader(MessageHeaders.ID));
}