1 /*

     2  * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.

     8  *

     9  * This code is distributed in the hope that it will be useful, but WITHOUT

    10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    12  * version 2 for more details (a copy is included in the LICENSE file that

    13  * accompanied this code).

    14  *

    15  * You should have received a copy of the GNU General Public License version

    16  * 2 along with this work; if not, write to the Free Software Foundation,

    17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    18  *

    19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    20  * or visit www.oracle.com if you need additional information or have any

    21  * questions.

    22  */

    23 

    24 /**

    25  * @test

    26  * @bug 8178518

    27  * @summary Add method JavaFileManager.contains

    28  * @library /tools/lib

    29  * @modules

    30  *      jdk.compiler/com.sun.tools.javac.api

    31  *      jdk.compiler/com.sun.tools.javac.main

    32  * @build toolbox.ToolBox toolbox.JarTask toolbox.JavacTask ModuleTestBase

    33  * @run main ContainsTest

    34  */

    35 

    36 import java.io.IOException;

    37 import java.nio.file.FileSystem;

    38 import java.nio.file.FileSystems;

    39 import java.nio.file.Files;

    40 import java.nio.file.Path;

    41 import java.nio.file.Paths;

    42 import java.util.EnumSet;

    43 import java.util.List;

    44 

    45 import javax.tools.FileObject;

    46 import javax.tools.JavaCompiler;

    47 import javax.tools.JavaFileManager.Location;

    48 import javax.tools.JavaFileObject;

    49 import javax.tools.StandardJavaFileManager;

    50 import javax.tools.StandardLocation;

    51 import javax.tools.ToolProvider;

    52 

    53 import toolbox.JarTask;

    54 import toolbox.JavacTask;

    55 

    56 public class ContainsTest extends ModuleTestBase {

    57     public static void main(String... args) throws Exception {

    58         ContainsTest t = new ContainsTest();

    59         t.runTests();

    60     }

    61 

    62     JavaCompiler javaCompiler = ToolProvider.getSystemJavaCompiler();

    63 

    64     @Test

    65     public void testSimplePath(Path base) throws IOException {

    66         // Test that we can look up in directories in the default file system.

    67         Path src = base.resolve("src");

    68         tb.writeJavaFiles(src, "package p; class C { }");

    69         Path c = src.resolve("p/C.java");

    70         Path x = base.resolve("src2/p/C.java");

    71         try (StandardJavaFileManager fm = javaCompiler.getStandardFileManager(null, null, null)) {

    72             fm.setLocationFromPaths(StandardLocation.SOURCE_PATH, List.of(src));

    73             checkContains(fm, StandardLocation.SOURCE_PATH, c, true);

    74             checkContains(fm, StandardLocation.SOURCE_PATH, x, false);

    75         }

    76     }

    77 

    78     @Test

    79     public void testJarPath(Path base) throws IOException {

    80         // Test that we can look up in jar files on a search path.

    81         // In this case, the path we look up must come from open file system

    82         // as used by the file manager.

    83         Path src = base.resolve("src");

    84         tb.writeJavaFiles(src, "package p; class C { }");

    85         Path classes = Files.createDirectories(base.resolve("classes"));

    86         new JavacTask(tb)

    87             .options("-sourcepath", src.toString())

    88             .outdir(classes)

    89             .files(findJavaFiles(src))

    90             .run()

    91             .writeAll();

    92 

    93         Path jar = base.resolve("classes.jar");

    94         new JarTask(tb).run("cf", jar.toString(), "-C", classes.toString(), "p");

    95 

    96         Path c = src.resolve("p/C.java");

    97         Path x = base.resolve("src2/p/C.java");

    98 

    99         try (StandardJavaFileManager fm = javaCompiler.getStandardFileManager(null, null, null)) {

   100             fm.setLocationFromPaths(StandardLocation.CLASS_PATH, List.of(src, jar));

   101 

   102             checkContains(fm, StandardLocation.CLASS_PATH, c, true);

   103             checkContains(fm, StandardLocation.CLASS_PATH, x, false);

   104 

   105             JavaFileObject fo = fm.list(StandardLocation.CLASS_PATH, "p",

   106                     EnumSet.of(JavaFileObject.Kind.CLASS), false).iterator().next();

   107 

   108             checkContains(fm, StandardLocation.CLASS_PATH, fo, true);

   109         }

   110     }

   111 

   112     @Test

   113     public void testJarFSPath(Path base) throws IOException {

   114         // Test that we can look up in non-default file systems on the search path,

   115         // such as an open jar file system.

   116         Path src = base.resolve("src");

   117         tb.writeJavaFiles(src, "package p; class C { }");

   118         Path classes = Files.createDirectories(base.resolve("classes"));

   119         new JavacTask(tb)

   120             .options("-sourcepath", src.toString())

   121             .outdir(classes)

   122             .files(findJavaFiles(src))

   123             .run()

   124             .writeAll();

   125 

   126         Path jar = base.resolve("classes.jar");

   127         new JarTask(tb).run("cf", jar.toString(), "-C", classes.toString(), "p");

   128 

   129         Path c = src.resolve("p/C.java");

   130         Path x = base.resolve("src2/p/C.java");

   131 

   132         try (FileSystem jarFS = FileSystems.newFileSystem(jar, null);

   133                 StandardJavaFileManager fm = javaCompiler.getStandardFileManager(null, null, null)) {

   134             Path jarRoot = jarFS.getRootDirectories().iterator().next();

   135             fm.setLocationFromPaths(StandardLocation.CLASS_PATH, List.of(src, jarRoot));

   136 

   137             checkContains(fm, StandardLocation.CLASS_PATH, c, true);

   138             checkContains(fm, StandardLocation.CLASS_PATH, x, false);

   139 

   140             JavaFileObject fo = fm.list(StandardLocation.CLASS_PATH, "p",

   141                     EnumSet.of(JavaFileObject.Kind.CLASS), false).iterator().next();

   142 

   143             checkContains(fm, StandardLocation.CLASS_PATH, fo, true);

   144             checkContains(fm, StandardLocation.CLASS_PATH, jarRoot.resolve("p/C.class"), true);

   145         }

   146     }

   147 

   148     void checkContains(StandardJavaFileManager fm, Location l, Path p, boolean expect) throws IOException {

   149         JavaFileObject fo = fm.getJavaFileObjects(p).iterator().next();

   150         checkContains(fm, l, fo, expect);

   151     }

   152 

   153     void checkContains(StandardJavaFileManager fm, Location l, FileObject fo, boolean expect) throws IOException {

   154         boolean found = fm.contains(l, fo);

   155         if (found) {

   156             if (expect) {

   157                 out.println("file found, as expected: " + l + " " + fo.getName());

   158             } else {

   159                 error("file not found: " + l + " " + fo.getName());

   160             }

   161         } else {

   162             if (expect) {

   163                 error("file found unexpectedly: " + l + " " + fo.getName());

   164             } else {

   165                 out.println("file not found, as expected: " + l + " " + fo.getName());

   166             }

   167         }

   168     }

   169 }