Add UT

Alluxio · Dec 22, 2023 · e8dc5dc · e8dc5dc
1 parent ae3a997
commit e8dc5dc
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/core/client/fs/src/main/java/alluxio/client/file/AlluxioFileInStream.java b/core/client/fs/src/main/java/alluxio/client/file/AlluxioFileInStream.java
@@ -423,7 +423,8 @@ private void updateStream() throws IOException {
     mBlockInStream.seek(offset);
   }
 
-  private boolean isStatusOutdated() {
+  @VisibleForTesting
+  public boolean isStatusOutdated() {
     return System.currentTimeMillis() > mStatusOutdatedTime;
   }
 

diff --git a/core/client/fs/src/test/java/alluxio/client/file/AlluxioFileInStreamTest.java b/core/client/fs/src/test/java/alluxio/client/file/AlluxioFileInStreamTest.java
@@ -53,6 +53,7 @@
 import alluxio.wire.WorkerNetAddress;
 
 import org.junit.After;
+import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -958,4 +959,17 @@ private void testReadBuffer(int dataRead) throws Exception {
   private void validatePartialCaching(int index, int readSize) {
     assertEquals(readSize, mInStreams.get(index).getBytesRead());
   }
+
+  @Test
+  public void testStatusOutdated() throws IOException {
+    OpenFilePOptions options =
+        OpenFilePOptions.newBuilder().setReadType(ReadPType.CACHE_PROMOTE).build();
+    mTestStream = new AlluxioFileInStream(mStatus, new InStreamOptions(mStatus, options, mConf,
+        mContext), mContext);
+    Assert.assertFalse(mTestStream.isStatusOutdated());
+    mConf.set(PropertyKey.USER_FILE_IN_STREAM_STATUS_EXPIRATION_TIME, 0L);
+    mTestStream = new AlluxioFileInStream(mStatus, new InStreamOptions(mStatus, options, mConf,
+        mContext), mContext);
+    Assert.assertTrue(mTestStream.isStatusOutdated());
+  }
 }