@Test
public void writeUnicodeFrameAndClose() throws Exception {
String unicodeStr = "\u0628\u00de";
_wsStream.setDataFormat(DataFormat.Text);
_wsStream.write(Buffer.buffer(unicodeStr.getBytes(StandardCharsets.UTF_8)));
_wsStream.finishCurrentMessage();
verify(_socket).writeFrame(textFrameEq(true, unicodeStr));
}
@Test
public void writeTwoDifferentMessages() throws Exception {
byte[] bytes = {0x1, 0x3, 0x5, 0x7};
String strContent = "foo";
_wsStream.write(Buffer.buffer(bytes));
_wsStream.finishCurrentMessage();
_wsStream.setDataFormat(DataFormat.Text);
_wsStream.write(Buffer.buffer(strContent.getBytes(StandardCharsets.UTF_8)));
_wsStream.finishCurrentMessage();
verify(_socket).writeFrame(binaryFrameEq(true, bytes));
verify(_socket).writeFrame(textFrameEq(true, strContent));
}
@BeforeMethod
public void setUp() throws Exception {
_socket = mock(ServerWebSocket.class);
_wsStream = new WebsocketWriteStream(_socket, MAX_FRAMESIZE);
_wsStream.setDataFormat(DataFormat.Binary);
}
