/** Tests that two simple receives work. */
public void testMultipleSimpleReceives() {
int len = 1000;
byte[] expected = getByteSequence(len);
ByteBuffer buf = ByteBuffer.allocate(len + 4, false);
buf.putInt(len);
buf.put(expected);
buf = buf.asReadOnlyBuffer();
buf.flip();
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
harness.recv(buf);
assertEquals(0, sentMessages.size());
assertEquals(1, receivedMessages.size());
byte[] actual = receivedMessages.get(0);
assertTrue("Incorrect recv!", Arrays.equals(actual, expected));
harness.recv(buf.rewind());
assertEquals(0, sentMessages.size());
assertEquals(2, receivedMessages.size());
actual = receivedMessages.get(1);
assertTrue("Incorrect recv!", Arrays.equals(actual, expected));
}
/** Tests handling of exceptions during dispatch. */
public void testReceiveHandlingException() {
int len = 1000;
byte[] expected = getByteSequence(len);
ByteBuffer buf = ByteBuffer.allocate(len + 4, false);
buf.putInt(len);
buf.put(expected);
buf = buf.asReadOnlyBuffer();
buf.flip();
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
RuntimeException expectedEx = new RuntimeException("Dummy exception for testing filter recv");
harness.setExceptionOnNextCompleteMessage(expectedEx);
// This recv will fail to process the message
harness.recv(buf);
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
// Send a second message, expecting the first to have been dropped.
buf.rewind();
harness.recv(buf);
assertEquals(0, sentMessages.size());
assertEquals(1, receivedMessages.size());
}
/**
* Tests that complete messages are delivered even when the message arrives one byte at a time
* from the network. Then tests that a second message can still be received all-at-once.
*/
public void testPartialOneByteReceives() {
int len = 32;
byte[] expected = getByteSequence(len);
ByteBuffer buf = ByteBuffer.allocate(len + 4, false);
buf.putInt(len);
buf.put(expected);
buf = buf.asReadOnlyBuffer();
buf.flip();
for (; buf.hasRemaining(); buf.skip(1)) {
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
ByteBuffer singleByte = buf.slice().limit(1);
harness.recv(singleByte);
}
assertEquals(0, sentMessages.size());
assertEquals(1, receivedMessages.size());
byte[] actual = receivedMessages.get(0);
assertTrue("Incorrect recv!", Arrays.equals(actual, expected));
harness.recv(buf.rewind());
assertEquals(0, sentMessages.size());
assertEquals(2, receivedMessages.size());
actual = receivedMessages.get(1);
assertTrue("Incorrect recv!", Arrays.equals(actual, expected));
}
/** Tests that a big message gets reassembled by the filter. */
public void testBigReceive() {
// Send a 62KB message in 1KB chunks
int len = 62 * 1024;
byte[] expected = new byte[len];
sizeBuf.putInt(len).flip();
harness.recv(sizeBuf.asReadOnlyBuffer());
ByteBuffer buf = ByteBuffer.wrap(getByteSequence(1024)).asReadOnlyBuffer();
// Do the recvs
for (int i = 0; i < 62; ++i) {
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
buf.rewind();
buf.get(expected, i * 1024, 1024);
buf.rewind();
harness.recv(buf);
}
assertEquals(0, sentMessages.size());
assertEquals(1, receivedMessages.size());
byte[] actual = receivedMessages.get(0);
assertTrue("Incorrect recv!", Arrays.equals(actual, expected));
}
/** Tests handling of exceptions during dispatch. */
public void testReceiveHandlingExceptionPartial() {
int len1 = 400;
int len2 = 600;
int len = len1 + len2;
byte[] expected = getByteSequence(len);
ByteBuffer buf = ByteBuffer.allocate(len + 4, false);
buf.putInt(len);
buf.put(expected);
buf = buf.asReadOnlyBuffer();
buf.flip();
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
ByteBuffer part1 = buf.slice().limit(len1 + 4);
ByteBuffer part2 = ByteBuffer.allocate(len2 + len2 + 4, false);
buf.rewind();
buf.skip(len1 + 4);
part2.put(buf);
buf.rewind();
buf.skip(len1 + 4);
part2.putInt(len2);
part2.put(buf);
part2.flip();
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
harness.recv(part1);
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
RuntimeException expectedEx = new RuntimeException("Dummy exception for testing filter recv");
harness.setExceptionOnNextCompleteMessage(expectedEx);
// This recv should process the second message.
harness.recv(part2);
assertEquals(0, sentMessages.size());
assertEquals(1, receivedMessages.size());
byte[] actual = receivedMessages.get(0);
assertEquals(len2, actual.length);
}
/** Tests handling of a valid message with payload length zero. */
public void testReceiveValidZeroLength() {
sizeBuf.putInt(0).flip();
harness.recv(sizeBuf.asReadOnlyBuffer());
assertEquals(0, sentMessages.size());
assertEquals(1, receivedMessages.size());
assertEquals(0, receivedMessages.get(0).length);
}
/** Tests that the send filter correctly prepends the message length. */
public void testSend() {
int len = 1000;
byte[] sendData = getByteSequence(len);
ByteBuffer buf = ByteBuffer.allocate(len + 4, false);
buf.putInt(len);
buf.put(sendData);
buf = buf.asReadOnlyBuffer();
buf.flip();
byte[] expected = new byte[buf.remaining()];
buf.get(expected);
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
harness.send(sendData);
assertEquals(1, sentMessages.size());
assertEquals(0, receivedMessages.size());
byte[] actual = sentMessages.get(0);
assertTrue("Incorrect send!", Arrays.equals(actual, expected));
}
/** Tests that zero-byte receives have no effect on complete message delivery. */
public void testPartialReceiveNoBytes() {
int len = 1000;
sizeBuf.putInt(len).flip();
harness.recv(sizeBuf.asReadOnlyBuffer());
ByteBuffer emptyBuf = sizeBuf.slice().limit(0).asReadOnlyBuffer();
harness.recv(emptyBuf);
assertEquals(0, sentMessages.size());
assertEquals(0, receivedMessages.size());
byte[] expected = getByteSequence(len);
ByteBuffer buf = ByteBuffer.wrap(expected).asReadOnlyBuffer();
harness.recv(buf);
assertEquals(0, sentMessages.size());
assertEquals(1, receivedMessages.size());
byte[] actual = receivedMessages.get(0);
assertTrue("Incorrect recv!", Arrays.equals(actual, expected));
harness.recv(emptyBuf);
assertEquals(0, sentMessages.size());
assertEquals(1, receivedMessages.size());
harness.recv(sizeBuf.rewind().asReadOnlyBuffer());
harness.recv(buf.rewind());
assertEquals(0, sentMessages.size());
assertEquals(2, receivedMessages.size());
actual = receivedMessages.get(1);
assertTrue("Incorrect recv!", Arrays.equals(actual, expected));
}
