JUnitとdbUnitを使ってPostgresのDBテストを実施する

はじめに

データベースの知識がなくって、DBのテスト方法がわからない。

そこで、データベースのテストを自動化するためのツール、dbUnitを調べてみた。

https://dbunit.sourceforge.net/

dbUnitをつかうと以下のようなことができる。

JUnitのテスト実施前に、データをデータベースに投入。
テスト実施後に、期待のデータと変更後のデータベースの値を比較。

データはxmlデータ, Excelデータ, csvデータが利用できる。

インストール

ビルドツールにgradleを利用しているので、gradleの設定を書く。

以下から、最新のバージョンをクリックして、gradleのコードをdependenciesに書く。

https://mvnrepository.com/artifact/org.dbunit/dbunit

dependencies {   
    compile group: 'org.dbunit', name: 'dbunit', version: '2.5.4'
}

データのテーブル用意

データのテーブルを用意する。ここでは、flywayを使った。

buildscript {
    dependencies {
        classpath 'com.h2database:h2:1.4.191'
    }
}

plugins {
    id "org.flywaydb.flyway" version "4.2.0"
}

flyway {
    url = 'jdbc:postgresql://localhost:5434/sampleDB'
    user = 'postgres'
    password = 'postgres'
}

本題ではないので、ここでは参考リンクを貼って詳細はそちらに譲る。

以下のようなテーブルをflywayMigrateで作成。V1__create_table.sql

CREATE TABLE HOGE (
    ID INT,
    NAME VARCHAR(255),
    STATE VARCHAR(10)
);

投入データと期待データの準備

データはxml形式のものを用意する。テスト前、テスト後のデータを用意。ID=3の stateを normal から errorに書き換える。

Before.xml

<?xml version="1.0" encoding="UTF-8"?>
<dataset>
    <HOGE ID="1" STATE="normal" NAME="name#1"/>
    <HOGE ID="2" STATE="normal" NAME="name#2"/>
    <HOGE ID="3" STATE="normal" NAME="name#3"/>
    <HOGE ID="4" STATE="normal" NAME="name#4"/>
    <HOGE ID="5" STATE="normal" NAME="name#5"/>
</dataset>

After.xml

<?xml version="1.0" encoding="UTF-8"?>
<dataset>
    <HOGE ID="1" STATE="normal" NAME="name#1"/>
    <HOGE ID="2" STATE="normal" NAME="name#2"/>
    <HOGE ID="3" STATE="error" NAME="name#3"/>
    <HOGE ID="4" STATE="normal" NAME="name#4"/>
    <HOGE ID="5" STATE="normal" NAME="name#5"/>
</dataset>

テスト対象コード

テスト対象コードをいかに示す。やっていることは、SQL文を実行しているだけ。

package sample;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;

public class DbUnitSample {
    private static final String SQL = "update HOGE set state = 'error' where name = 'name#3';";

    public void execute() {
        try(Connection conn = DriverManager.getConnection(
                "jdbc:postgresql://localhost:5434/sampleDB", "postgres", "postgres");
        ) {
            PreparedStatement stmt = conn.prepareStatement(SQL);
            stmt.executeUpdate();
        } catch(Exception e) {
            System.out.println("failed.");
        }
    }
}

テストコード

テストコードをいかに示す。

package sample;

import org.dbunit.Assertion;
import org.dbunit.IDatabaseTester;
import org.dbunit.JdbcDatabaseTester;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.ITable;
import org.dbunit.dataset.SortedTable;
import org.dbunit.dataset.xml.FlatXmlDataSetBuilder;
import org.dbunit.operation.DatabaseOperation;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.io.File;

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat;

public class DbUnitSampleTest {
    private static IDatabaseTester databaseTester;
    private DbUnitSample sample;

    @Before
    public void setUp() throws Exception {
        databaseTester = new JdbcDatabaseTester("org.postgresql.Driver",
                "jdbc:postgresql://localhost:5434/sampleDB",
                "postgres",
                "postgres",
                "public");

        IDataSet dataSet = new FlatXmlDataSetBuilder().build(
                new File("src/test/resources/db/dbunit/Before.xml"));
        databaseTester.setDataSet(dataSet);
        databaseTester.setSetUpOperation(DatabaseOperation.CLEAN_INSERT);
        databaseTester.onSetup();

        sample = new DbUnitSample();
    }

    @After
    public void tearDown() throws Exception {
        databaseTester.setTearDownOperation(DatabaseOperation.NONE);
        databaseTester.onTearDown();
    }

    @Test
    public void state更新() throws Exception {
        sample.execute();
        IDataSet expectedDataSet = new FlatXmlDataSetBuilder().build(
                new File("src/test/resources/db/dbunit/After.xml"));
        ITable expectedTable = expectedDataSet.getTable("HOGE");

        IDataSet databaseDataSet = databaseTester.getConnection().createDataSet();
        ITable actualTable = databaseDataSet.getTable("HOGE");
        ITable sortedTable = new SortedTable(actualTable, new String[]{"id"});

        Assertion.assertEquals(expectedTable, sortedTable);
    }

    @Test
    public void state更新2() throws Exception {
        sample.execute();
        IDataSet databaseDataSet = databaseTester.getConnection().createDataSet();
        ITable actualTable = databaseDataSet.getTable("HOGE");

        String actualState = "normal";
        for(int i=0; i < actualTable.getRowCount(); i++) {
            if (actualTable.getValue(i, "name").toString().equals("name#3")) {
                actualState = actualTable.getValue(i, "state").toString();
                break;
            }
        }
        assertThat("error", is(actualState));
    }
}

ちょっと注意するところは、sqlのupdateを実行すると、updateしたid=3が、rowの一番最後になってしまうところ。なので、テスト1では、わざわざソートして対応、テスト２では、for文で名前をキーにして検索をしている。

        ITable sortedTable = new SortedTable(actualTable, new String[]{"id"});

おわりに

手動でデータベースを書き換えてテストするのではなくて、flywayやらdbunitやら、使えるものは使って効率よく開発すべき。

しかし問題は、テストを書く時間すら今は確保できないくらい追い詰められていること。

進捗ヤバメです。(´・ω・`)

code:

https://github.com/tsu-nera/java_spock_playground

はじめに#

インストール#

データのテーブル用意#

投入データと期待データの準備#

テスト対象コード#

テストコード#

おわりに#