--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/langtools/test/tools/javac/api/file/SJFM_TestBase.java Tue Jan 06 14:51:10 2015 -0800
@@ -0,0 +1,226 @@
+/*
+ * Copyright (c) 2014, 2015, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.io.UncheckedIOException;
+import java.lang.annotation.Annotation;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+import java.nio.file.FileSystem;
+import java.nio.file.FileSystems;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
+
+import javax.tools.JavaCompiler;
+import javax.tools.StandardJavaFileManager;
+import javax.tools.ToolProvider;
+
+/**
+ * Base class for unit tests for StandardJavaFileManager.
+ */
+class SJFM_TestBase {
+
+ /** Shared compiler instance. */
+ JavaCompiler comp;
+
+ /** A list of items to be closed when the test is complete. */
+ List<AutoCloseable> closeables;
+
+ /**
+ * Runs a test. This is the primary entry point and should generally be
+ * called from each test's main method.
+ * It calls all methods annotated with {@code @Test} with the instances
+ * of StandardJavaFileManager to be tested.
+ *
+ * @throws Exception if the test fails.
+ */
+ void run() throws Exception {
+ comp = ToolProvider.getSystemJavaCompiler();
+ closeables = new ArrayList<>();
+
+ try (StandardJavaFileManager systemFileManager = comp.getStandardFileManager(null, null, null);
+ StandardJavaFileManager customFileManager = new MyStandardJavaFileManager(systemFileManager)) {
+ test(systemFileManager);
+ test(customFileManager);
+ } finally {
+ for (AutoCloseable c: closeables) {
+ try {
+ c.close();
+ } catch (IOException e) {
+ error("Exception closing " + c + ": " + e);
+ }
+ }
+ }
+
+ if (errors > 0)
+ throw new Exception(errors + " errors occurred");
+ }
+
+ /**
+ * Get the file managers to be tested.
+ *
+ * Currently, two are provided:
+ * <ol>
+ * <li>the system-provided file manager
+ * <li>a custom file manager, which relies on the default methods provided in the
+ * StandardJavaFileManager interface
+ * </li>
+ *
+ * @return the file managers to be tested
+ */
+ List<StandardJavaFileManager> getTestFileManagers() {
+ StandardJavaFileManager systemFileManager = comp.getStandardFileManager(null, null, null);
+ StandardJavaFileManager customFileManager = new MyStandardJavaFileManager(systemFileManager);
+ return Arrays.asList(systemFileManager, customFileManager);
+ }
+
+ /**
+ * Tests a specific file manager, by calling all methods annotated
+ * with {@code @Test} passing this file manager as an argument.
+ *
+ * @param fm the file manager to be tested
+ * @throws Exception if the test fails
+ */
+ void test(StandardJavaFileManager fm) throws Exception {
+ System.err.println("Testing " + fm);
+ for (Method m: getClass().getDeclaredMethods()) {
+ Annotation a = m.getAnnotation(Test.class);
+ if (a != null) {
+ try {
+ System.err.println("Test " + m.getName());
+ m.invoke(this, new Object[] { fm });
+ } catch (InvocationTargetException e) {
+ Throwable cause = e.getCause();
+ throw (cause instanceof Exception) ? ((Exception) cause) : e;
+ }
+ System.err.println();
+ }
+ }
+ }
+
+ /** Marker annotation for test cases. */
+ @Retention(RetentionPolicy.RUNTIME)
+ @interface Test { }
+
+ /**
+ * Returns a series of paths for artifacts in the default file system.
+ * The paths are for the .java files in the test.src directory.
+ *
+ * @return a list of paths
+ * @throws IOException
+ */
+ List<Path> getTestFilePaths() throws IOException {
+ String testSrc = System.getProperty("test.src");
+ return Files.list(Paths.get(testSrc))
+ .filter(p -> p.getFileName().toString().endsWith(".java"))
+ .collect(Collectors.toList());
+ }
+
+ private FileSystem zipfs;
+ private List<Path> zipPaths;
+
+ /**
+ * Returns a series of paths for artifacts in a non-default file system.
+ * A zip file is created containing copies of the .java files in the
+ * test.src directory. The paths that are returned refer to these files.
+ *
+ * @return a list of paths
+ * @throws IOException
+ */
+ List<Path> getTestZipPaths() throws IOException {
+ if (zipfs == null) {
+ Path testZip = createSourceZip();
+ zipfs = FileSystems.newFileSystem(testZip, null);
+ closeables.add(zipfs);
+ zipPaths = Files.list(zipfs.getRootDirectories().iterator().next())
+ .filter(p -> p.getFileName().toString().endsWith(".java"))
+ .collect(Collectors.toList());
+ }
+ return zipPaths;
+ }
+
+ /**
+ * Create a zip file containing the contents of the test.src directory.
+ *
+ * @return a path for the zip file.
+ * @throws IOException if there is a problem creating the file
+ */
+ private Path createSourceZip() throws IOException {
+ Path testSrc = Paths.get(System.getProperty("test.src"));
+ Path testZip = Paths.get("test.zip");
+ try (OutputStream os = Files.newOutputStream(testZip)) {
+ try (ZipOutputStream zos = new ZipOutputStream(os)) {
+ Files.list(testSrc)
+ .filter(p -> p.getFileName().toString().endsWith(".java"))
+ .forEach(p -> {
+ try {
+ zos.putNextEntry(new ZipEntry(p.getFileName().toString()));
+ zos.write(Files.readAllBytes(p));
+ zos.closeEntry();
+ } catch (IOException ex) {
+ throw new UncheckedIOException(ex);
+ }
+ });
+ }
+ }
+ return testZip;
+ }
+
+ /**
+ * Tests whether it is expected that a file manager will be able
+ * to create a series of file objects from a series of paths.
+ *
+ * MyStandardJavaFileManager does not support paths referring to
+ * non-default file systems.
+ *
+ * @param fm the file manager to be tested
+ * @param paths the paths to be tested
+ * @return
+ */
+ boolean isGetFileObjectsSupported(StandardJavaFileManager fm, List<Path> paths) {
+ return !(fm instanceof MyStandardJavaFileManager
+ && (paths.get(0).getFileSystem() != FileSystems.getDefault()));
+ }
+
+ /**
+ * Report an error.
+ */
+ void error(String msg) {
+ System.err.println("Error: " + msg);
+ errors++;
+ }
+
+ /** Count of errors reported. */
+ int errors;
+
+}