From 5470de420b2660e3042a5fec566ecc6b0d8ff93a Mon Sep 17 00:00:00 2001
From: Nanda kumar <nanda@apache.org>
Date: Sat, 15 Sep 2018 01:38:13 +0530
Subject: [PATCH] HDDS-454. TestChunkStreams#testErrorReadGroupInputStream &
 TestChunkStreams#testReadGroupInputStream are failing. Contributed by
 chencan.

---
 .../apache/hadoop/ozone/om/TestChunkStreams.java   | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/hadoop-ozone/ozone-manager/src/test/java/org/apache/hadoop/ozone/om/TestChunkStreams.java b/hadoop-ozone/ozone-manager/src/test/java/org/apache/hadoop/ozone/om/TestChunkStreams.java
index 7ce916a9cc3..7ff9d636534 100644
--- a/hadoop-ozone/ozone-manager/src/test/java/org/apache/hadoop/ozone/om/TestChunkStreams.java
+++ b/hadoop-ozone/ozone-manager/src/test/java/org/apache/hadoop/ozone/om/TestChunkStreams.java
@@ -121,6 +121,7 @@ public class TestChunkStreams {
         int tempOffset = offset;
         ChunkInputStream in =
             new ChunkInputStream(null, null, null, new ArrayList<>(), null) {
+              private long pos = 0;
               private ByteArrayInputStream in =
                   new ByteArrayInputStream(buf, tempOffset, 100);
 
@@ -131,7 +132,7 @@ public class TestChunkStreams {
 
               @Override
               public long getPos() throws IOException {
-                throw new UnsupportedOperationException();
+                return pos;
               }
 
               @Override
@@ -147,7 +148,9 @@ public class TestChunkStreams {
 
               @Override
               public int read(byte[] b, int off, int len) throws IOException {
-                return in.read(b, off, len);
+                int readLen = in.read(b, off, len);
+                pos += readLen;
+                return readLen;
               }
             };
         inputStreams.add(in);
@@ -175,6 +178,7 @@ public class TestChunkStreams {
         int tempOffset = offset;
         ChunkInputStream in =
             new ChunkInputStream(null, null, null, new ArrayList<>(), null) {
+              private long pos = 0;
               private ByteArrayInputStream in =
                   new ByteArrayInputStream(buf, tempOffset, 100);
 
@@ -185,7 +189,7 @@ public class TestChunkStreams {
 
               @Override
               public long getPos() throws IOException {
-                throw new UnsupportedOperationException();
+                return pos;
               }
 
               @Override
@@ -201,7 +205,9 @@ public class TestChunkStreams {
 
               @Override
               public int read(byte[] b, int off, int len) throws IOException {
-                return in.read(b, off, len);
+                int readLen = in.read(b, off, len);
+                pos += readLen;
+                return readLen;
               }
             };
         inputStreams.add(in);