Maxim Shafirov
113 lines
4.3 KiB
Java
113 lines
4.3 KiB
Java
/*
|
|
* Copyright 2010-2012 JetBrains s.r.o.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package org.jetbrains.jet;
|
|
|
|
import com.intellij.openapi.application.PathManager;
|
|
import junit.framework.Test;
|
|
import junit.framework.TestSuite;
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
|
import java.io.File;
|
|
import java.io.FileFilter;
|
|
import java.io.FilenameFilter;
|
|
import java.util.Arrays;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
|
|
/**
|
|
* @author abreslav
|
|
*/
|
|
public abstract class JetTestCaseBuilder {
|
|
public static FilenameFilter emptyFilter = new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File file, String name) {
|
|
return true;
|
|
}
|
|
};
|
|
|
|
public static String getTestDataPathBase() {
|
|
return getHomeDirectory() + "/compiler/testData";
|
|
}
|
|
|
|
public static String getHomeDirectory() {
|
|
return new File(PathManager.getResourceRoot(JetTestCaseBuilder.class, "/org/jetbrains/jet/JetTestCaseBuilder.class")).getParentFile().getParentFile().getParent();
|
|
}
|
|
|
|
public interface NamedTestFactory {
|
|
@NotNull Test createTest(@NotNull String dataPath, @NotNull String name, @NotNull File file);
|
|
}
|
|
|
|
@NotNull
|
|
public static TestSuite suiteForDirectory(String baseDataDir, @NotNull final String dataPath, boolean recursive, @NotNull NamedTestFactory factory) {
|
|
return suiteForDirectory(baseDataDir, dataPath, recursive, emptyFilter, factory);
|
|
}
|
|
|
|
@NotNull
|
|
public static TestSuite suiteForDirectory(String baseDataDir, @NotNull final String dataPath, boolean recursive, final FilenameFilter filter, @NotNull NamedTestFactory factory) {
|
|
TestSuite suite = new TestSuite(dataPath);
|
|
appendTestsInDirectory(baseDataDir, dataPath, recursive, filter, factory, suite);
|
|
return suite;
|
|
}
|
|
|
|
public static void appendTestsInDirectory(String baseDataDir, String dataPath, boolean recursive, final FilenameFilter filter, NamedTestFactory factory, TestSuite suite) {
|
|
final String extensionJet = ".jet";
|
|
final String extensionKt = ".kt";
|
|
final FilenameFilter extensionFilter = new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File dir, String name) {
|
|
return name.endsWith(extensionJet) || name.endsWith(extensionKt);
|
|
}
|
|
};
|
|
FilenameFilter resultFilter;
|
|
if (filter != emptyFilter) {
|
|
resultFilter = new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File file, String s) {
|
|
return extensionFilter.accept(file, s) && filter.accept(file, s);
|
|
}
|
|
};
|
|
}
|
|
else {
|
|
resultFilter = extensionFilter;
|
|
}
|
|
File dir = new File(baseDataDir + dataPath);
|
|
FileFilter dirFilter = new FileFilter() {
|
|
@Override
|
|
public boolean accept(File pathname) {
|
|
return pathname.isDirectory();
|
|
}
|
|
};
|
|
if (recursive) {
|
|
File[] files = dir.listFiles(dirFilter);
|
|
assert files != null : dir;
|
|
List<File> subdirs = Arrays.asList(files);
|
|
Collections.sort(subdirs);
|
|
for (File subdir : subdirs) {
|
|
suite.addTest(suiteForDirectory(baseDataDir, dataPath + "/" + subdir.getName(), recursive, filter, factory));
|
|
}
|
|
}
|
|
List<File> files = Arrays.asList(dir.listFiles(resultFilter));
|
|
Collections.sort(files);
|
|
for (File file : files) {
|
|
String fileName = file.getName();
|
|
assert fileName != null;
|
|
String extension = fileName.endsWith(extensionJet) ? extensionJet : extensionKt;
|
|
suite.addTest(factory.createTest(dataPath, fileName.substring(0, fileName.length() - extension.length()), file));
|
|
}
|
|
}
|
|
}
|