Unit Test 测试采用H2数据库，两种方式导入数据

大家好，我是神韵，是一个技术&生活博主。出文章目的主要是两个，一是好记忆不如烂笔头，记录总结中提高自己。二是希望我的文章可以帮到大家。欢迎来点赞打卡，你们的行动将是我无限的动力。
本篇主题是：Unit Test 测试采用H2数据库，两种方式导入数据

Unit Test 测试采用H2-嵌入式内存数据库作为测试数据来源，因为数据比较复杂，如果从代码上写，会造成很多没有用的代码。

代码在Github上可以找到：FYR
采用H2作为内存数据库，配置文件内容可以如下：

spring:
  #不存在会自动创建/包括库、表等
  jpa:
    hibernate:
      #测试时每次删掉
      ddl-auto: create-drop
      naming:
        #对表字段采用驼峰命名方式，取消默认的下划线命名(默认，SpringPhysicalNamingStrategy：表名，字段为小写，当有大写字母的时候会添加下划线分隔符号)
        physical-strategy: org.hibernate.boot.model.naming.PhysicalNamingStrategyStandardImpl
    generate-ddl: true
    show-sql: true
    properties:
      hibernate:
        #防止懒加载
        enable_lazy_load_no_trans: true
        dialect: org.hibernate.dialect.H2Dialect
    #配置ture，JPA自动建表会在执行sql文件之前，否则会报找不到table
    defer-datasource-initialization: true
  sql:
    init:
      data-locations: classpath:sql/boot_backend_init.sql


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

一、通过SpringBoot自带的方式导入

我之前写过一篇文章介绍，建议可以先看一下https://blog.csdn.net/qq_41055045/article/details/122038093?spm=1001.2014.3001.5501

因为我采用的是SpringBoot2.5.1，直接使用以下配置加载文件到H2数据库内存中（老版本使用spring.datasource.data: xxx.sql）

spring:
  sql:
    init:
      data-locations: classpath:sql/boot_backend_init.sql
1
2
3
4

如果不是单个文件，而是sql目录下全部sql文件，则用*代替名称
如果加载，指定的文件，而不是全部文件，则用“，”分隔需要加载的文件

二、通过在Spring容器启动后调用执行

通过注解：@RunWith(SpringRunner.class)启动Spring容器，然后注入DataSource和ApplicationContext，通过ResourceDatabasePopulator类将资源包装，然后通过DatabasePopulatorUtils的execute执行SQL文件。

这里放到Before注解中，在该类执行测试类之前会执行SQL加载，但是之后如果执行其它的测试类，此次加载的数据仍然有效果，所以就需要避免主键重复的问题。