import org.apache.hadoop.hdfs.server.datanode.DataNode.DataNodeDiskChecker; //导入依赖的package包/类
@Test (timeout = 30000)
public void testDataDirValidation() throws Throwable {
  
  DataNodeDiskChecker diskChecker = mock(DataNodeDiskChecker.class);
  doThrow(new IOException()).doThrow(new IOException()).doNothing()
    .when(diskChecker).checkDir(any(LocalFileSystem.class), any(Path.class));
  LocalFileSystem fs = mock(LocalFileSystem.class);
  AbstractList<StorageLocation> locations = new ArrayList<StorageLocation>();

  locations.add(StorageLocation.parse("file:/p1/"));
  locations.add(StorageLocation.parse("file:/p2/"));
  locations.add(StorageLocation.parse("file:/p3/"));

  List<StorageLocation> checkedLocations =
      DataNode.checkStorageLocations(locations, fs, diskChecker);
  assertEquals("number of valid data dirs", 1, checkedLocations.size());
  String validDir = checkedLocations.iterator().next().getFile().getPath();
  assertThat("p3 should be valid", new File("/p3/").getPath(), is(validDir));
}
 

import org.apache.hadoop.hdfs.server.datanode.DataNode.DataNodeDiskChecker; //导入依赖的package包/类
@Test(timeout = 30000)
public void testDataDirValidation() throws Throwable {

  DataNodeDiskChecker diskChecker = mock(DataNodeDiskChecker.class);
  doThrow(new IOException()).doThrow(new IOException()).doNothing()
      .when(diskChecker)
      .checkDir(any(LocalFileSystem.class), any(Path.class));
  LocalFileSystem fs = mock(LocalFileSystem.class);
  AbstractList<StorageLocation> locations = new ArrayList<StorageLocation>();

  locations.add(StorageLocation.parse("file:/p1/"));
  locations.add(StorageLocation.parse("file:/p2/"));
  locations.add(StorageLocation.parse("file:/p3/"));

  List<StorageLocation> checkedLocations =
      DataNode.checkStorageLocations(locations, fs, diskChecker);
  assertEquals("number of valid data dirs", 1, checkedLocations.size());
  String validDir = checkedLocations.iterator().next().getFile().getPath();
  assertThat("p3 should be valid", new File("/p3/").getPath(), is(validDir));
}
 

import org.apache.hadoop.hdfs.server.datanode.DataNode.DataNodeDiskChecker; //导入依赖的package包/类
@Test (timeout = 10000)
public void testGetDataDirsFromURIs() throws Throwable {
  
  DataNodeDiskChecker diskChecker = mock(DataNodeDiskChecker.class);
  doThrow(new IOException()).doThrow(new IOException()).doNothing()
    .when(diskChecker).checkDir(any(LocalFileSystem.class), any(Path.class));
  LocalFileSystem fs = mock(LocalFileSystem.class);
  Collection<URI> uris = Arrays.asList(new URI("file:/p1/"),
      new URI("file:/p2/"), new URI("file:/p3/"));

  List<File> dirs = DataNode.getDataDirsFromURIs(uris, fs, diskChecker);
  assertEquals("number of valid data dirs", 1, dirs.size());
  String validDir = dirs.iterator().next().getPath();
  assertEquals("p3 should be valid", new File("/p3").getPath(), validDir);
}