1 /*
   2  * Copyright (c) 2016, 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 8160489
  27  * @summary tests for --patch-modules
  28  * @library /tools/lib
  29  * @modules
  30  *      jdk.compiler/com.sun.tools.javac.api
  31  *      jdk.compiler/com.sun.tools.javac.file
  32  *      jdk.compiler/com.sun.tools.javac.main
  33  * @build toolbox.ToolBox toolbox.JavacTask toolbox.ModuleBuilder ModuleTestBase
  34  * @run main PatchModulesTest
  35  */
  36 
  37 import java.io.File;
  38 import java.io.IOException;
  39 import java.io.PrintWriter;
  40 import java.io.StringWriter;
  41 import java.lang.reflect.Field;
  42 import java.nio.file.Path;
  43 import java.nio.file.Paths;
  44 import java.util.List;
  45 import java.util.Map;
  46 import java.util.stream.Collectors;
  47 
  48 import javax.tools.JavaFileObject;
  49 import javax.tools.ToolProvider;
  50 
  51 import com.sun.source.util.JavacTask;
  52 import com.sun.tools.javac.api.JavacTool;
  53 import com.sun.tools.javac.file.BaseFileManager;
  54 import com.sun.tools.javac.file.JavacFileManager;
  55 import com.sun.tools.javac.file.Locations;
  56 
  57 import static java.util.Arrays.asList;
  58 
  59 
  60 public class PatchModulesTest extends ModuleTestBase {
  61 
  62     public static void main(String... args) throws Exception {
  63         PatchModulesTest t = new PatchModulesTest();
  64         t.init();
  65         t.runTests();
  66     }
  67 
  68     private static String PS = File.pathSeparator;
  69 
  70     void init() throws IOException {
  71         tb.createDirectories("a", "b", "c", "d", "e");
  72         tb.writeJavaFiles(Paths.get("."), "class C { }");
  73     }
  74 
  75     @Test
  76     public void testSimple(Path base) throws Exception {
  77         test(asList("java.base=a"),
  78             "{java.base=[a]}");
  79     }
  80 
  81     @Test
  82     public void testPair(Path base) throws Exception {
  83         test(asList("java.base=a", "java.compiler=b"),
  84             "{java.base=[a], java.compiler=[b]}");
  85     }
  86 
  87     @Test
  88     public void testMultiple(Path base) throws Exception {
  89         test(asList("java.base=a:b"),
  90             "{java.base=[a, b]}");
  91     }
  92 
  93     @Test
  94     public void testLastOneWins(Path base) throws Exception {
  95         test(asList("java.base=a", "java.compiler=b", "java.base=c"),
  96             "{java.base=[c], java.compiler=[b]}");
  97     }
  98 
  99     void test(List<String> patches, String expect) throws Exception {
 100         JavacTool tool = (JavacTool) ToolProvider.getSystemJavaCompiler();
 101         StringWriter sw = new StringWriter();
 102         try (PrintWriter pw = new PrintWriter(sw)) {
 103             JavacFileManager fm = tool.getStandardFileManager(null, null, null);
 104             List<String> opts = patches.stream()
 105                 .map(p -> "--patch-module=" + p.replace(":", PS))
 106                 .collect(Collectors.toList());
 107             Iterable<? extends JavaFileObject> files = fm.getJavaFileObjects("C.java");
 108             JavacTask task = tool.getTask(pw, fm, null, opts, null, files);
 109 
 110             Field locationsField = BaseFileManager.class.getDeclaredField("locations");
 111             locationsField.setAccessible(true);
 112             Object locations = locationsField.get(fm);
 113 
 114             Field patchMapField = Locations.class.getDeclaredField("patchMap");
 115             patchMapField.setAccessible(true);
 116             Map<?,?> patchMap = (Map<?,?>) patchMapField.get(locations);
 117             String found = patchMap.toString();
 118 
 119             if (!found.equals(expect)) {
 120                 tb.out.println("Expect: " + expect);
 121                 tb.out.println("Found:  " + found);
 122                 error("output not as expected");
 123             }
 124         }
 125     }
 126 }
 127