Untitled

package dbunitsample;

 

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.net.MalformedURLException;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.SQLException;

 

import org.dbunit.Assertion;

import org.dbunit.DatabaseUnitException;

import org.dbunit.database.AmbiguousTableNameException;

import org.dbunit.database.DatabaseConnection;

import org.dbunit.database.IDatabaseConnection;

import org.dbunit.database.QueryDataSet;

import org.dbunit.dataset.DataSetException;

import org.dbunit.dataset.IDataSet;

import org.dbunit.dataset.ITable;

import org.dbunit.dataset.xml.FlatXmlDataSet;

import org.dbunit.dataset.xml.FlatXmlDataSetBuilder;

import org.dbunit.operation.DatabaseOperation;

import org.junit.After;

import org.junit.Before;

import org.junit.Test;

 

 

public class MembersTest {

       

        private String testDataDir = "test_data";

 

       

        private Connection getMyConnection() throws ClassNotFoundException, SQLException{

                Class.forName("com.ibm.db2.jcc.DB2Driver");

               

                return DriverManager.getConnection(

                                "jdbc:db2://localhost:50000/SAMPLE"

                                , "db2inst1"

                                , "password"

                );

        }

 

       

        private IDataSet readDataSetFromXMLFile(String path) throws MalformedURLException, DataSetException{

                return new FlatXmlDataSetBuilder().build(new File(path));

        }

 

       

        private void escapeTable(String tableName, String path)

                        throws DatabaseUnitException, AmbiguousTableNameException,

                        IOException, DataSetException, FileNotFoundException,

                        ClassNotFoundException, SQLException {

                Connection con = getMyConnection();

                File file = new File(path); // プロジェクトのルートからのパスで作成

 

                QueryDataSet queryDataSet = new QueryDataSet(new DatabaseConnection(con));

                queryDataSet.addTable(tableName);

                FlatXmlDataSet.write(queryDataSet, new FileOutputStream(file));

        }

 

       

        @Before

        public void setUp() throws Exception {

                // テーブルのデータを退避

                escapeTable("MEMBERS", testDataDir + "/escaped_members.xml" );

        }

 

 

        @After

        public void tearDown() throws Exception {

                IDataSet dataSet = readDataSetFromXMLFile( testDataDir + "/escaped_members.xml");

                Connection con = getMyConnection();

               

                // テーブルのデータをすべて削除して退避したデータを戻す

                DatabaseOperation.CLEAN_INSERT.execute(new DatabaseConnection(con), dataSet);

               

                con.close();

        }

 

 

        @Test

        public void testUpdate() throws DatabaseUnitException, SQLException, FileNotFoundException, ClassNotFoundException, MalformedURLException{

                Connection con = null;

                IDatabaseConnection dbcon = null;

                try{

                        con = getMyConnection();

                        dbcon = new DatabaseConnection(con);

 

                        // テストデータをロード

                        IDataSet testDataSet = readDataSetFromXMLFile(testDataDir

                                        + "/test_update.xml");

 

                        // テーブルのデータを全て消去＋テストデータ投入

                        DatabaseOperation.CLEAN_INSERT.execute(dbcon, testDataSet);

 

                        // 更新処理（テスト対象のメソッド）

                        new Memebers().update();

 

                        // 期待値データをロード

                        IDataSet expectedDataSet = readDataSetFromXMLFile(testDataDir

                                        + "/test_update_expected.xml");

                        ITable expectedTable = expectedDataSet.getTable("MEMBERS");

 

                        // DBのテーブルからデータを取得

                        IDataSet actualDataSet = dbcon.createDataSet();

                        ITable actualTable = actualDataSet.getTable("MEMBERS");

 

                        // 期待値と実際の値を比較

                        Assertion.assertEquals(expectedTable, actualTable);

                       

                } finally {

                        if (dbcon != null) {

                                dbcon.close();

                        }

                        if (con != null) {

                                con.close();

                        }

                }

        }

}