IntegrationTest.java revision 15591:967c9ee04457 
1/*
2 * Copyright (c) 2015, 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
24import java.io.File;
25import java.io.FileReader;
26import java.io.IOException;
27import java.io.UncheckedIOException;
28import java.nio.file.Files;
29import java.nio.file.Path;
30import java.nio.file.Paths;
31import java.util.ArrayList;
32import java.util.Collections;
33import java.util.HashMap;
34import java.util.HashSet;
35import java.util.List;
36import java.util.Map;
37import java.util.Properties;
38import java.util.Set;
39import java.util.function.Function;
40import jdk.tools.jlink.Jlink;
41import jdk.tools.jlink.Jlink.JlinkConfiguration;
42import jdk.tools.jlink.Jlink.PluginsConfiguration;
43import jdk.tools.jlink.builder.DefaultImageBuilder;
44import jdk.tools.jlink.plugin.ResourcePool;
45import jdk.tools.jlink.plugin.ResourcePoolBuilder;
46import jdk.tools.jlink.plugin.Plugin;
47import jdk.tools.jlink.internal.ExecutableImage;
48import jdk.tools.jlink.internal.PostProcessor;
49import jdk.tools.jlink.internal.plugins.DefaultCompressPlugin;
50import jdk.tools.jlink.internal.plugins.StripDebugPlugin;
51
52import tests.Helper;
53import tests.JImageGenerator;
54
55/*
56 * @test
57 * @summary Test integration API
58 * @author Jean-Francois Denise
59 * @library ../lib
60 * @modules java.base/jdk.internal.jimage
61 *          jdk.jdeps/com.sun.tools.classfile
62 *          jdk.jlink/jdk.tools.jlink
63 *          jdk.jlink/jdk.tools.jlink.builder
64 *          jdk.jlink/jdk.tools.jlink.internal
65 *          jdk.jlink/jdk.tools.jlink.internal.plugins
66 *          jdk.jlink/jdk.tools.jmod
67 *          jdk.jlink/jdk.tools.jimage
68 *          jdk.compiler
69 * @build tests.*
70 * @run main IntegrationTest
71 */
72public class IntegrationTest {
73
74    private static final List<Integer> ordered = new ArrayList<>();
75
76    public static class MyPostProcessor implements PostProcessor, Plugin {
77
78        public static final String NAME = "mypostprocessor";
79
80        @Override
81        public List<String> process(ExecutableImage image) {
82            try {
83                Files.createFile(image.getHome().resolve("toto.txt"));
84                return null;
85            } catch (IOException ex) {
86                throw new UncheckedIOException(ex);
87            }
88        }
89
90        @Override
91        public String getName() {
92            return NAME;
93        }
94
95        @Override
96        public Category getType() {
97            return Category.PROCESSOR;
98        }
99
100        @Override
101        public void configure(Map<String, String> config) {
102            throw new UnsupportedOperationException("Shouldn't be called");
103        }
104
105        @Override
106        public ResourcePool transform(ResourcePool in, ResourcePoolBuilder out) {
107            in.transformAndCopy(Function.identity(), out);
108            return out.build();
109        }
110    }
111
112    public static void main(String[] args) throws Exception {
113
114        Helper helper = Helper.newHelper();
115        if (helper == null) {
116            System.err.println("Test not run");
117            return;
118        }
119        apitest();
120        test();
121    }
122
123    private static void apitest() throws Exception {
124        boolean failed = false;
125        Jlink jl = new Jlink();
126
127        try {
128            jl.build(null);
129            failed = true;
130        } catch (Exception ex) {
131            // XXX OK
132        }
133        if (failed) {
134            throw new Exception("Should have failed");
135        }
136        System.out.println(jl);
137
138        JlinkConfiguration config
139                = new JlinkConfiguration(null, null, null, null);
140
141        System.out.println(config);
142
143        Plugin p = Jlink.newPlugin("toto", Collections.emptyMap(), null);
144        if (p != null) {
145            throw new Exception("Plugin should be null");
146        }
147
148        Plugin p2 = Jlink.newPlugin("compress", Collections.emptyMap(), null);
149        if (p2 == null) {
150            throw new Exception("Plugin should not be null");
151        }
152    }
153
154    private static void test() throws Exception {
155        Jlink jlink = new Jlink();
156        Path output = Paths.get("integrationout");
157        List<Path> modulePaths = new ArrayList<>();
158        File jmods
159                = JImageGenerator.getJModsDir(new File(System.getProperty("test.jdk")));
160        modulePaths.add(jmods.toPath());
161        Set<String> mods = new HashSet<>();
162        mods.add("java.management");
163        Set<String> limits = new HashSet<>();
164        limits.add("java.management");
165        JlinkConfiguration config = new Jlink.JlinkConfiguration(output,
166                modulePaths, mods, limits, null);
167
168        List<Plugin> lst = new ArrayList<>();
169
170        //Strip debug
171        {
172            Map<String, String> config1 = new HashMap<>();
173            config1.put(StripDebugPlugin.NAME, "");
174            Plugin strip = Jlink.newPlugin("strip-debug", config1, null);
175            lst.add(strip);
176        }
177        // compress
178        {
179            Map<String, String> config1 = new HashMap<>();
180            config1.put(DefaultCompressPlugin.NAME, "2");
181            Plugin compress
182                    = Jlink.newPlugin("compress", config1, null);
183            lst.add(compress);
184        }
185        // Post processor
186        {
187            lst.add(new MyPostProcessor());
188        }
189        // Image builder
190        DefaultImageBuilder builder = new DefaultImageBuilder(output);
191        PluginsConfiguration plugins
192                = new Jlink.PluginsConfiguration(lst, builder, null);
193
194        jlink.build(config, plugins);
195
196        if (!Files.exists(output)) {
197            throw new AssertionError("Directory not created");
198        }
199        File jimage = new File(output.toString(), "lib" + File.separator + "modules");
200        if (!jimage.exists()) {
201            throw new AssertionError("jimage not generated");
202        }
203        File release = new File(output.toString(), "release");
204        if (!release.exists()) {
205            throw new AssertionError("release not generated");
206        }
207
208        Properties props = new Properties();
209        try (FileReader reader = new FileReader(release)) {
210            props.load(reader);
211        }
212
213        checkReleaseProperty(props, "JAVA_VERSION");
214        checkReleaseProperty(props, "JAVA_FULL_VERSION");
215        checkReleaseProperty(props, "OS_NAME");
216        checkReleaseProperty(props, "OS_ARCH");
217        checkReleaseProperty(props, "OS_VERSION");
218
219        if (!Files.exists(output.resolve("toto.txt"))) {
220            throw new AssertionError("Post processing not called");
221        }
222
223    }
224
225    static void checkReleaseProperty(Properties props, String name) {
226        if (! props.containsKey(name)) {
227            throw new AssertionError("release file does not contain property : " + name);
228        }
229
230        // property value is of min. length 3 and double quoted at the ends.
231        String value = props.getProperty(name);
232        if (value.length() < 3 ||
233            value.charAt(0) != '"' ||
234            value.charAt(value.length() - 1) != '"') {
235            throw new AssertionError("release property " + name + " is not quoted property");
236        }
237    }
238}
239