如何为这种方法编写单元测试用例？

Question

public String createUser() {

    Session session = HibernateUtility.getSession();
    Transaction tx = session.beginTransaction();

    session.save(this);
    tx.commit();
    session.close();

    return "accountCreated";

}

如何为这种方法编写单元测试用例？

Answer 1

很难对包含对静态方法（例如HibernateUtility.getSession()调用的代码进行单元测试。

您可能考虑注入Session实例（也许通过向createUser()方法中添加一个参数，或者通过给包含该方法的对象一个Session Factory对象，它可以用来获取Session实例）。 然后，您可以使用Mockito或类似的框架为该方法提供一个模拟Session对象，并验证对它的预期调用。

public String createUser(Session session) {
    Transaction tx = session.beginTransaction();

    session.save(this);
    tx.commit();
    session.close();

    return "accountCreated";
}

使用Mockito，单元测试可能如下所示：

public testCreateUser(){
    Session session = mock(Session.class);
    Transaction trans = mock(Transaction.class);

    when(session.beginTransaction()).thenReturn(trans);

    obj.createUser(session);

    verify(session).save(obj);
    verify(session).close();
    verify(trans).commit();
}