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:+open
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 testDuplicates(Path base) throws Exception {
95 test(asList("java.base=a", "java.compiler=b", "java.base=c"),
96 false, "--patch-module specified more than once for java.base");
97 }
98
99 @Test
100 public void testEmpty(Path base) throws Exception {
101 test(asList(""),
102 false, "no value for --patch-module option");
103 }
104
105 @Test
106 public void testInvalid(Path base) throws Exception {
107 test(asList("java.base/java.lang=."),
108 false, "bad value for --patch-module option: 'java.base/java.lang=.'");
109 }
110
111 void test(List<String> patches, String expect) throws Exception {
112 test(patches, true, expect);
113 }
114
115 void test(List<String> patches, boolean expectOK, String expect) throws Exception {
116 JavacTool tool = (JavacTool) ToolProvider.getSystemJavaCompiler();
117 StringWriter sw = new StringWriter();
118 try (PrintWriter pw = new PrintWriter(sw)) {
119 JavacFileManager fm = tool.getStandardFileManager(null, null, null);
120 List<String> opts = patches.stream()
121 .map(p -> "--patch-module=" + p.replace(":", PS))
122 .collect(Collectors.toList());
123 Iterable<? extends JavaFileObject> files = fm.getJavaFileObjects("C.java");
124 JavacTask task = tool.getTask(pw, fm, null, opts, null, files);
125
126 Field locationsField = BaseFileManager.class.getDeclaredField("locations");
127 locationsField.setAccessible(true);
128 Object locations = locationsField.get(fm);
129
130 Field patchMapField = Locations.class.getDeclaredField("patchMap");
131 patchMapField.setAccessible(true);
132 Map<?,?> patchMap = (Map<?,?>) patchMapField.get(locations);
133 String found = patchMap.toString();
134
135 if (!found.equals(expect)) {
136 tb.out.println("Expect: " + expect);
137 tb.out.println("Found: " + found);
138 error("output not as expected");
139 }
140 } catch (IllegalArgumentException e) {
141 if (expectOK) {
142 error("unexpected exception: " + e);
143 throw e;
144 }
145 String found = e.getMessage();
146 if (!found.equals(expect)) {
147 tb.out.println("Expect: " + expect);
148 tb.out.println("Found: " + found);
149 error("output not as expected");
150 }
151 }
152 }
153 }
154