如何Mock EntityListener中的doThis()方法以测试监听逻辑？

JPA的EntityListener实例由JPA提供者（如Hibernate）直接创建，不属于Spring IoC容器管理范畴。你用@MockBean创建的Mock实例和JPA实际调用的Listener实例完全是两个对象，所以Mock不会生效。

方案一：解耦逻辑到独立服务（推荐） #

把doThis()的业务逻辑抽成单独的Spring Bean，让myEntityListener依赖这个Bean，这样就能通过Mock这个服务实现需求，同时保留myListener()的真实逻辑。

步骤1：抽取独立服务 #

@Component
public class MyBusinessService {
    // 原doThis()的复杂逻辑
    public String doThis() {
        return "expensiveToCompute";
    }
}

步骤2：修改EntityListener为Spring管理Bean并注入服务 #

@Component
public class myEntityListener {
    
    @Autowired
    private MyBusinessService businessService;
    
    @PrePersist
    public void myListener(myEntity e) {
         if (complexConditionToTest) {
             e.setSomething(businessService.doThis());
         }
    }
}

步骤3：修改测试代码Mock服务 #

public class MyEntityListenerTest {

    @Autowired
    myEntityRepository repo;
    
    @MockBean
    MyBusinessService businessService;
    
    @Test
    public void myListenerTest() throws Exception {
        String expected = "fake-text";
        Mockito.when(businessService.doThis()).thenReturn(expected);
        
        myEntity e = new myEntity();
        // 设置满足complexConditionToTest的属性
        myEntity stored = repo.save(e);
        assertThat(stored.getSomething()).isEqualTo(expected);
    }
}

方案二：使用Mockito Spy替换JPA的Listener实例 #

如果不想修改原有代码结构，可以通过Mockito创建myEntityListener的Spy对象（保留真实的myListener()逻辑，MockdoThis()），手动注册到JPA的事件监听体系中。

测试代码示例（基于Hibernate） #

@Autowired
myEntityRepository repo;

@Autowired
EntityManager entityManager;

@Test
public void myListenerTest() throws Exception {
    String expected = "fake-text";
    
    // 创建Spy对象，保留myListener真实实现，Mock doThis方法
    myEntityListener spyListener = Mockito.spy(new myEntityListener());
    Mockito.when(spyListener.doThis()).thenReturn(expected);
    
    // 获取Hibernate事件注册器，替换原有Listener
    Session session = entityManager.unwrap(Session.class);
    EventListenerRegistry registry = session.getSessionFactory().unwrap(EventListenerRegistry.class);
    
    // 清除原有PRE_PERSIST事件的Listener，添加Spy实例
    registry.getEventListenerGroup(EventType.PRE_PERSIST).clear();
    registry.getEventListenerGroup(EventType.PRE_PERSIST).appendListener(spyListener);
    
    myEntity e = new myEntity();
    // 设置满足complexConditionToTest的属性
    myEntity stored = repo.save(e);
    assertThat(stored.getSomething()).isEqualTo(expected);
}

注意事项 #

• 该方案依赖Hibernate的API，若使用其他JPA提供者（如EclipseLink），需对应调整事件注册代码。
• 测试完成后建议恢复原有Listener，避免影响其他测试用例。

内容的提问来源于stack exchange，提问作者Ateig