H2：构建单元测试的本地内存数据库

dao层的单元测试不能依赖测试环境的数据库以避免环境的影响，H2是一个内存数据库，支持标准SQL，相当于把数据库本地化，可以避免对测试环境的依赖，也可以提升单测的速度

SpringBoot与H2的整合

由于springBoot提供了H2的默认配置，如果使用是的springBoot则不需要新增h2的配置。默认配置如下：

提示：如果不是基于SpringBoot的测试，则需要加上这几项配置

H2数据库初始化
由于h2是内存数据库，不能保存表结构，在每次测试环境准备的时候都要先初始化好需要用到的表结构。
我们在在test/resources目录下新建一个schema.sql文件，这个文件将在初始化数据源的时候被引用到。建表语句如下：

CREATE TABLE `User` (
  `AutoId` bigint(20) NOT NULL AUTO_INCREMENT,
  `UserId` bigint(20) NOT NULL COMMENT '用户Id',
  `UserName` varchar(64) NOT NULL COMMENT '用户姓名',
  `Age` int(10) NOT NULL COMMENT '年龄',
  `PointValue` int(11) NOT NULL DEFAULT '0' COMMENT '积分',
  `Status` smallint(6) NOT NULL DEFAULT '0' COMMENT '记录可用状态',
  `CreateTime` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '记录创建日期',
  `LastModifyTime` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '最后修改日期',
  PRIMARY KEY (`AutoId`)
);

建表语句后面不能有ENGINE=InnoDB DEFAULT CHARSET=utf8，不然将会报错。
下面是初始化数据源的代码：

@Bean
    public DataSource h2DataSource(){

        EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder();
        EmbeddedDatabase database = builder.setType(EmbeddedDatabaseType.H2)
                        .addScript("classpath:schema.sql") ///启动时初始化建表语句
                        .build();
        return database;
    }

H2的pom依赖

        <dependency>
            <groupId>com.h2database</groupId>
            <artifactId>h2</artifactId>
            <version>1.4.195</version>
            <scope>test</scope>
        </dependency>

至此，H2的初始化完成，构建了一个基于H2的本地内存数据库，这样我们就实现了单元测试与测试环境数据库的解耦，零依赖。

更多内容欢迎关注个人微信公众号，一起成长！

H2：构建单元测试的本地内存数据库

猜你喜欢