DBのユニットテスト③ Java

前回の続きです。
UserDaoのテストクラスを作成してみます。

UserDaoTest.java

JUnit4っぽく作ってみました。見やすいコードが書けたかも。

アノテーションは、こんな感じです。
　@Test があるのがテストメソッドです。
　@Before は各テストメソッドの実行前に呼ばれます。
　@After は各テストメソッドの実行後に呼ばれます。

テストの流れは、こんな感じです。
　1.setUpメソッドでuserテーブルを初期化
　2.テストメソッド実行
　　UserDaoのメソッドを実行
　　抽出結果比較 or 更新結果比較（userテーブル）
　3.tearDownメソッドで後始末

package sample;

import java.sql.Connection;
import java.util.ArrayList;
import java.util.List;

import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

public class UserDaoTest {

  private DbTester tester = new DbTester();
  private UserDao dao;

  @Before
  public void setUp() throws Throwable {
    Connection conn = Dbcp.getInstance().getConnection();
    this.tester.setUp(conn);
    this.tester.initDb("testdata/DaoSampleTest/init.xls");
    this.dao = new UserDao(conn);
  }

  @After
  public void tearDown() throws Throwable {
    this.tester.tearDown();
  }

  @Test
  public void selectAll() throws Throwable {

    List<User> list = this.dao.selectAll();

    List<User> expected = new ArrayList<User>();
    expected.add(new User(1, "userA"));
    expected.add(new User(2, "userB"));
    expected.add(new User(3, "userC"));

    Assert.assertArrayEquals(expected.toArray(), list.toArray());
  }

  @Test
  public void select() throws Throwable {

    User user = this.dao.select(1);

    Assert.assertEquals(new User(1, "userA"), user);
  }

  @Test
  public void insert() throws Throwable {

    User user = new User(4, "userD");
    this.dao.insert(user);

    this.tester.assertDb("testdata/DaoSampleTest/insert.xls");
  }

  @Test
  public void update() throws Throwable {

    User user = new User(2, "userE");
    this.dao.update(user);

    this.tester.assertDb("testdata/DaoSampleTest/update.xls");
  }

  @Test
  public void delete() throws Throwable {

    this.dao.delete(2);

    this.tester.assertDb("testdata/DaoSampleTest/delete.xls");
  }

}

テストデータExcelファイルの中身

各テストデータExcelファイルにテスト実行後の期待値を記述します。
シート名がテーブル名で、１行目がカラム名、２行目からデータになります。

■init.xls（テスト実行前の初期状態）

■insert.xls（insertメソッド実行後）
[id=4、name=userD]のレコードが追加されているはず。

■update.xls（updateメソッド実行後）
[id=2、name=userB]が[id=2、name=userE]に変更されているはず。

■delete.xls（deleteメソッド実行後）
[id=2、name=userB]のレコードが削除されているはず。

Eclipseから実行してみる

Eclipseから実行してみます。

テストクラス名を選択して、
右クリック→「実行」→「JUnit テスト」で全テストが実行されます。

どれか一つのテストを実行したい場合は、
テストメソッド名を選択して、右クリック→「実行」→「JUnit テスト」で実行できます。

わざとエラーにしてみます。
selectメソッドの期待値を変更して実行すると・・・

  @Test
  public void select() throws Throwable {

    User user = this.dao.select(1);

　　//Assert.assertEquals(new User(1, "userA"), user);
    Assert.assertEquals(new User(1, "HOGEHOGE"), user);
  }

ちゃんとエラーになりますね！
ユニットテストを作るときはわざと失敗して、
エラーになる事を確認しながら作るとテストの作成ミスを防げると思います。