Fix test

Alluxio · Dec 6, 2017 · 1c4d72f · 1c4d72f
1 parent 65808c1
commit 1c4d72f
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/tests/src/test/java/alluxio/master/file/UfsSyncIntegrationTest.java b/tests/src/test/java/alluxio/master/file/UfsSyncIntegrationTest.java
@@ -44,6 +44,7 @@
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
+import java.util.stream.Collectors;
 
 /**
  * Tests the loading of metadata and the available options.
@@ -235,25 +236,27 @@ public void unpersistedFileSync() throws Exception {
     ListStatusOptions options =
         ListStatusOptions.defaults().setLoadMetadataType(LoadMetadataType.Never)
             .setCommonOptions(SYNC_ALWAYS);
-    List<URIStatus> initialStatusList =
+    List<String> initialStatusList =
-        mFileSystem.listStatus(new AlluxioURI(alluxioPath(ROOT_DIR)), options);
+        mFileSystem.listStatus(new AlluxioURI(alluxioPath(ROOT_DIR)), options).stream()
+            .map(URIStatus::getName).collect(Collectors.toList());
 
     // write a MUST_CACHE file
     writeMustCacheFile(alluxioPath(NEW_FILE), 1);
 
     // List the status with force sync.
-    List<URIStatus> syncStatusList =
+    List<String> syncStatusList =
-        mFileSystem.listStatus(new AlluxioURI(alluxioPath(ROOT_DIR)), options);
+        mFileSystem.listStatus(new AlluxioURI(alluxioPath(ROOT_DIR)), options).stream()
+            .map(URIStatus::getName).collect(Collectors.toList());
 
-    Set<URIStatus> initialSet = Sets.newHashSet(initialStatusList);
+    Set<String> initialSet = Sets.newHashSet(initialStatusList);
-    Set<URIStatus> syncSet = Sets.newHashSet(syncStatusList);
+    Set<String> syncSet = Sets.newHashSet(syncStatusList);
     Assert.assertTrue(syncSet.size() > initialSet.size());
     syncSet.removeAll(initialSet);
 
     // only the MUST_CACHE file should remain.
     Assert.assertTrue(syncSet.size() == 1);
-    URIStatus file = syncSet.iterator().next();
+    String file = syncSet.iterator().next();
-    Assert.assertTrue(file.getName().equals(new AlluxioURI(NEW_FILE).getName()));
+    Assert.assertTrue(file.equals(new AlluxioURI(NEW_FILE).getName()));
   }
 
   private String ufsPath(String path) {