第java單元測試對h2數(shù)據(jù)庫數(shù)據(jù)清理方式目錄java單元測試對h2數(shù)據(jù)庫數(shù)據(jù)清理前因junit單元測試使用H2內(nèi)存數(shù)據(jù)庫首先導(dǎo)入H2內(nèi)存數(shù)據(jù)庫其次使用H2數(shù)據(jù)源模擬Oracle下面來寫個Junit4的單元測試類例子

java單元測試對h2數(shù)據(jù)庫數(shù)據(jù)清理

前因

寫測試框架的時候使用的精簡測試框架不需要啟動整個springboot，并不支持@Transactional測試后回滾h2數(shù)據(jù)庫，而是在基礎(chǔ)測試類里聲明cleandb函數(shù)供使用，這就需要適配任意表的數(shù)據(jù)清除，不過更推薦不清理，以方法名為id使數(shù)據(jù)不重復(fù)即可

JdbcTemplatejdbcTemplate=獲取DataSource結(jié)構(gòu)，反射或加載Bean等都可以，不要對業(yè)務(wù)代碼侵入;

Connectionconnection=jdbcTemplate.getDataSource().getConnection();//獲取連接

ResultSettables=connection.getMetaData().getTables(null,null,null,newString[]{"TABLE"});//獲取表名

connection.close();//記得關(guān)閉連接，或者使用連接池

while(tables.next()){

//若是測試類的測試方法超過7,java.sql.Connection無法獲取連接導(dǎo)致死循環(huán)

jdbcTemplate.update("DELETEFROM"+tables.getObject("TABLE_NAME"));

//spring-test-5.2.1.RELEASE-sources.jar!\org\springframework\test\jdbc\JdbcTestUtils.deleteFromTables如果使用了springtest可以代替使用該方法

}catch(SQLExceptione){

e.printStackTrace();

}

junit單元測試使用H2內(nèi)存數(shù)據(jù)庫

首先導(dǎo)入H2內(nèi)存數(shù)據(jù)庫

Junit4單元測試依賴。

dependency

groupIdcom.h2database/groupId

artifactIdh2/artifactId

version1.4.185/version

scopetest/scope

/dependency

dependency

groupIdjunit/groupId

artifactIdjunit/artifactId

version4.11/version

scopetest/scope

/dependency

其次使用H2數(shù)據(jù)源模擬Oracle

@Component

@Profile("testOne")//指定單元測試活動所匹配的數(shù)據(jù)源

publicclassDataSourceConfig4Test{

@Bean(name="dataSourceUtils")//要和模擬的數(shù)據(jù)源名稱相對應(yīng)

DataSourcedataSourceUtils()throwsSQLException{

returnnewEmbeddedDatabaseBuilder()

.setType(EmbeddedDatabaseType.H2)

.addScript("classpath:/H2_TYPE.sql")

.addScript("classpath:/INIT_TABLE.sql")

.addScript("classpath:/H2_FUNCTION.sql")

.addScript("classpath:/INIT_DATA.sql")

.build();

}

H2_TYPE.sql（設(shè)置H2所模擬的數(shù)據(jù)庫）

SETMODEOracle;

INIT_TABLE.sql（單元測試需要的數(shù)據(jù)庫表初始化）

CREATETABLETEST

IDNUMBER(38,0)PRIMARYKEYNOTNULL,

PARAM1VARCHAR2(6)NOTNULL,

PARAM2VARCHAR2(6)NOTNULL,

PARAM3VARCHAR2(1)NOTNULL,

PARAM4VARCHAR2(50)NOTNULL

);

H2_FUNCTION.sql（不需要特殊方法時，此不可忽略）

CREATEALIASTO_DATEFOR"com.mvn.task.one.Function.toDate";

INIT_DATA.sql（初始化測試數(shù)據(jù)）

INSERTINTOTEST(ID,PARAM1,PARAM2,PARAM3,PARAM4)

VALUES(100,'TEST1','TEST2','2','JKL');

至此H2部分準備完畢

下面來寫個Junit4的單元測試類例子

@ActiveProfiles("testOne")//需要和H2模擬的數(shù)據(jù)源對應(yīng)上

@RunWith(SpringJUnit4ClassRunner.class)

@ContextConfiguration(locations="classpath:testspring/applicationContext-test-one.xml")

publicclassoneTest{

@Autowired

privateITestServiceiTestService;

@Test

publicvoidtestBengin(){

Testmodel=iSubcService.fi