public void testDelete() {
   Exception ex = new ObjectRetrievalFailureException(User.class, "foo");
   dao.expects(once()).method("remove").isVoid();
   dao.expects(once()).method("get").will(throwException(ex));
   manager.remove(User.class, "foo");
   new AssertThrows(ObjectRetrievalFailureException.class) {
     public void test() {
       manager.get(User.class, "foo");
     }
   }.runTest();
 }
 public void testUpdate() {
   User user = createUser();
   dao.expects(once()).method("save").isVoid();
   user.getAddress().setCountry("USA");
   user = (User) manager.save(user);
 }
 /** Simple test to verify BaseDao works. */
 public void testCreate() {
   User user = createUser();
   dao.expects(once()).method("save").will(returnValue(user));
   user = (User) manager.save(user);
 }
 public void testRetrieve() {
   User user = createUser();
   dao.expects(once()).method("get").will(returnValue(user));
   user = (User) manager.get(User.class, user.getUsername());
 }
 protected void setUp() throws Exception {
   super.setUp();
   dao = new Mock(UniversalDao.class);
   manager.setDao((UniversalDao) dao.proxy());
 }