149 lines
5.5 KiB
Java
149 lines
5.5 KiB
Java
|
/*
|
||
|
* Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.
|
||
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
||
|
*
|
||
|
* This code is free software; you can redistribute it and/or modify it
|
||
|
* under the terms of the GNU General Public License version 2 only, as
|
||
|
* published by the Free Software Foundation.
|
||
|
*
|
||
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
||
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||
|
* version 2 for more details (a copy is included in the LICENSE file that
|
||
|
* accompanied this code).
|
||
|
*
|
||
|
* You should have received a copy of the GNU General Public License version
|
||
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
||
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
||
|
*
|
||
|
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
||
|
* or visit www.oracle.com if you need additional information or have any
|
||
|
* questions.
|
||
|
*/
|
||
|
|
||
|
/*
|
||
|
* @test
|
||
|
* @summary Test files copy plugin
|
||
|
* @author Jean-Francois Denise
|
||
|
* @modules jdk.jlink/jdk.tools.jlink.internal
|
||
|
* jdk.jlink/jdk.tools.jlink.internal.plugins
|
||
|
* @run main FileCopierPluginTest
|
||
|
*/
|
||
|
|
||
|
import java.io.File;
|
||
|
import java.nio.file.Files;
|
||
|
import java.nio.file.Path;
|
||
|
import java.util.Collections;
|
||
|
import java.util.HashMap;
|
||
|
import java.util.Map;
|
||
|
import jdk.tools.jlink.internal.PoolImpl;
|
||
|
import jdk.tools.jlink.builder.DefaultImageBuilder;
|
||
|
|
||
|
import jdk.tools.jlink.internal.plugins.FileCopierPlugin;
|
||
|
import jdk.tools.jlink.plugin.Pool;
|
||
|
import jdk.tools.jlink.plugin.Pool.ModuleData;
|
||
|
import jdk.tools.jlink.plugin.Pool.ModuleDataType;
|
||
|
|
||
|
public class FileCopierPluginTest {
|
||
|
|
||
|
public static void main(String[] args) throws Exception {
|
||
|
new FileCopierPluginTest().test();
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* 3 cases - Absolute, no target ==> copy in image root dir - Absolute and
|
||
|
* target ==> copy in image root dir/target - Relative ==> copy from JDK
|
||
|
* home dir.
|
||
|
*
|
||
|
* @throws Exception
|
||
|
*/
|
||
|
public void test() throws Exception {
|
||
|
FileCopierPlugin plug = new FileCopierPlugin();
|
||
|
String content = "You \n should \n be \bthere.\n";
|
||
|
String name = "sample.txt";
|
||
|
File src = new File("src");
|
||
|
src.mkdir();
|
||
|
// Need a fake bin
|
||
|
File bin = new File("bin");
|
||
|
bin.mkdir();
|
||
|
|
||
|
File txt = new File(src, name);
|
||
|
txt.createNewFile();
|
||
|
|
||
|
String target = "target" + File.separator + name;
|
||
|
Files.write(txt.toPath(), content.getBytes());
|
||
|
File lic = new File(System.getProperty("java.home"), "LICENSE");
|
||
|
StringBuilder builder = new StringBuilder();
|
||
|
int expected = lic.exists() ? 4 : 3;
|
||
|
if (lic.exists()) {
|
||
|
builder.append("LICENSE,");
|
||
|
}
|
||
|
builder.append(txt.getAbsolutePath()+",");
|
||
|
builder.append(txt.getAbsolutePath() + "=" + target+",");
|
||
|
builder.append(src.getAbsolutePath() + "=src2");
|
||
|
|
||
|
Map<String, String> conf = new HashMap<>();
|
||
|
conf.put(FileCopierPlugin.NAME, builder.toString());
|
||
|
plug.configure(conf);
|
||
|
Pool pool = new PoolImpl();
|
||
|
plug.visit(new PoolImpl(), pool);
|
||
|
if (pool.getContent().size() != expected) {
|
||
|
throw new AssertionError("Wrong number of added files");
|
||
|
}
|
||
|
for (ModuleData f : pool.getContent()) {
|
||
|
if (!f.getType().equals(ModuleDataType.OTHER)) {
|
||
|
throw new AssertionError("Invalid type " + f.getType()
|
||
|
+ " for file " + f.getPath());
|
||
|
}
|
||
|
if (f.stream() == null) {
|
||
|
throw new AssertionError("Null stream for file " + f.getPath());
|
||
|
}
|
||
|
|
||
|
}
|
||
|
Path root = new File(".").toPath();
|
||
|
DefaultImageBuilder imgbuilder = new DefaultImageBuilder(false,
|
||
|
root);
|
||
|
imgbuilder.storeFiles(pool, "");
|
||
|
|
||
|
if (lic.exists()) {
|
||
|
File license = new File(root.toFile(), "LICENSE");
|
||
|
if (!license.exists() || license.length() == 0) {
|
||
|
throw new AssertionError("Invalide license file "
|
||
|
+ license.getAbsoluteFile());
|
||
|
}
|
||
|
}
|
||
|
|
||
|
File sample1 = new File(root.toFile(), txt.getName());
|
||
|
if (!sample1.exists() || sample1.length() == 0) {
|
||
|
throw new AssertionError("Invalide sample1 file "
|
||
|
+ sample1.getAbsoluteFile());
|
||
|
}
|
||
|
if (!new String(Files.readAllBytes(sample1.toPath())).equals(content)) {
|
||
|
throw new AssertionError("Invalid Content in sample1");
|
||
|
}
|
||
|
|
||
|
File sample2 = new File(root.toFile(), target);
|
||
|
if (!sample2.exists() || sample2.length() == 0) {
|
||
|
throw new AssertionError("Invalide sample2 file "
|
||
|
+ sample2.getAbsoluteFile());
|
||
|
}
|
||
|
if (!new String(Files.readAllBytes(sample2.toPath())).equals(content)) {
|
||
|
throw new AssertionError("Invalid Content in sample2");
|
||
|
}
|
||
|
|
||
|
File src2 = new File(root.toFile(), "src2");
|
||
|
if (!src2.exists() || src2.list().length != 1) {
|
||
|
throw new AssertionError("Invalide src2 dir "
|
||
|
+ src2.getAbsoluteFile());
|
||
|
}
|
||
|
File f = src2.listFiles()[0];
|
||
|
if (!f.getName().equals(txt.getName())) {
|
||
|
throw new AssertionError("Invalide file name in src2 dir "
|
||
|
+ f.getAbsoluteFile());
|
||
|
}
|
||
|
if (!new String(Files.readAllBytes(f.toPath())).equals(content)) {
|
||
|
throw new AssertionError("Invalid Content in src2 dir");
|
||
|
}
|
||
|
}
|
||
|
}
|