Question

我正在尝试为一个现有功能编写单元测试，该功能使用EntityManager更新数据库字段的值。我的单元测试应验证该类是否正确更新了该值。当前代码如下：

public class MyClass {
    MyClass(EntityManager em) {
      this.em = em;
    }

    void updateValue() {
      em.getTransaction().begin();
      TypedQuery query = em.createQuery("select a from ...", MyResult.class);
      List<MyResult> results = query.getResultList(); 

      MyResult result = results.get(0);
      result.setInterestingValue(1);
      em.getTransaction().commit();
    }
}

我当前的单元测试方法：

@Mock EntityManager em;
@Mock TypedQuery query;

publid void test() {
   MyClass myClass = new MyClass(em);
   MyResult result = new MyResult();
   ArrayList<MyResult> resultList = new ArrayList<>();
   resultList.add(result);

   when(em.getTransaction()).thenReturn(mock(EntityTransaction.class));
   when(em.createQuery(anyString(), any())).thenReturn(query);
   when(query.getResultList()).thenReturn(resultList);

   myClass.updateValue();

   assertEquals(1, result.getInterestingValue());
}

上面的我的单元测试验证MyResult实例是否已更新。但是，它不会验证该值是否实际上是通过EntityManager更新的。换句话说，我想测试EntityManager是否在现有代码中正确使用。例如，即使我在setInterestinvValue（1）之前调用commit（），我的测试仍会通过。有测试的好方法吗？

Answer 1

如上所述，如果您使用模仿，解决方案将是-使用inOrder()：

MyResult result = mock(MyResult.class);
EntityTransaction transaction = mock(EntityTransaction.class)

inOrder.verify(result).setInterestingValue(1);
inOrder.verify(transaction).commit();

如何编写EntityManager事务的单元测试？

1 个答案: