[开发测试] 利用h2内存数据库编写Dao层的单元测试用例

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 开发测试 -> 利用h2内存数据库编写Dao层的单元测试用例 -> 正文阅读

[开发测试]利用h2内存数据库编写Dao层的单元测试用例

文章目录

摘要

利用h2内存数据库编写Dao层的单元测试用例，代码以pg为例，mysql也是同样适用的。

一、背景

单元测试时提升代码质量和保证代码正确性的重要保证，而在写DAO层的单元测试时，我们往往会遇到一个问题，测试用例所依赖的数据库数据被修改或修改，或者在一个新的环境下所依赖的数据库不存在，导致单元测试无法通过，进而构建失败。

二、技术选型

DAO的单元测试有常见的两种方案：

使用H2这类内存数据库进行单元测试
使用真实的数据库类型，如Mysql/Postgres, 测试用例执行后回滚。

两种方案各有利弊，个人更加倾向于使用第一种，因为更加高效和解耦。

三、流程图

确定好技术方案之后就是落地的流程了，我这里花了一张图

引入依赖库

配置spring参数

数据库初始化sql

公共类编写

测试用例编写

测试效果

四、步骤

4.1 引入依赖库

pom文件中添加

        <dependency>
            <groupId>com.h2database</groupId>
            <artifactId>h2</artifactId>
            <scope>runtime</scope>
        </dependency>

4.2 配置spring参数

定义一个配置文件：application-testPg.yml

spring:
  #初始化基本的配置项
  datasource:
    url: jdbc:h2:mem:learn_db;DB_CLOSE_DELAY=-1;MODE=PostgreSQL
#    url: jdbc:h2:mem:test;DB_CLOSE_DELAY=-1;MODE=PostgreSQL
    username: root
    password:
    driver-class-name: org.h2.Driver
  sql:
    #启动时需要初始化的建表语句
    init:
      username: root
      password:
      platform: pg
      schema-locations: classpath:sql/init/createDB-{{platform}}.sql
      #      mode: always # 如果启动springboot要执行记得开启always， 关闭下面的never
      mode: always
      separator: ;
      data-locations: classpath:sql/init/h2/data-{{platform}}.sql

4.3 数据库初始化sql

CREATE TABLE IF NOT EXISTS t_persons
(
    Id_P      varchar(255) NOT NULL,
    LastName  varchar(255) NOT NULL,
    FirstName varchar(255),
    Address   varchar(255),
    City      varchar(255),
    PRIMARY KEY (Id_P)
);
COMMENT on table t_persons is 'The persons table';

还有数据初始化的sql

INSERT INTO t_persons (Id_P, LastName, FirstName, Address, City) VALUES ('1', 'h', 'jk', 'china', 'dongguan');

4.4 公共的入口配置类编写

package com.dreamkite.pg;

import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.context.annotation.Profile;

@SpringBootTest(classes={UsePgApplication.class})
@Profile(value="testPg")
public class TestPgDao {
}

4.5 测试用例编写

package com.dreamkite.pg.service;

import com.dreamkite.pg.TestPgDao;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;

@Slf4j
class PersonServiceTest extends TestPgDao {
    @Autowired
    PersonService personService;

    @Test
    void delete() {
        System.out.println(personService.queryPersonById("1"));
        personService.delete("1");
        System.out.println(personService.queryPersonById("1"));
        personService.insert("1", "h", "jk", "china", "dongguan");
        System.out.println(personService.queryPersonById("1"));
    }
}