SpringBoot中单元测试如何对包含AopContext.currentProxy()的方法进行测试

springboot,单元测试,如何,包含,aopcontext,currentproxy,方法,进行,测试 · 浏览次数 : 179

小编点评

**问题分析：** 在AopContext中调用自身的函数时，出现 `java.lang.IllegalStateException: Cannot find current proxy: Set 'exposeProxy' property on Advised to 'true' to make it available, and ensure that AopContext.currentProxy() is invoked in the same thread as the AOP invocation context` 错误。 **解决方案：** 1. **使用`@Mock`注解模拟`AopContext`中的静态方法：** - 创建一个`MockStatic`对象，其中包含 `currentProxy()` 方法的返回值。 - 在`@BeforeEach`方法中设置`aopContextMockedStatic`。 2. **使用`@Mock`注解模拟`OrderServiceImpl`的`insertDemo()`方法：** - 创建一个`Mock`对象，其中包含`insertDemo()`方法的行为。 - 在`@Mock`中注入`orderServiceMock`。 3. **使用`@Autowired`注解注入`OrderServiceImpl`：** - 在`@BeforeEach`方法中打开Mocks，将`orderServiceMock`注入到`aopContextMockedStatic`中。 4. **在测试方法中访问`AopContext`：** - 在测试方法中使用`aopContextMockedStatic`访问`AopContext`中的静态方法。 **示例代码：** ```java // Mock AopContext @MockStatic private MockedStatic aopContextMockedStatic; // Mock OrderServiceImpl @Mock private OrderServiceImpl orderServiceMock; // Inject OrderServiceImpl into AopContext @Autowired private AopContext aopContext; // Set 'exposeProxy' property to 'true' aopContextMockedStatic.when(AopContext::currentProxy).thenReturn(orderServiceMock); // Perform test int result = orderServiceUnderTest.insertOrder(); // Assert result assertThat(result).isEqualTo(1); ```

正文

今天在工作中遇到一个问题，一个Service类中有一个方法，其中使用了 AopContext.currentProxy() 去访问自身的函数，例如

int result = ((OrderServiceImpl) AopContext.currentProxy()).save();

单元测试方法如下：

@InjectMocks
private OrderServiceImpl orderServiceUnderTest;

@Test
@DisplayName("AopContext测试")
public void test() {
    int result = orderServiceUnderTest.insertDemo();
    assertThat(result).isEqualTo(1);
}

但是在运行测试的时候出现了 java.lang.IllegalStateException: Cannot find current proxy: Set 'exposeProxy' property on Advised to 'true' to make it available, and ensure that AopContext.currentProxy() is invoked in the same thread as the AOP invocation context.

这时候需要将AopContext的静态方法也mock，同时，需要将被测类 OrderServiceImpl 再注入一次，注入到AopContext中。

具体代码如下：

public class SpringTestDemoApplicationTests {
    @InjectMocks
    private OrderServiceImpl orderServiceUnderTest;

    @Spy
    private orderServiceImpl orderServiceMock;

    @Mock
    private MockedStatic<AopContext> aopContextMockedStatic;

    @BeforeEach
    public void setup() {
        openMocks(this);
        aopContextMockedStatic.when(AopContext::currentProxy).thenReturn(orderServiceMock);
    }

    @Test
    @DisplayName("AopContext测试")
    public void test() {
        int result = orderServiceUnderTest.insertOrder();
        assertThat(result).isEqualTo(1);
    }

    @AfterEach
    public void tearDown() {
        aopContextMockedStatic.close();
    }
}

如果proxy对象调用的方法是通过@Autowired注解注入的对象，可能会出现NPE。此时需要将proxy对象的注解由@Spy 改成 @Mock，并通过mockito模拟该方法的行为。而这个方法本身需要另外写一个测试方法。