/
MultiReleaseTestBase.java
136 lines (118 loc) · 5.21 KB
/
MultiReleaseTestBase.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
/*
* Copyright (c) 2023 Contributors to the Eclipse Foundation.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v. 2.0, which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* This Source Code may also be made available under the following Secondary
* Licenses when the conditions for such availability set forth in the
* Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
* version 2 with the GNU Classpath Exception, which is available at
* https://www.gnu.org/software/classpath/license.html.
*
* SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
*/
package org.glassfish.main.test.app.web.mrjar;
import org.glassfish.main.itest.tools.GlassFishTestEnvironment;
import org.glassfish.main.itest.tools.asadmin.Asadmin;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.asset.Asset;
import org.jboss.shrinkwrap.api.asset.ByteArrayAsset;
import org.jboss.shrinkwrap.api.spec.JavaArchive;
import org.objectweb.asm.ClassWriter;
import org.objectweb.asm.MethodVisitor;
import org.objectweb.asm.Type;
import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.util.jar.Manifest;
import static java.util.jar.Attributes.Name.MANIFEST_VERSION;
import static java.util.jar.Attributes.Name.MULTI_RELEASE;
import static org.objectweb.asm.Opcodes.ACC_PUBLIC;
import static org.objectweb.asm.Opcodes.ACC_SUPER;
import static org.objectweb.asm.Opcodes.ALOAD;
import static org.objectweb.asm.Opcodes.ARETURN;
import static org.objectweb.asm.Opcodes.INVOKESPECIAL;
import static org.objectweb.asm.Opcodes.RETURN;
import static org.objectweb.asm.Opcodes.V11;
import static org.objectweb.asm.Opcodes.V17;
import static org.objectweb.asm.Opcodes.V1_8;
public class MultiReleaseTestBase {
private static final String LIB_FILE_NAME = "mrlib.jar";
protected static final Asadmin ASADMIN = GlassFishTestEnvironment.getAsadmin();
protected String readResponse(HttpURLConnection connection) throws Exception {
StringBuilder response = new StringBuilder();
try (BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream()))) {
String line;
while ((line = reader.readLine()) != null) {
response.append(line);
}
}
return response.toString();
}
protected static JavaArchive createMultiReleaseLibrary() throws IOException {
return ShrinkWrap.create(JavaArchive.class, LIB_FILE_NAME)
.setManifest(generateManifest())
.addClass(Version.class)
.add(generateVersionClass(V1_8), packageJarEntryFor(VersionImpl.class, V1_8), classFileNameFor(VersionImpl.class))
.add(generateVersionClass(V11), packageJarEntryFor(VersionImpl.class, V11), classFileNameFor(VersionImpl.class))
.add(generateVersionClass(V17), packageJarEntryFor(VersionImpl.class, V17), classFileNameFor(VersionImpl.class));
}
private static String classFileNameFor(Class<?> c) {
return c.getSimpleName() + ".class";
}
private static String packageJarEntryFor(Class<?> c, int version) {
StringBuilder jarTarget = new StringBuilder();
switch (version) {
case V11:
jarTarget.append("META-INF/versions/11/");
break;
case V17:
jarTarget.append("META-INF/versions/17/");
break;
default:
break;
}
jarTarget.append(c.getPackageName().replace('.', '/'));
return jarTarget.toString();
}
private static Asset generateManifest() throws IOException {
Manifest manifest = new Manifest();
manifest.getMainAttributes().put(MANIFEST_VERSION, "1.0");
manifest.getMainAttributes().put(MULTI_RELEASE, String.valueOf(Boolean.TRUE));
try (ByteArrayOutputStream out = new ByteArrayOutputStream()) {
manifest.write(out);
return new ByteArrayAsset(out.toByteArray());
}
}
private static Asset generateVersionClass(int version) {
ClassWriter cw = new ClassWriter(0);
cw.visit(version,
ACC_PUBLIC + ACC_SUPER,
Type.getInternalName(VersionImpl.class),
null,
"java/lang/Object",
new String[] {Type.getInternalName(Version.class)});
// Generate default constructor
MethodVisitor mv = cw.visitMethod(ACC_PUBLIC, "<init>", "()V", null, null);
mv.visitCode();
mv.visitVarInsn(ALOAD, 0);
mv.visitMethodInsn(INVOKESPECIAL, "java/lang/Object", "<init>", "()V", false);
mv.visitInsn(RETURN);
mv.visitMaxs(1, 1);
mv.visitEnd();
// Generate actual getVersion() method
mv = cw.visitMethod(ACC_PUBLIC, "getVersion", "()Ljava/lang/String;", null, null);
mv.visitCode();
mv.visitVarInsn(ALOAD, 0);
mv.visitLdcInsn(String.valueOf(version));
mv.visitInsn(ARETURN);
mv.visitMaxs(2, 1);
mv.visitEnd();
cw.visitEnd();
return new ByteArrayAsset(cw.toByteArray());
}
}