dbunit mysql_基于Dbunit的数据库层数据验证测试

public class TestDbunit  {

public static JdbcDatabaseTester tester;

IDataSet dataSet_Insert ;

IDataSet dataSet_init;

IDataSet dataSet_bak;

@BeforeClass

protected void setUp() throws Exception{

System.out.println("BeforeClass");

IDatabaseConnection conn=tester.getConnection();

String[] depTableNames=new String[]{"ROLES","USERS","USER_ROLES"};

try{

IDataSet utput=conn.createDataSet(depTableNames);

FileOutputStream fos=new FileOutputStream("user-roles-bak.xml");

//备份数据

FlatXmlDataSet.write(output, fos);

//初始化数据

//dataSet_init=new FlatXmlDataSet(new File("user-roles.xml"));

//DatabaseOperation.CLEAN_INSERT.execute(conn, dataSet_init);

}finally{

conn.close();

}

}

@BeforeTest

public void loadDB()throws Exception{

System.out.println("@BeforeTest");

tester=new  JdbcDatabaseTester( "com.mysql.jdbc.Driver" ,

"jdbc:mysql://localhost:3306/test" ,  "root" , "" );

dataSet_init=new FlatXmlDataSet(new File("user-roles.xml"));

IDatabaseConnection conn=tester.getConnection();

try{

DatabaseOperation.CLEAN_INSERT.execute(conn, dataSet_init);

}finally{

conn.close();

}

}

@Test

public void testInsert()throws Exception{

dataSet_Insert=new FlatXmlDataSet(new File("user_insert.xml"));

IDatabaseConnection conn=tester.getConnection();

try{

DatabaseOperation.INSERT.execute(conn, dataSet_Insert);

assert conn.getRowCount("users")==3;

}finally{

conn.close();

}

}

@Test

public void testUpdate()throws Exception{

IDataSet dataSet_Update=new FlatXmlDataSet(new File("user_update.xml"));

IDatabaseConnection conn=tester.getConnection();

IDataSet expect=new  FlatXmlDataSet(new File("user_update_expected.xml"));

IDataSet act=tester.getConnection().createDataSet();

try{

DatabaseOperation.UPDATE.execute(conn, dataSet_Update);

System.out.println(act.getTable("users").getRowCount());

System.out.println(act.getTable("users").getValue(act.getTable("users").getRowCount()-2,"last_name"));

assert act.getTable("users").getValue(act.getTable("users").getRowCount()-2,"last_name").toString().equals("shifan");

}finally{

conn.close();

}

}

@AfterClass

public void clearDB() throws Exception{

IDatabaseConnection conn=tester.getConnection();

try{

//数据还原

DatabaseOperation.CLEAN_INSERT.execute(conn, dataSet_init);

}finally{

conn.close();

}

}

}