Addendum for HDFS-7748.

This commit is contained in:
Haohui Mai 2015-03-30 12:23:45 -07:00
parent cba4ed1678
commit 02fcf622fb
1 changed files with 4 additions and 5 deletions

View File

@ -33,7 +33,6 @@ import java.net.Socket;
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;
@ -540,17 +539,17 @@ public class TestDataTransferProtocol {
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));
}