// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
package org.chromium.base.test.util;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.Reader;
import java.io.Writer;
import java.util.Arrays;
/** Utility class for dealing with files for test. */
public class TestFileUtil {
public static void createNewHtmlFile(String name, String title, String body)
throws IOException {
createNewHtmlFile(new File(name), title, body);
}
public static void createNewHtmlFile(File file, String title, String body) throws IOException {
if (!file.createNewFile()) {
throw new IOException("File \"" + file.getAbsolutePath() + "\" already exists");
}
Writer writer = null;
try {
writer = new OutputStreamWriter(new FileOutputStream(file), "UTF-8");
writer.write(
"<html><meta charset=\"UTF-8\" />"
+ " <head><title>"
+ title
+ "</title></head>"
+ " <body>"
+ (body != null ? body : "")
+ " </body>"
+ " </html>");
} finally {
if (writer != null) {
writer.close();
}
}
}
public static void deleteFile(String name) {
deleteFile(new File(name));
}
public static void deleteFile(File file) {
boolean deleted = file.delete();
assert (deleted || !file.exists());
}
/**
* @param fileName the file to read in.
* @param sizeLimit cap on the file size: will throw an exception if exceeded
* @return Array of chars read from the file
* @throws FileNotFoundException file does not exceed
* @throws IOException error encountered accessing the file
*/
public static char[] readUtf8File(String fileName, int sizeLimit)
throws FileNotFoundException, IOException {
Reader reader = null;
try {
File f = new File(fileName);
if (f.length() > sizeLimit) {
throw new IOException(
"File "
+ fileName
+ " length "
+ f.length()
+ " exceeds limit "
+ sizeLimit);
}
char[] buffer = new char[(int) f.length()];
reader = new InputStreamReader(new FileInputStream(f), "UTF-8");
int charsRead = reader.read(buffer);
// Debug check that we've exhausted the input stream (will fail e.g. if the
// file grew after we inspected its length).
assert !reader.ready();
return charsRead < buffer.length ? Arrays.copyOfRange(buffer, 0, charsRead) : buffer;
} finally {
if (reader != null) reader.close();
}
}
}
Codebase Browser
chromium