Addendum for HDFS-7748.

This commit is contained in:
Haohui Mai 2015-03-30 12:23:45 -07:00
parent 2a945d24f7
commit 0967b1d99d

View File

@ -33,7 +33,6 @@
import java.nio.ByteBuffer;
import java.util.Random;
import com.sun.xml.internal.messaging.saaj.util.ByteOutputStream;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.hadoop.conf.Configuration;
@ -534,17 +533,17 @@ public void TestPipeLineAckCompatibility() throws IOException {
Status.CHECKSUM_OK))
.build();
ByteOutputStream oldAckBytes = new ByteOutputStream();
ByteArrayOutputStream oldAckBytes = new ByteArrayOutputStream();
proto.writeDelimitedTo(oldAckBytes);
PipelineAck oldAck = new PipelineAck();
oldAck.readFields(new ByteArrayInputStream(oldAckBytes.getBytes()));
oldAck.readFields(new ByteArrayInputStream(oldAckBytes.toByteArray()));
assertEquals(PipelineAck.combineHeader(PipelineAck.ECN.DISABLED, Status
.CHECKSUM_OK), oldAck.getHeaderFlag(0));
PipelineAck newAck = new PipelineAck();
ByteOutputStream newAckBytes = new ByteOutputStream();
ByteArrayOutputStream newAckBytes = new ByteArrayOutputStream();
newProto.writeDelimitedTo(newAckBytes);
newAck.readFields(new ByteArrayInputStream(newAckBytes.getBytes()));
newAck.readFields(new ByteArrayInputStream(newAckBytes.toByteArray()));
assertEquals(PipelineAck.combineHeader(PipelineAck.ECN.SUPPORTED, Status
.CHECKSUM_OK), newAck.getHeaderFlag(0));
}