注意DatabaseFixture的createProxy()方法,它將一個普通的DAO對象包裝為在事務范圍內執行的代理對象,即對于一個普通的DAO對象的方法調用前后,自動地開啟事務并根據異常情況提交或回滾事務。
下面是UserDaoImpl的單元測試類:
public class UserDaoImplTest extends DatabaseFixture {
private UserDao userDao = new UserDaoImpl();
private UserDao proxy = (UserDao)createProxy(userDao); @Test
public void testQueryUser() {
User user = newUser("test");
proxy.createUser(user);
User t = proxy.queryUser("test");
assertEquals(user.getEmail(), t.getEmail());
}
}
注意到UserDaoImplTest持有兩個UserDao引用,userDao是普通的UserDaoImpl對象,而proxy則是將userDao進行了事務封裝的對象。
由于UserDaoImplTest從DatabaseFixture繼承,因此,@Before方法在每個@Test方法調用前自動調用,這樣,每個@Test方法執行前,數據庫都是一個經過初始化的“干凈”的表。
對于普通的測試,如UserDao.queryUser()方法,直接調用proxy.queryUser()即可在事務內執行查詢,獲得返回結果。
對于異常測試,例如期待一個ResourceNotFoundException,就不能直接調用proxy.queryUser()方法,否則,將得到一個UndeclaredThrowableException:
這是因為通過反射調用拋出的異常被代理類包裝為UndeclaredThrowableException,因此,對于異常測試,只能使用原始的userDao對象配合TransactionCallback實現:
@Test(expected=ResourceNotFoundException.class)
public void testQueryNonExistUser() throws Exception {