2

我想在方法代码中添加说明。这些指令应该在到达方法之后和离开方法之前执行。为了确保在离开之前总是执行后面的指令，我想将它们放在 finally 块中。（我知道该类AdviceAdapter，但它不能确保在调用的方法引发异常时执行退出代码。）

我的问题是结果中的说明顺序错误。

处理方法：

@Test
public void original() {
    assertTrue(true);
    assertTrue(!(false));
}

期望的结果：

@Test
public void desired() {
    //some logging X

    try {
        assertTrue(true);
        assertTrue(!(false));
    }
    finally {
        //some logging Y
    }
}

（记录 X 也可以发生在 try 块的第一行。）

（期望结果的字节码等于以下 Java 代码的字节码：）

@Test
public void desired() {
    //some logging X

    try {
        assertTrue(true);
        assertTrue(!(false));
        //some logging Y
    }
    catch (Throwable t) {
        //some logging Y
        throw t;
    }
}

我使用 ASM 处理方法的代码：

@Override
public void visitCode() {
    before();

    super.visitCode();

    after();
}

private void before() {
    insertInstructionToSetMode(LoggingMode.TESTING);

    this.l0 = new Label();
    this.l1 = new Label();
    visitLabel(l0);
}

private void after() {
    visitTryCatchBlock(l0, l1, l1, null);
    Label l2 = new Label();
    visitJumpInsn(GOTO, l2);
    visitLabel(this.l1);
    visitFrame(Opcodes.F_SAME1, 0, null, 1, new Object[] {"java/lang/Throwable"});
    visitVarInsn(ASTORE, 1);

    insertInstructionToSetMode(LoggingMode.FRAMING);

    visitVarInsn(ALOAD, 1);
    visitInsn(ATHROW);
    visitLabel(l2);
    visitFrame(Opcodes.F_SAME, 0, null, 0, null);

    insertInstructionToSetMode(LoggingMode.FRAMING);
}

private void insertInstructionToSetMode(LoggingMode mode) {
    String modeValue = (mode == LoggingMode.TESTING ? FIELD_NAME_TESTING : FIELD_NAME_FRAMING);

    visitFieldInsn(Opcodes.GETSTATIC, CP_LOGGING_MODE, modeValue, FIELD_DESC_LOGGING_MODE);
    visitMethodInsn(INVOKESTATIC, CP_INVOCATION_LOGGER, METHOD_NAME_SET_MODE, METHOD_DESC_SET_MODE);
}

生成的字节码（指令顺序错误）：

// logging X
01 getstatic instrumentation/LoggingMode/TESTING Linstrumentation/LoggingMode;
02 invokestatic instrumentation/InvocationLogger/setMode(Linstrumentation/LoggingMode;)V

// successfully passed the try block
03 goto 9

// catch block for the finally behaviour
04 astore_1
05 getstatic instrumentation/LoggingMode/FRAMING Linstrumentation/LoggingMode;
06 invokestatic instrumentation/InvocationLogger/setMode(Linstrumentation/LoggingMode;)V
07 aload_1
08 athrow

// logging Y
09 getstatic instrumentation/LoggingMode/FRAMING Linstrumentation/LoggingMode;
10 invokestatic instrumentation/InvocationLogger/setMode(Linstrumentation/LoggingMode;)V

// original code
11 iconst_1
12 invokestatic org/junit/Assert/assertTrue(Z)V
13 iconst_1
14 invokestatic org/junit/Assert/assertTrue(Z)V
15 return

01-02 很好，但是09-10 需要在原始代码（14）之后，但在返回指令之前。 11-14 需要在 03 之前。

4

3 回答 3

2

我不确定您的方法中的错误在哪里。但是在使用 AdviceAdapter 进行了一些试验和错误之后，我实现了这样的目标。

看

http://code.google.com/p/pitestrunner/source/browse/pitest/src/main/java/org/pitest/coverage/codeassist/CoverageMethodVisitor.java

于 2013-06-26T16:34:22.390 回答
0

警告：此解决方案仅在方法仅包含一个返回指令时才有效（例如：仅在抛出异常时无效）。请参阅：在 try-finally 块中嵌入方法的现有代码 (2)

visitCode我发现了问题：调用时没有在方法中插入现有代码super.visitCode。此方法在超类中为空。这清楚地表明现有代码是在其他地方添加的。

解决方案： 我在方法中调用我的方法before（为需要在开头的新行添加代码）visitCode。如果操作码是返回语句，我会调用afterin 。visitVarInsn

@Override
public void visitCode()
{
    before();
}

@Override
public void visitInsn(int opcode)
{
    if (OpcodesUtil.isXRETURN(opcode))
    {
        after();
    }

    super.visitInsn(opcode);
}

（这AdviceAdapter也有效，但是在确保使用调用everyClassReader的方法时存在一些问题。此外，它可能会建议更多的退出点，并且在恰好关闭一个 try 块时不起作用。）acceptEXPAND_FRAMES

于 2013-06-27T08:30:15.203 回答
0

您可以将 JUnit 注释放在应该在测试方法之前和之后调用的方法上@Before。@After

于 2013-06-25T07:59:05.500 回答