AspectJ 的 JUnit 测试

Question

我正在尝试为自定义方面编写 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();

    }

}

因此，每当 joinpoint 与切入点匹配时，上述方面就会进行拦截。它工作正常。

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

@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);

    }

因此，Junit 中的 mongoTemplate.save(test) 被 SampleAspect 拦截，因为它与切入点匹配。但是我应该如何在 junits 中（可能通过断言）确保我的 SampleAspect 在调用该联合点时进行拦截？

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

如果提供有关方面测试的任何代码片段示例，那就太好了。谢谢

Answer 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 测试，以便单独测试方面逻辑，而不是布线/编织逻辑。对于后者，您需要另一种类型的测试。

PS：我只为你使用了 JUnit 和 Mockito。通常我只使用 Spock 及其内置的模拟功能。 <代码>;-)

Answer 2

试试这个：

@RunWith(MockitoJUnitRunner.class)
public class SampleAspectTest {

    @InjectMocks
    private SampleAspect sampleAspect;

    @Mock
    private MongoOperations mongoOperations;

    private MongoOperations mongoOperationsWithAspect;

    @Before
    public void init() {
        AspectJProxyFactory factory = new AspectJProxyFactory(mongoOperations);
        factory.addAspect(sampleAspect);
        mongoOperationsWithAspect = factory.getProxy();
    }

    @Test
    public void testIntercept() {
        // stub
        when(mongoOperations.xxx(...)).thenReturn(...);
        // test
        mongoOperationsWithAspect.xxx();
        // verify
        verify(mongoOperationsWithAspect, times(1)).intercept(any());
    }
}