DBunit 如何在Spring Boot中使用

這篇文章給大家介紹DBunit 如何在Spring Boot中使用，內容非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

Springboot 添加 DBunit 依賴

// https://mvnrepository.com/artifact/org.dbunit/dbunit
testCompile group: 'org.dbunit', name: 'dbunit', version: '2.5.4'

編寫Test.java

import org.dbunit.DBTestCase;
import org.dbunit.DatabaseUnitException;
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.xml.FlatXmlDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSetBuilder;
import org.dbunit.operation.DatabaseOperation;

@RunWith(SpringRunner.class)
@SpringBootTest
public class DBunit extends DBTestCase {

  @Resource
  DataSource dataSource;
  IDatabaseConnection iDatabaseConnection;


  @Override
  protected IDataSet getDataSet() throws Exception {
    return iDatabaseConnection.createDataSet();
  }

  @Before
  public void before() throws Exception{
      iDatabaseConnection = new DatabaseConnection(dataSource.getConnection());
    
  }
}

將數據庫數據轉換為FlatXml

  @Test
  public void testPartialExport() throws DataSetException, IOException {
    QueryDataSet queryDataSet = new QueryDataSet(iDatabaseConnection);
    queryDataSet.addTable("user", "select * from user");
    FlatXmlDataSet.write(queryDataSet, new FileOutputStream("user.xml"));
  }

執行后，將會得到一個user.xml文件，里面記錄了數據庫user表的所有數據，看起來大概是這個樣子

<?xml version='1.0' encoding='UTF-8'?>
<dataset>
 <user id="1" username="mechanists" password="ABA3FC1EB2997E318E43CA099AE175CA"/>
 <user id="2" username="reporter" password="ABA3FC1EB2997E318E43CA099AE175CA" />
 
</dataset>

關于DBunit 如何在Spring Boot中使用就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。