Flooyd space
 
Search this site

Navigation
Home‎ > ‎Java Code‎ > ‎

SampleDBUnitTest

package com.freizl.dbunit;

import static org.junit.Assert.assertEquals;

import org.dbunit.IDatabaseTester;
import org.dbunit.JdbcDatabaseTester;
import org.dbunit.database.QueryDataSet;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.dbunit.operation.DatabaseOperation;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class SampleTest {
    private IDatabaseTester databaseTester;

    @Before
    public void setUp() throws Exception {
        initJdbcDatabaseTester();
        databaseTester.setDataSet(getDataSet());
        databaseTester.onSetup();

    }

    private void initJdbcDatabaseTester() throws ClassNotFoundException {
        if (databaseTester == null) {
            databaseTester = new JdbcDatabaseTester(
                    "oracle.jdbc.driver.OracleDriver",
                    "jdbc:oracle:thin:@localhost:1521:orcl", "weblogic",
                    "weblogic", "weblogic");

            databaseTester.setSetUpOperation(DatabaseOperation.REFRESH);
            databaseTester.setTearDownOperation(DatabaseOperation.DELETE);
        }
    }

    private IDataSet getDataSet() throws Exception {
        FlatXmlDataSet dataset = new FlatXmlDataSet(getClass().getClassLoader()
                .getResourceAsStream("hr-seed.xml"));
        return dataset;
    }

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

    @Test
    public void testMe() throws Exception {

        QueryDataSet qds = new QueryDataSet(this.databaseTester.getConnection());

        String testTable = "EMPLOYEE";
        qds
                .addTable(testTable,
                        "select first_name from EMPLOYEE where employee_id >= 2 and employee_id <= 4");
        assertEquals(3, qds.getTable(testTable).getRowCount());
        assertEquals("Drew", qds.getTable(testTable).getValue(0, "First_Name"));

    }

}

Attachments (2)

  • folder-structure.png - on Sep 22, 2009 4:16 AM by Haisheng Wu (version 1)
    11k View Download
  • hr-seed.xml - on Sep 22, 2009 4:16 AM by Haisheng Wu (version 1)
    1k Download

Comments (2)

Haisheng Wu - Sep 22, 2009 4:16 AM

Folder Structure
src
- com.freizl.dbunit
- SampleTest.java
- hr-seed.xml

Haisheng Wu - Sep 29, 2009 1:50 AM

TODO:
check project to http://code.google.com/p/personal-study/