Как смоделировать entityManager.getTransaction (). Begin () в Hibernate с помощью Mockito?

Question

Я работаю над реализацией DAO, которая должна сохранять сущность, но я застрял, когда имитирую начало транзакции.

В частности, я не знаю, как издеваться над Мокито:

entityManager.getTransaction().begin();

Answer 1

Вы можете издеваться entityManager и заставить его возвращаться transaction издеваться.

public class DaoTest {

  EntityManager entityManager = Mockito.mock(EntityManager.class);
  EntityTransaction transaction = Mockito.mock(EntityTransaction.class);
  Dao dao = new MyDao();

  @Test
  void whenSave_thenBeginTransaction(){
    when(entityManager.getTransaction()).thenReturn(transaction);
    dao.save(new MyObject());
    verify(transaction).begin();
  }            
}