package org.jboss.as.test.shared;
import java.io.BufferedInputStream;
import java.io.Closeable;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.StandardCopyOption;
/**
* Shared fs utils for the test suite.
*
* Keep this class dependent only on JDK classes as it's packed to deployed WAR's.
*
* @author Stuart Douglas
*/
public class FileUtils {
private FileUtils() {
}
public static String readFile(Class<?> testClass, String fileName) {
final URL res = testClass.getResource(fileName);
return readFile(res);
}
public static String readFile(URL url) {
BufferedInputStream stream = null;
try {
stream = new BufferedInputStream(url.openStream());
byte[] buff = new byte[1024];
StringBuilder builder = new StringBuilder();
int read = -1;
while ((read = stream.read(buff)) != -1) {
builder.append(new String(buff, 0, read, StandardCharsets.UTF_8));
}
return builder.toString();
} catch (IOException e) {
throw new RuntimeException(e);
} finally {
if (stream != null) {
try {
stream.close();
} catch (IOException e) {
//ignore
}
}
}
}
public static File getFileOrCheckParentsIfNotFound( String baseStr, String path ) throws FileNotFoundException {
//File f = new File( System.getProperty("jbossas.project.dir", "../../..") );
File base = new File( baseStr );
if( ! base.exists() ){
throw new FileNotFoundException( "Base path not found: " + base.getPath() );
}
base = base.getAbsoluteFile();
File f = new File( base, path );
if ( f.exists() )
return f;
File fLast = f;
while( ! f.exists() ){
int slash = path.lastIndexOf( File.separatorChar );
if( slash <= 0 ) // no slash or "/xxx"
throw new FileNotFoundException("Path not found: " + f.getPath());
path = path.substring( 0, slash );
fLast = f;
f = new File( base, path );
}
// When first existing is found, report the last non-existent.
throw new FileNotFoundException("Path not found: " + fLast.getPath());
}
public static void copyFile(final File src, final File dest) throws IOException {
Files.copy(src.toPath(),dest.toPath(), StandardCopyOption.REPLACE_EXISTING);
}
public static void copyFile(final InputStream in, final File dest) throws IOException {
dest.getParentFile().mkdirs();
Files.copy(in,dest.toPath(), StandardCopyOption.REPLACE_EXISTING);
}
/**
* Waits for either the specified files to exist for up to 20 seconds.
*
* Some tests have problems on windows as file creation does not show up straight away, this is basically
* a workaround. 20 seconds was selected as this will generally not wait at all and so will not add any time
* to the test suite runs, and will only wait for the full 20 seconds on a genuine failure.
*
* Returns true if the files exist, false otherwise.
* @param files The files to wait for
*/
public static boolean waitForFiles(File... files) {
long exitTime = 20000 + System.currentTimeMillis();
do {
boolean allExist = true;
for(File file : files) {
if(!file.exists()) {
allExist = false;
break;
}
}
if(allExist) {
return true;
}
} while (System.currentTimeMillis() < exitTime);
return false;
}
public static void close(Closeable closeable) {
try {
closeable.close();
} catch (IOException ignore) {
}
}
}// class