我最近遇到了一個障礙。我正在編寫一些單元測試來運行我的庫的一些部分,負責代碼生成。代碼接受配置對象的輸入,使用CodeModel編寫代碼,將這些類轉儲爲String,然後使用Java 6編譯器API將它們實時編譯爲可用的類。生成代碼測試合併PowerMock
我碰到的障礙是我在生成的代碼中測試的方法之一,它使用帶有最終方法(Android Bundle)的類,並在使用時拋出異常(「java.lang.RuntimeException:Stub!」)。所以,爲了解決這個問題,我使用PowerMock來模擬最終的方法。然而,在這種情況下,編譯器API會拋出一個NPE。我認爲這是因爲PowerMock在幕後使用的設施,但我不確定。
這裏的例外:
at org.androidtransfuse.gen.classloader.MemoryFileManager.<init>(MemoryFileManager.java:11)
at org.androidtransfuse.gen.classloader.MemoryClassLoader.<init>(MemoryClassLoader.java:16)
at org.androidtransfuse.gen.ParcelableGeneratorTest.setup(ParcelableGeneratorTest.java:67)
at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:39)
at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:25)
at org.junit.internal.runners.MethodRoadie.runBefores(MethodRoadie.java:129)
at org.junit.internal.runners.MethodRoadie.runBeforesThenTestThenAfters(MethodRoadie.java:93)
at org.powermock.modules.junit4.internal.impl.PowerMockJUnit44RunnerDelegateImpl$PowerMockJUnit44MethodRunner.executeTest(PowerMockJUnit44RunnerDelegateImpl.java:296)
MemoryFileManager.java:
class MemoryFileManager extends ForwardingJavaFileManager<JavaFileManager> {
public final Map<String, Output> map = new HashMap<String, Output>();
MemoryFileManager(JavaCompiler compiler) {
super(compiler.getStandardFileManager(null, null, null));
}
@Override
public Output getJavaFileForOutput
(Location location, String name, JavaFileObject.Kind kind, FileObject source) {
Output mc = new Output(name, kind);
this.map.put(name, mc);
return mc;
}
}
MemoryClassLoader.java(使用MemoryFileManager):
public class MemoryClassLoader extends ClassLoader {
private final JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
private final MemoryFileManager manager = new MemoryFileManager(this.compiler);
public void add(String classname, String filecontent) {
add(Collections.singletonMap(classname, filecontent));
}
public void add(Map<String, String> map) {
List<Source> list = new ArrayList<Source>();
for (Map.Entry<String, String> entry : map.entrySet()) {
list.add(new Source(entry.getKey(), JavaFileObject.Kind.SOURCE, entry.getValue()));
}
this.compiler.getTask(null, this.manager, null, null, null, list).call();
}
@Override
protected Class<?> findClass(String name) throws ClassNotFoundException {
synchronized (this.manager) {
Output mc = this.manager.map.remove(name);
if (mc != null) {
byte[] array = mc.toByteArray();
return defineClass(name, array, 0, array.length);
}
}
return super.findClass(name);
}
}
和一個非常基本的測試證明了該問題:
@RunWith(PowerMockRunner.class)
@PrepareForTest(Parcel.class)
public class ExampleTest {
private static final String TEST_VALUE = "test value";
String classToCompile =
"import android.os.Parcel;\n" +
"\n" +
"public class ClassToCompile\n" +
"{\n" +
"\n" +
" private Parcel parcel;\n" +
"}";
@Test
public void test() throws ClassNotFoundException, IOException, NoSuchMethodException, InvocationTargetException, IllegalAccessException, InstantiationException {
MemoryClassLoader classLoader = new MemoryClassLoader();
classLoader.add("ClassToCompile", classToCompile);
classLoader.loadClass("ClassToCompile");
}
}