Как проверить ClassFileTransformer / javaagent? - PullRequest
Новая
       60

Как проверить ClassFileTransformer / javaagent?

6 голосов
/ 05 июля 2011

Я реализовал ClassFileTransformer для javaagent, используя ASM. Поскольку в нем есть ошибки, я хочу написать для него тестовый пример JUnit. Как мне это сделать?

Используя псевдокод, я подумал: 

// Have a test class as subject
public static class Subject {
  public void doSomething(){...}
}
// Manually load and transform the subject
...?
// Normally execute some now transformed methods of the subject
new Subject().doSomething();
// Check the result of the call (i.e. whether the correct attached methods were called)
Assert.assertTrue(MyClassFileTransformer.wasCalled());

Теперь вопрос: как мне вручную загрузить и преобразовать тему и заставить JVM / Classloader использовать мою манипулированную версию? Или я что-то полностью пропускаю?

1 Ответ

6 голосов
/ 07 июля 2011

Я понял. Нужно реализовать собственный ClassLoader, который выполняет то же преобразование с испытуемым, что и ClassFileTransformer (например, вызывает его). И, конечно, предметный класс может быть еще не загружен, поэтому его использование может быть невозможно. Поэтому я использовал API отражения Java для выполнения методов класса объекта.

В отдельном файле: 

public static class Subject {
    public void doSomething(){...}
}

В тесте: 

private static class TransformingClassLoader extends ClassLoader {

    private final String className;

    public TransformingClassLoader(String className) {
        super();
        this.className = className;
    }

    @Override
    public Class<?> loadClass(String name) throws ClassNotFoundException {
        if (name.equals(className)) {
            byte[] byteBuffer = instrumentByteCode(fullyQualifiedSubjectClass);
            return defineClass(className, byteBuffer, 0, byteBuffer.length);
        }
        return super.loadClass(name);
    }
}

@Test
public void testSubject(){
    ClassLoader classLoader = new TransformingClassLoader(fullyQualifiedSubjectClass);
    Class<?> subjectClass = classLoader.loadClass(fullyQualifiedSubjectClass);
    Constructor<?> constructor = subjectClass.getConstructor();
    Object subject = constructor.newInstance();
    Method doSomething = subjectClass.getMethod("doSomething");
    doSomething.invoke(subject);
    Assert.assertTrue(MyClassFileTransformer.wasCalled());
}
...