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 24 /* 25 * @test 26 * @summary Test files copy plugin 27 * @author Jean-Francois Denise 28 * @modules jdk.jlink/jdk.tools.jlink.internal 29 * jdk.jlink/jdk.tools.jlink.builder 30 * jdk.jlink/jdk.tools.jlink.internal.plugins 31 * @run main FileCopierPluginTest 32 */ 33 34 import java.io.File; 35 import java.net.URI; 36 import java.nio.file.Files; 37 import java.nio.file.Path; 38 import java.nio.file.Paths; 39 import java.util.Collections; 40 import java.util.HashMap; 41 import java.util.Map; 42 import jdk.tools.jlink.internal.ResourcePoolManager; 43 import jdk.tools.jlink.builder.DefaultImageBuilder; 44 45 import jdk.tools.jlink.internal.plugins.FileCopierPlugin; 46 import jdk.tools.jlink.plugin.PluginException; 47 import jdk.tools.jlink.plugin.ResourcePoolEntry; 48 import jdk.tools.jlink.plugin.ResourcePool; 49 50 public class FileCopierPluginTest { 51 52 public static void main(String[] args) throws Exception { 53 new FileCopierPluginTest().test(); 54 } 55 56 /** 57 * 3 cases - Absolute, no target ==> copy in image root dir - Absolute and 58 * target ==> copy in image root dir/target - Relative ==> copy from JDK 59 * home dir. 60 * 61 * @throws Exception 62 */ 63 public void test() throws Exception { 64 FileCopierPlugin plug = new FileCopierPlugin(); 65 String content = "You \n should \n be \bthere.\n"; 66 String name = "sample.txt"; 67 File src = new File("src"); 68 src.mkdir(); 69 // Need a fake bin 70 File bin = new File("bin"); 71 bin.mkdir(); 72 73 File txt = new File(src, name); 74 txt.createNewFile(); 75 76 String target = "target" + File.separator + name; 77 Files.write(txt.toPath(), content.getBytes()); 78 File lic = new File(System.getProperty("java.home"), "LICENSE.txt"); 79 StringBuilder builder = new StringBuilder(); 80 int expected = lic.exists() ? 4 : 3; 81 if (lic.exists()) { 82 builder.append("LICENSE.txt,"); 83 } 84 builder.append(txt.getAbsolutePath()+","); 85 builder.append(txt.getAbsolutePath() + "=" + target+","); 86 builder.append(src.getAbsolutePath() + "=src2"); 87 88 Map<String, String> conf = new HashMap<>(); 89 conf.put(FileCopierPlugin.NAME, builder.toString()); 90 plug.configure(conf); 91 ResourcePoolManager poolMgr = new ResourcePoolManager(); 92 // java.base/module-info.class is used to add "release" file 93 // We read it from jrt-fs and add a ResourcePoolEntry 94 poolMgr.add( 95 ResourcePoolEntry.create("/java.base/module-info.class", 96 ResourcePoolEntry.Type.CLASS_OR_RESOURCE, getJavaBaseModuleInfo())); 97 expected++; 98 ResourcePool pool = plug.transform( 99 new ResourcePoolManager().resourcePool(), 100 poolMgr.resourcePoolBuilder()); 101 if (pool.entryCount() != expected) { 102 throw new AssertionError("Wrong number of added files"); 103 } 104 pool.entries().forEach(f -> { 105 if (!f.type().equals(ResourcePoolEntry.Type.OTHER) && 106 !f.type().equals(ResourcePoolEntry.Type.CLASS_OR_RESOURCE)) { 107 throw new AssertionError("Invalid type " + f.type() 108 + " for file " + f.path()); 109 } 110 if (f.content() == null) { 111 throw new AssertionError("Null stream for file " + f.path()); 112 } 113 }); 114 Path root = new File(".").toPath(); 115 DefaultImageBuilder imgbuilder = new DefaultImageBuilder(root); 116 imgbuilder.storeFiles(pool); 117 118 if (lic.exists()) { 119 File license = new File(root.toFile(), "LICENSE.txt"); 120 if (!license.exists() || license.length() == 0) { 121 throw new AssertionError("Invalid license file " 122 + license.getAbsoluteFile()); 123 } 124 } 125 126 File sample1 = new File(root.toFile(), txt.getName()); 127 if (!sample1.exists() || sample1.length() == 0) { 128 throw new AssertionError("Invalide sample1 file " 129 + sample1.getAbsoluteFile()); 130 } 131 if (!new String(Files.readAllBytes(sample1.toPath())).equals(content)) { 132 throw new AssertionError("Invalid Content in sample1"); 133 } 134 135 File sample2 = new File(root.toFile(), target); 136 if (!sample2.exists() || sample2.length() == 0) { 137 throw new AssertionError("Invalide sample2 file " 138 + sample2.getAbsoluteFile()); 139 } 140 if (!new String(Files.readAllBytes(sample2.toPath())).equals(content)) { 141 throw new AssertionError("Invalid Content in sample2"); 142 } 143 144 File src2 = new File(root.toFile(), "src2"); 145 if (!src2.exists() || src2.list().length != 1) { 146 throw new AssertionError("Invalide src2 dir " 147 + src2.getAbsoluteFile()); 148 } 149 File f = src2.listFiles()[0]; 150 if (!f.getName().equals(txt.getName())) { 151 throw new AssertionError("Invalide file name in src2 dir " 152 + f.getAbsoluteFile()); 153 } 154 if (!new String(Files.readAllBytes(f.toPath())).equals(content)) { 155 throw new AssertionError("Invalid Content in src2 dir"); 156 } 157 } 158 159 // read java.base/module-info.class from jrt-fs 160 private static Path getJavaBaseModuleInfo() { 161 return Paths.get(URI.create("jrt:/modules/java.base/module-info.class")); 162 } 163 }