對DAO編寫單元測試[6]

　　對DAO編寫單元測試[6]

　　下面是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 {

　　new TransactionCallback() {

　　protected Object doInTransaction() throws Exception {

　　userDao.queryUser("nonexist");

　　return null;

　　}

　　}.execute();

　　}

　　到此為止，對DAO組件的單元測試已經實現完畢。下一步，我們需要使用HibernateTool自動生成數據庫腳本，免去維護SQL語句的麻煩。相關的Ant腳本片段如下：

原文轉自：http://www.kjueaiud.com