@Test
public void locationTest() throws IOException {
int fileId = TachyonFSTestUtils.createByteFile(mTfs, "/testFile", WriteType.MUST_CACHE, 10);
mFsShell.location(new String[] {"location", "/testFile"});
TachyonFile tFile = mTfs.getFile(new TachyonURI("/testFile"));
Assert.assertNotNull(tFile);
List<String> locationsList = tFile.getLocationHosts();
String[] commandParameters = new String[3 + locationsList.size()];
commandParameters[0] = "location";
commandParameters[1] = "/testFile";
commandParameters[2] = String.valueOf(fileId);
Iterator<String> iter = locationsList.iterator();
int i = 3;
while (iter.hasNext()) {
commandParameters[i++] = iter.next();
}
Assert.assertEquals(getCommandOutput(commandParameters), mOutput.toString());
}
@Test
public void locationNotExistTest() throws IOException {
int ret = mFsShell.location(new String[] {"location", "/NotExistFile"});
Assert.assertEquals("/NotExistFile does not exist.\n", mOutput.toString());
Assert.assertEquals(-1, ret);
}
