java 单元测试 对h2数据库数据清理方式

try {

       JdbcTemplate jdbcTemplate = 获取DataSource结构，反射或加载Bean等都可以，不要对业务代码侵入;

       Connection connection = jdbcTemplate.getDataSource().getConnection(); //获取连接

       ResultSet tables = connection.getMetaData().getTables( null , null , null , new String[]{ "TABLE" }); //获取表名

       connection.close(); //记得关闭连接，或者使用连接池

       while (tables.next()){

         //若是测试类的测试方法超过7,java.sql.Connection无法获取连接导致死循环

         jdbcTemplate.update( "DELETE FROM " + tables.getObject( "TABLE_NAME" ));

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

       }

     } catch (SQLException e) {

       e.printStackTrace();

     }

< dependency >

  < groupId >com.h2database</ groupId >

  < artifactId >h2</ artifactId >

  < version >1.4.185</ version >

  < scope >test</ scope >

</ dependency >

< dependency >

  < groupId >junit</ groupId >

  < artifactId >junit</ artifactId >

  < version >4.11</ version >

  < scope >test</ scope >

</dependency

@Component

@Profile ( "testOne" ) // 指定单元测试活动所匹配的数据源

public class DataSourceConfig4Test {

     @Bean (name= "dataSourceUtils" ) // 要和模拟的数据源名称相对应

     DataSource dataSourceUtils() throws SQLException {

         return new EmbeddedDatabaseBuilder()

                 .setType(EmbeddedDatabaseType.H2)

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

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

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

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

                 .build();

     }

}

SET MODE Oracle;

CREATE TABLE TEST

(

     ID      NUMBER(38,0) PRIMARY KEY NOT NULL ,

     PARAM1  VARCHAR2(6) NOT NULL ,

     PARAM2 VARCHAR2(6) NOT NULL ,

  PARAM3 VARCHAR2(1) NOT NULL ,

  PARAM4 VARCHAR2(50) NOT NULL

);

CREATE ALIAS TO_DATE FOR "com.mvn.task.one.Function.toDate" ;

INSERT INTO TEST(ID,PARAM1,PARAM2,PARAM3,PARAM4)

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

@ActiveProfiles ( "testOne" ) // 需要和H2模拟的数据源对应上

@RunWith (SpringJUnit4ClassRunner. class )

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

public class oneTest {

  @Autowired

  private ITestService iTestService;

  @Test

  public void testBengin(){

      Test model = iSubcService.findTest( "TEST1" , "TEST2" );

      Assert.assertNotNull(model);

  }

}