public void testVersion() {
WireProtocolHeader header = new WireProtocolHeader();
header.setVersion((byte) 15);
assertTrue(15 == header.getVersion());
header.setVersion(WireProtocolHeader.VERSION_2);
assertTrue(WireProtocolHeader.VERSION_2 == header.getVersion());
boolean exception = false;
try {
header.setVersion((byte) 0);
} catch (Exception e) {
exception = true;
}
assertTrue(exception);
exception = false;
try {
header.setVersion((byte) 16);
} catch (Exception e) {
exception = true;
}
assertTrue(exception);
exception = false;
try {
header.setVersion((byte) -1);
} catch (Exception e) {
exception = true;
}
assertTrue(exception);
}
public void testOptions() {
WireProtocolHeader header = new WireProtocolHeader();
byte oldLength = header.getHeaderLength();
header.setOptions(new byte[] {});
byte newLength = header.getHeaderLength();
assertTrue(newLength >= oldLength);
assertTrue(newLength == (WireProtocolHeader.MIN_LENGTH / 4));
byte[] maxOptions = new byte[WireProtocolHeader.MAX_LENGTH - WireProtocolHeader.MIN_LENGTH];
Arrays.fill(maxOptions, (byte) 0xFF);
header.setOptions(maxOptions);
assertTrue(header.getHeaderLength() * 4 == WireProtocolHeader.MAX_LENGTH);
Arrays.equals(maxOptions, header.getOptions());
header.setOptions(null);
assertTrue(header.getHeaderLength() * 4 == WireProtocolHeader.MIN_LENGTH);
assertTrue(header.getOptions().length == 0);
}
public void testGoodHeader() {
byte[] data = getGoodHeader();
TCByteBuffer buffer = TCByteBufferFactory.getInstance(false, WireProtocolHeader.MAX_LENGTH);
buffer.put(data);
buffer.flip();
WireProtocolHeader header = new WireProtocolHeader(buffer);
System.out.println(header);
assertTrue(header.isChecksumValid());
assertTrue(header.getVersion() == WireProtocolHeader.VERSION_2);
assertTrue(header.getHeaderLength() == 8);
assertTrue(header.getTypeOfService() == 2);
assertTrue(header.getTimeToLive() == 3);
assertTrue(header.getTotalPacketLength() == 255);
assertTrue(header.getChecksum() == 2744585179L);
assertTrue(
Arrays.equals(
header.getSourceAddress(), new byte[] {(byte) 0xFF, (byte) 1, (byte) 0xFF, (byte) 1}));
assertTrue(
Arrays.equals(
header.getDestinationAddress(),
new byte[] {(byte) 1, (byte) 0xFF, (byte) 1, (byte) 0xFF}));
assertTrue(header.getSourcePort() == 43605);
assertTrue(header.getDestinationPort() == 21930);
assertTrue(header.getMessageCount() == 5);
assertTrue(header.getOptions().length == 0);
try {
header.validate();
} catch (WireProtocolHeaderFormatException e) {
fail(e.getMessage());
}
// changing data in the header should cause the checksum to need to be recomputed
header.setVersion((byte) (header.getVersion() + 1));
assertFalse(header.isChecksumValid());
// Fix and validate the checksum
header.computeChecksum();
assertTrue(header.isChecksumValid());
}
