Have dicttool test tidy up after itself.

Bug: 13776363
Change-Id: Icb1d3fc0efe71e0339b434928e8aed507f2fb590
This commit is contained in:
Jean Chalard 2014-05-21 15:58:49 +09:00
parent 75cb258ee4
commit 7086d88d3e
2 changed files with 36 additions and 4 deletions

View file

@ -16,6 +16,8 @@
package android.test; package android.test;
import com.android.inputmethod.latin.dicttool.Test;
import junit.framework.TestCase; import junit.framework.TestCase;
import java.io.File; import java.io.File;
@ -27,7 +29,11 @@ import java.io.File;
*/ */
public class AndroidTestCase extends TestCase { public class AndroidTestCase extends TestCase {
public File getCacheDir() { public File getCacheDir() {
return new File("."); final File dir = Test.TEST_TMP_DIR;
if (!dir.isDirectory()) {
dir.mkdirs();
}
return dir;
} }
public AndroidTestCase getContext() { public AndroidTestCase getContext() {
return this; return this;

View file

@ -19,16 +19,29 @@ package com.android.inputmethod.latin.dicttool;
import com.android.inputmethod.latin.makedict.BinaryDictDecoderEncoderTests; import com.android.inputmethod.latin.makedict.BinaryDictDecoderEncoderTests;
import com.android.inputmethod.latin.makedict.BinaryDictEncoderFlattenTreeTests; import com.android.inputmethod.latin.makedict.BinaryDictEncoderFlattenTreeTests;
import com.android.inputmethod.latin.makedict.FusionDictionaryTest; import com.android.inputmethod.latin.makedict.FusionDictionaryTest;
import com.android.inputmethod.latin.utils.FileUtils;
import java.io.File;
import java.io.IOException;
import java.lang.reflect.Constructor; import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException; import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method; import java.lang.reflect.Method;
import java.nio.file.Files;
import java.util.ArrayList; import java.util.ArrayList;
/** /**
* Dicttool command implementing self-tests. * Dicttool command implementing self-tests.
*/ */
public class Test extends Dicttool.Command { public class Test extends Dicttool.Command {
private static final String getTmpDir() {
try {
return Files.createTempDirectory("dicttool").toString();
} catch (IOException e) {
throw new RuntimeException("Can't get temporary directory", e);
}
}
private static final String TEST_TMP_DIR_BASE = getTmpDir();
public static final File TEST_TMP_DIR = new File(TEST_TMP_DIR_BASE);
public static final String COMMAND = "test"; public static final String COMMAND = "test";
private static final int DEFAULT_MAX_UNIGRAMS = 1500; private static final int DEFAULT_MAX_UNIGRAMS = 1500;
private long mSeed = System.currentTimeMillis(); private long mSeed = System.currentTimeMillis();
@ -56,8 +69,12 @@ public class Test extends Dicttool.Command {
@Override @Override
public String getHelp() { public String getHelp() {
final StringBuilder s = new StringBuilder("test [-s seed] [-m maxUnigrams] [testName...]\n" final StringBuilder s = new StringBuilder(
+ "If seed is not specified, the current time is used.\nTest list is:\n"); "test [-s seed] [-m maxUnigrams] [-n] [testName...]\n"
+ "If seed is not specified, the current time is used.\n"
+ "If -n option is provided, do not delete temporary files in "
+ TEST_TMP_DIR_BASE + "/*.\n"
+ "Test list is:\n");
for (final Method m : mAllTestMethods) { for (final Method m : mAllTestMethods) {
s.append(" "); s.append(" ");
s.append(m.getName()); s.append(m.getName());
@ -70,17 +87,26 @@ public class Test extends Dicttool.Command {
public void run() throws IllegalAccessException, InstantiationException, public void run() throws IllegalAccessException, InstantiationException,
InvocationTargetException { InvocationTargetException {
int i = 0; int i = 0;
boolean deleteTmpDir = true;
while (i < mArgs.length) { while (i < mArgs.length) {
final String arg = mArgs[i++]; final String arg = mArgs[i++];
if ("-s".equals(arg)) { if ("-s".equals(arg)) {
mSeed = Long.parseLong(mArgs[i++]); mSeed = Long.parseLong(mArgs[i++]);
} else if ("-m".equals(arg)) { } else if ("-m".equals(arg)) {
mMaxUnigrams = Integer.parseInt(mArgs[i++]); mMaxUnigrams = Integer.parseInt(mArgs[i++]);
} else if ("-n".equals(arg)) {
deleteTmpDir = false;
} else { } else {
mUsedTestMethods.add(arg); mUsedTestMethods.add(arg);
} }
} }
runChosenTests(); try {
runChosenTests();
} finally {
if (deleteTmpDir) {
FileUtils.deleteRecursively(TEST_TMP_DIR);
}
}
} }
private void runChosenTests() throws IllegalAccessException, InstantiationException, private void runChosenTests() throws IllegalAccessException, InstantiationException,