针对 AspectJ 的 JUnit 测试

java junit mockito spring-aop aspectj

2022-09-01 19:17:00

我正在尝试为自定义方面编写 Junit 测试。下面是 Aspect 类代码段：

@Aspect
@Component
public class SampleAspect {

    private static Logger log = LoggerFactory.getLogger(SampleAspect.class);

    @Around("execution(* org.springframework.data.mongodb.core.MongoOperations.*(..)) || execution(* org.springframework.web.client.RestOperations.*(..))")
    public Object intercept(final ProceedingJoinPoint point) throws Throwable {
        logger.info("invoked Cutom aspect");
         return point.proceed();

    }

}

因此，只要关节点与切入点匹配，上述方面就会截获。它工作正常。

但我的问题是如何对该类进行单元测试。我有以下朱尼特测试：

@Test(expected = MongoTimeoutException.class)
    public void TestWithMongoTemplate() {
        //MongoDocument class
        TestDocument test = new TestDocument();

        ApplicationContext ctx = new AnnotationConfigApplicationContext(TestMongoConfigurationMain.class);
        MongoTemplate mongoTemplate = ctx.getBean(MongoTemplate.class);

        //this call is being intercepted by SampleAspect
        mongoTemplate.save(test);

    }

所以我在朱尼特被拦截，因为它匹配切点。但是，当调用该连接点时，我应该如何确保在 junits（可能通过断言）中我正在拦截？mongoTemplate.save(test)SampleAspectSampleAspect

我不能断言返回值，因为它除了执行关节点之外没有任何特殊作用。因此，我的 Junit 找不到任何区别，无论是按方面执行还是基于返回值的常规执行。intercept()

如果提供任何有关方面测试的代码片段示例，都将很棒。谢谢

答案 1

我认为您试图测试的是方面编织和切入点匹配。请注意，这将是一个集成，而不是一个单元测试。如果你真的想对你的方面逻辑进行单元测试，并且因为你已经用“mockito”标记了这个问题，我建议你这样做：编写一个单元测试，模拟方面的连接点，也许还有它的其他参数，如果有的话。下面是一个稍微复杂一些的示例，其中包含一些方面内逻辑：

Java 类的目标方面：

package de.scrum_master.app;

public class Application {
    public static void main(String[] args) {
        new Application().doSomething(11);
        new Application().doSomething(-22);
        new Application().doSomething(333);
    }

    public void doSomething(int number) {
        System.out.println("Doing something with number " + number);
    }
}

被测方面：

package de.scrum_master.aspect;

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;

@Aspect
public class SampleAspect {
    @Around("execution(* doSomething(int)) && args(number)")
    public Object intercept(final ProceedingJoinPoint thisJoinPoint, int number) throws Throwable {
        System.out.println(thisJoinPoint + " -> " + number);
        if (number < 0)
            return thisJoinPoint.proceed(new Object[] { -number });
        if (number > 99)
            throw new RuntimeException("oops");
        return thisJoinPoint.proceed();
    }
}

运行 Application.main（..） 时的控制台日志：

如您所见，该方面在 11 上传递，否定 -22 并为 333 引发异常：

execution(void de.scrum_master.app.Application.doSomething(int)) -> 11
Doing something with number 11
execution(void de.scrum_master.app.Application.doSomething(int)) -> -22
Doing something with number 22
execution(void de.scrum_master.app.Application.doSomething(int)) -> 333
Exception in thread "main" java.lang.RuntimeException: oops
    at de.scrum_master.aspect.SampleAspect.intercept(SampleAspect.aj:15)
    at de.scrum_master.app.Application.doSomething(Application.java:10)
    at de.scrum_master.app.Application.main(Application.java:7)

方面单元测试：

现在我们真的想验证该方面是否做了它应该做的事情，并涵盖了所有执行路径：

package de.scrum_master.aspect;

import org.aspectj.lang.ProceedingJoinPoint;
import org.junit.Rule;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

import static org.mockito.Mockito.*;

public class SampleAspectTest {
    @Rule
    public MockitoRule mockitoRule = MockitoJUnit.rule();

    @Mock
    private ProceedingJoinPoint proceedingJoinPoint;

    private SampleAspect sampleAspect = new SampleAspect();

    @Test
    public void testPositiveSmallNumber() throws Throwable {
        sampleAspect.intercept(proceedingJoinPoint, 11);
        // 'proceed()' is called exactly once
        verify(proceedingJoinPoint, times(1)).proceed();
        // 'proceed(Object[])' is never called
        verify(proceedingJoinPoint, never()).proceed(null);
    }

    @Test
    public void testNegativeNumber() throws Throwable {
        sampleAspect.intercept(proceedingJoinPoint, -22);
        // 'proceed()' is never called
        verify(proceedingJoinPoint, never()).proceed();
        // 'proceed(Object[])' is called exactly once
        verify(proceedingJoinPoint, times(1)).proceed(new Object[] { 22 });
    }

    @Test(expected = RuntimeException.class)
    public void testPositiveLargeNumber() throws Throwable {
        sampleAspect.intercept(proceedingJoinPoint, 333);
    }
}

现在运行这个简单的JUnit + Mockito测试，以便单独测试方面逻辑，而不是布线/编织逻辑。对于后者，您需要另一种类型的测试。

P.S.：只有你，我才用了JUnit和Mockito。通常我只使用Spock及其内置的模拟功能。;-)

答案 2