@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(); }