Space fix

Author: anmolanmol1234

hadoop-tools/hadoop-azure/src/test/java/org/apache/hadoop/fs/azurebfs/ITestAbfsReadWriteAndSeek.java

@@ -58,17 +58,17 @@ public class ITestAbfsReadWriteAndSeek extends AbstractAbfsScaleTest {
   @Parameterized.Parameters(name = "Size={0}-readahead={1}")
   public static Iterable<Object[]> sizes() {
     return Arrays.asList(new Object[][]{{MIN_BUFFER_SIZE, true},
-            {DEFAULT_READ_BUFFER_SIZE, false},
-            {DEFAULT_READ_BUFFER_SIZE, true},
-            {APPENDBLOB_MAX_WRITE_BUFFER_SIZE, false},
-            {MAX_BUFFER_SIZE, true}});
+        {DEFAULT_READ_BUFFER_SIZE, false},
+        {DEFAULT_READ_BUFFER_SIZE, true},
+        {APPENDBLOB_MAX_WRITE_BUFFER_SIZE, false},
+        {MAX_BUFFER_SIZE, true}});
   }
 
   private final int size;
   private final boolean readaheadEnabled;
 
   public ITestAbfsReadWriteAndSeek(final int size,
-                                   final boolean readaheadEnabled) throws Exception {
+      final boolean readaheadEnabled) throws Exception {
     this.size = size;
     this.readaheadEnabled = readaheadEnabled;
   }
@@ -93,7 +93,7 @@ private void testReadWriteAndSeek(int bufferSize) throws Exception {
     try {
       stream.write(b);
     } finally{
-      stream.close();
+    stream.close();
     }
     logIOStatisticsAtLevel(LOG, IOSTATISTICS_LOGGING_LEVEL_INFO, stream);
 
@@ -103,10 +103,10 @@ private void testReadWriteAndSeek(int bufferSize) throws Exception {
     try (FSDataInputStream inputStream = fs.open(testPath)) {
       statisticsSource = inputStream;
       ((AbfsInputStream) inputStream.getWrappedStream()).registerListener(
-              new TracingHeaderValidator(abfsConfiguration.getClientCorrelationId(),
-                      fs.getFileSystemId(), FSOperationType.READ, true, 0,
-                      ((AbfsInputStream) inputStream.getWrappedStream())
-                              .getStreamID()));
+          new TracingHeaderValidator(abfsConfiguration.getClientCorrelationId(),
+              fs.getFileSystemId(), FSOperationType.READ, true, 0,
+              ((AbfsInputStream) inputStream.getWrappedStream())
+                  .getStreamID()));
       inputStream.seek(bufferSize);
       result = inputStream.read(readBuffer, bufferSize, bufferSize);
       assertNotEquals(-1, result);
@@ -138,10 +138,10 @@ public void testReadAheadRequestID() throws java.io.IOException {
     Path testPath = path(TEST_PATH);
     try (FSDataOutputStream stream = fs.create(testPath)) {
       ((AbfsOutputStream) stream.getWrappedStream()).registerListener(
-              new TracingHeaderValidator(abfsConfiguration.getClientCorrelationId(),
-                      fs.getFileSystemId(), FSOperationType.WRITE, false, 0,
-                      ((AbfsOutputStream) stream.getWrappedStream())
-                              .getStreamID()));
+          new TracingHeaderValidator(abfsConfiguration.getClientCorrelationId(),
+              fs.getFileSystemId(), FSOperationType.WRITE, false, 0,
+              ((AbfsOutputStream) stream.getWrappedStream())
+                  .getStreamID()));
       stream.write(b);
       logIOStatisticsAtLevel(LOG, IOSTATISTICS_LOGGING_LEVEL_INFO, stream);
     }
@@ -150,17 +150,17 @@ public void testReadAheadRequestID() throws java.io.IOException {
     final byte[] readBuffer = new byte[4 * bufferSize];
     int result;
     fs.registerListener(
-            new TracingHeaderValidator(abfsConfiguration.getClientCorrelationId(),
-                    fs.getFileSystemId(), FSOperationType.OPEN, false, 0));
+        new TracingHeaderValidator(abfsConfiguration.getClientCorrelationId(),
+            fs.getFileSystemId(), FSOperationType.OPEN, false, 0));
     try (FSDataInputStream inputStream = fs.open(testPath)) {
       ((AbfsInputStream) inputStream.getWrappedStream()).registerListener(
-              new TracingHeaderValidator(abfsConfiguration.getClientCorrelationId(),
-                      fs.getFileSystemId(), FSOperationType.READ, false, 0,
-                      ((AbfsInputStream) inputStream.getWrappedStream())
-                              .getStreamID()));
+          new TracingHeaderValidator(abfsConfiguration.getClientCorrelationId(),
+              fs.getFileSystemId(), FSOperationType.READ, false, 0,
+              ((AbfsInputStream) inputStream.getWrappedStream())
+                  .getStreamID()));
       result = inputStream.read(readBuffer, 0, bufferSize*4);
       logIOStatisticsAtLevel(LOG, IOSTATISTICS_LOGGING_LEVEL_INFO, inputStream);
     }
     fs.registerListener(null);
   }
-}
+}

hadoop-tools/hadoop-azure/src/test/java/org/apache/hadoop/fs/azurebfs/services/TestAbfsInputStream.java

@@ -39,6 +39,7 @@
 import org.apache.hadoop.fs.azurebfs.contracts.services.ReadBufferStatus;
 import org.apache.hadoop.fs.azurebfs.utils.TestCachedSASToken;
 import org.apache.hadoop.fs.azurebfs.utils.TracingContext;
+import org.mockito.Mockito;
 
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.doReturn;