package junittest;

import junit.framework.*;
import data.*;
import com.borland.dx.dataset.*;

public class TestWalletData extends TestCase {
  private WalletData walletData = null;
  protected void setUp() throws Exception {
    super.setUp;
    walletData = new WalletData();
   }
  protected void tearDown() throw Exception {
    walletData = null;
    super.tearDown();
  }
  //測試檢查用戶的方法
  public void testCheckUser() {
    //測試正確的用戶名和密碼
    String userName = "ally";
    String userPassword = "ally";
    int expectedReturn = 1;
    int actualReturn = walletData.checkUser(userName,userPassword);
    assertEquals("return value",expectedReturn,actualReturn);
    userName = "張先";
    userPassword = "111";
    expectedReturn = 4;
    actualReturn = walletData.checkUser(userName,userPassword);
    assertEquals("return value",expectedReturn,actualReturn);
    //測試錯誤的用戶名和密碼
    userName = "Ally";
    userPassword = "ally";
    expectedReturn = -1;
    actualReturn = walletData.checkUser(userName,userPassword);
    assertEquals("return value",expectedReturn,actualReturn);
  }
  //測試取得擁護數據類的方法
  public void testGetUsertableDataSet() {
    TableDataSet actualReturn = walletData.getUsertableDataSet();
    assertEquals("return value", 5,actualReturn.getRowCount());
  }
  //測試取得錢包數據類的方法
  public void testGetWalletDataSet() {
    TableDataSet actualReturn = walletData.getWalletDataSet();
    assertEquals("return value", 7,actualReturn.getRowCount());
  } 
  //測試按照日期查詢數據的方法
  public void testDataFilter() {
    //創建日期類
    java.util.Calendar date = java.util.Calendar.getInstance();
    date.set(2004,1,22,0,0,0);
    java.sql.Timestamp startDate = new java.sql.Timestamp(date.getTimeInMillis());
    date.set(2004,1,22,23,59,59);
    java.sql.Timestamp endDate = new java.sql.Timestamp(date.getTimeInMillis());
    walletData.setStartDate(startDate);
    walletData.setEndDate(endDate);
    walletDate.setShowAll(false);
    //重新取得數據類的數據
    walletData.getWalletDataSet().refilter();
    this.assertEquals(2,walletData.getWalletDataSet().getRowCount());
  }
  //測試取得余額的方法
  public void testGetBalance() {
    double actualReturn = walletData.getBalance();
    assertEquals(return value", 1330.0,actualReturn,0);
  }
  //測試取得新記錄ID的方法
  public void testGetWalletId() {
    int actualReturn = walletData.getWalletId();
    assertEquals("return value", 37, actualReturn);
  }
  //測試存款的方法
  public void testDeposit() {
    //向ally用戶的錢包存3元
    double actualReturn = walletData.deposit(1, 3);
    assertEquals("return value", 1333.0,actualReturn, 0);
  }
  //測試取款的方法
  public void testWithdraw() {
    //測試當取款金額大于余額時，返回的測試結果
    double actualReturn = walletData.withdraw(1, 10000);
    assertEquals("return value",-1, actualReturn, 0);
    //向ally用戶的錢包取3元錢
    actualReturn = walletData.withdraw(1, 3);
    assertEquals("return value",1330.0, actualReturn, 0); 
  }
}