@Test
public void shouldWriteWriteTextMessage() throws Exception {
final WriteRequestEx writeRequest = context.mock(WriteRequestEx.class);
final WsTextMessage message = new WsTextMessage(BUFFER_ALLOCATOR.wrap(BYTES));
context.checking(
new Expectations() {
{
allowing(writeRequest).getMessage();
will(returnValue(message));
oneOf(nextFilter).filterWrite(session, writeRequest);
}
});
filter.filterWrite(nextFilter, session, writeRequest);
context.assertIsSatisfied();
}
@Test
public void shouldWriteEscapeTextMessageConsistingOfControlBytes() throws Exception {
final WriteRequestEx writeRequest = context.mock(WriteRequestEx.class);
final WsTextMessage message =
new WsTextMessage(BUFFER_ALLOCATOR.wrap(ByteBuffer.wrap(CONTROL_BYTES)));
context.checking(
new Expectations() {
{
allowing(writeRequest).getMessage();
will(returnValue(message));
oneOf(nextFilter).filterWrite(with(session), with(hasMessage(ESCAPE_MESSAGE)));
oneOf(nextFilter).filterWrite(session, writeRequest);
}
});
filter.filterWrite(nextFilter, session, writeRequest);
context.assertIsSatisfied();
}
