@Test(expected = InsufficientBalanceException.class) public void testWithdrawWithInsufficientBalance() { Account account = new Account(TEST_ACCOUNT_NO, 100); accountDao.findAccount(TEST_ACCOUNT_NO); easyMock.expectLastCall().andReturn(account); // mockControl.setReturnValue(account); easyMock.replay(); accountService.withdraw(TEST_ACCOUNT_NO, 150); easyMock.verify(); }
@Test public void withdrawWithSufficientBalance() { Account account = new Account(TEST_ACCOUNT_NO, 100); accountDao.findAccount(TEST_ACCOUNT_NO); easyMock.expectLastCall().andReturn(account); // mockControl.setReturnValue(account); account.setBalance(50); accountDao.updateAccount(account); easyMock.replay(); accountService.withdraw(TEST_ACCOUNT_NO, 50); easyMock.verify(); }