测试结构如下:
1 定义测试数据
2 将测试数据导入到数据库中
3 将预期结果和实际结果进行对比
目的:
测试selectAllDemo方法,检测查询结果是否有效
使用DBuint框架完成测试
返回测试用的数据库连接对象
protected abstract IDatabaseConnection getConnection() throws Exception;
返回测试用XML数据集对象
protected abstract IDataSet getDataSet() throws Exception;
部分参考csdn文章
1 定义测试数据
2 将测试数据导入到数据库中
3 将预期结果和实际结果进行对比
目的:
测试selectAllDemo方法,检测查询结果是否有效
使用DBuint框架完成测试
返回测试用的数据库连接对象
protected abstract IDatabaseConnection getConnection() throws Exception;
返回测试用XML数据集对象
protected abstract IDataSet getDataSet() throws Exception;
部分参考csdn文章
package baor.src.dbunit;
import java.sql.ResultSet;
import java.sql.SQLException;
import baor.test.sql.ConnPropInfo;
import baor.test.sql.ConnUtil;
import java.sql.Statement;
public interface CRUDDemo {
public abstract ResultSet selectAllDemo(Statement stmt, String SQLString)
throws SQLException;
public abstract Statement setSTMT(ConnUtil connUtil, ConnPropInfo demo)
throws SQLException;
}
package baor.src.dbunit;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import baor.test.sql.ConnPropInfo;
import baor.test.sql.ConnUtil;
public class SelectAllDemo implements CRUDDemo {
@Override
public ResultSet selectAllDemo(Statement stmt, String SQLString)
throws SQLException {
// TODO Auto-generated method stub
// 创建数据集对象
ResultSet rs = stmt.executeQuery(SQLString);
// System.out.println("TestNo" + "\t" + "Description" + "\t" +
// "Author");
return rs;
}
@Override
public Statement setSTMT(ConnUtil connUtil, ConnPropInfo demo)
throws SQLException {
// TODO Auto-generated method stub
// 创建stmt对象
Statement stmt = connUtil.getConnection(demo).createStatement();
return stmt;
}
}
package baor.test.sql;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
public class ConnUtil {
Connection connection = null;
public Connection getConnection(ConnPropInfo prop) {
if (connection == null) {
try {
// 调用Class.forName()方法加载驱动程序
Class.forName(prop.getDriver());
System.out.println("成功加载MySQL驱动!");
connection = DriverManager.getConnection(prop.getUrl(),
prop.getUser(), prop.getPwd());
System.out.print("成功连接到数据库!");
} catch (ClassNotFoundException e1) {
System.out.println("找不到MySQL驱动!");
e1.printStackTrace();
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
return connection;
}
public void closeConnection(Connection conn) {
try {
if (conn != null) {
conn.close();
conn = null;
System.out.println("连接关闭!!");
}
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
package baor.src.dbunit;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.sql.Connection;
import java.sql.SQLException;
import org.dbunit.DatabaseUnitException;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.database.QueryDataSet;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSet;
public class DataUtil {
public void simgleTableExport(Connection conn, String tableName,
String xmlFile) {
IDatabaseConnection connection = null;
try {
connection = new DatabaseConnection(conn);
QueryDataSet dataSet = new QueryDataSet(connection);
dataSet.addTable(tableName);
FlatXmlDataSet.write(dataSet, new FileOutputStream(xmlFile));
} catch (DatabaseUnitException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
public void allTableExport(Connection conn, String fileName) {
IDatabaseConnection connection = null;
try {
connection = new DatabaseConnection(conn);
IDataSet dataSet = connection.createDataSet();
FlatXmlDataSet.write(dataSet, new FileOutputStream(fileName));
} catch (DatabaseUnitException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (SQLException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (FileNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
package baor.src.dbunit;
import java.io.FileInputStream;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.ResultSet;
import org.dbunit.DatabaseTestCase;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.dbunit.dataset.xml.FlatXmlProducer;
import org.dbunit.ext.mysql.MySqlConnection;
import org.dbunit.operation.DatabaseOperation;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.xml.sax.InputSource;
import java.sql.Statement;
import baor.test.sql.ConnPropInfo;
import baor.test.sql.ConnUtil;
import baor.test.sql.PropInfoFactory;
import baor.test.sql.PropMySQLFactory;
public class DemoTest extends DatabaseTestCase {
Connection conn;
ResultSet rs;
Statement stmt;
FlatXmlProducer fxp;
SelectAllDemo testDemo;
ConnPropInfo demo;
ConnUtil testConnUtil;
String sql = "select * from testdemo where TestNo = 1";
@Before
public void setUp() throws Exception {
// 数据库连接对象
PropInfoFactory test = new PropMySQLFactory();
demo = test.createPropInfo();
testConnUtil = new ConnUtil();
conn = testConnUtil.getConnection(demo);
// xml文件对象
InputStream fis = new FileInputStream("C://testdemo1.xml");
InputSource is = new InputSource(fis);
fxp = new FlatXmlProducer(is);
}
@After
public void tearDown() throws Exception {
testConnUtil.closeConnection(conn);
}
@Test
public void testGet() throws Exception {
testDemo = new SelectAllDemo();
stmt = testDemo.setSTMT(testConnUtil, demo);
rs = testDemo.selectAllDemo(stmt, sql);
while (rs.next()) {
assertEquals(1, rs.getInt(1));
assertEquals("test", rs.getString(2));
assertEquals("paur", rs.getString(3));
}
stmt.close();
rs.close();
}
@Override
protected IDatabaseConnection getConnection() throws Exception {
// TODO Auto-generated method stub
return new MySqlConnection(conn, "test");
}
@Override
protected IDataSet getDataSet() throws Exception {
// TODO Auto-generated method stub
return new FlatXmlDataSet(fxp);
}
/**
* 在把xml里的数据导入到db里之前,需要对数据库里的表的数据做的操作。
*/
protected DatabaseOperation getSetUpOperation() throws Exception {
System.out.println("setup...");
return DatabaseOperation.CLEAN_INSERT;
}
/**
* 测试执行完以后,需要对数据库里的表里的数据做的操作
*/
protected DatabaseOperation getTearDownOperation() throws Exception {
System.out.println("teardown...");
return DatabaseOperation.NONE;
}
}