Play Framework新手：测试DAO时如何指定使用H2内存数据源？

嘿，作为Play Framework新手，想用H2内存库测试DAO其实很简单，Play本身就支持环境隔离的配置，我给你一步步讲怎么做：

1. 配置文件区分生产与测试数据源 #

在你的application.conf里，Play支持用%环境名前缀来指定不同环境的配置，直接添加测试环境的H2配置即可，它会在测试时自动生效：

# 生产环境MySQL配置（保留你原有的配置）
db.default.driver = com.mysql.cj.jdbc.Driver
db.default.url = "jdbc:mysql://localhost:3306/your_prod_db?useSSL=false&serverTimezone=UTC"
db.default.username = your_prod_user
db.default.password = your_prod_pass

# 测试环境专属H2内存库配置
%test.db.default.driver = org.h2.Driver
%test.db.default.url = "jdbc:h2:mem:testdb;DB_CLOSE_DELAY=-1"  # DB_CLOSE_DELAY=-1防止连接关闭时销毁数据库
%test.db.default.username = sa
%test.db.default.password = ""
%test.db.default.jdbcUrl = ${db.default.url}  # 部分ORM（比如Ebean）需要这个参数

2. 确保测试类加载测试环境 #

Play的测试框架会自动识别测试环境，但我们可以通过注解或继承类来明确：

如果你用JUnit 5（推荐） #

直接用@PlayTest注解，它会自动加载测试配置并处理依赖注入：

import org.junit.jupiter.api.Test;
import play.test.PlayTest;
import javax.inject.Inject;

@PlayTest
public class YourDAOTest {

    // 直接注入你的DAO，它会自动使用H2数据源
    @Inject
    private YourDAO yourDAO;

    @Test
    public void testYourDAOMethod() {
        // 在这里调用DAO方法并编写断言
        var testResult = yourDAO.someBusinessMethod();
        // 比如：assertEquals("预期值", testResult);
    }
}

如果你用JUnit 4 #

继承WithApplication并使用PlayJUnitRunner：

import org.junit.Test;
import org.junit.runner.RunWith;
import play.test.PlayJUnitRunner;
import play.test.WithApplication;
import javax.inject.Inject;

@RunWith(PlayJUnitRunner.class)
public class YourDAOTest extends WithApplication {

    @Inject
    private YourDAO yourDAO;

    @Test
    public void testDAOLogic() {
        running(provideApplication(), () -> {
            var result = yourDAO.getData();
            // 断言逻辑
        });
    }
}

3. DAO类要通过依赖注入获取数据源 #

你的DAO不能硬编码数据源，要通过Play的依赖注入拿到Database实例，这样才能在不同环境自动切换：

import javax.inject.Inject;
import play.db.Database;

public class YourDAO {

    private final Database db;

    // 构造函数注入Database
    @Inject
    public YourDAO(Database db) {
        this.db = db;
    }

    // 示例DAO方法
    public String fetchSampleData() {
        return db.withConnection(conn -> {
            try (var stmt = conn.createStatement()) {
                var rs = stmt.executeQuery("SELECT content FROM test_table LIMIT 1");
                return rs.next() ? rs.getString("content") : null;
            }
        });
    }
}

4. 补充：添加H2依赖并初始化测试数据 #

• 首先要在build.sbt里添加H2的测试依赖，这样它只会在测试环境生效：

libraryDependencies += "com.h2database" % "h2" % "2.2.224" % Test

• 如果需要每次测试前初始化数据库（比如建表、插测试数据），可以在测试类里加@BeforeEach方法：

import org.junit.jupiter.api.BeforeEach;
import java.nio.file.Files;
import java.nio.file.Paths;

@BeforeEach
public void setupTestDB() {
    db.withConnection(conn -> {
        // 读取测试资源目录下的初始化SQL脚本
        String initSql = Files.readString(Paths.get("test/resources/init_test_db.sql"));
        try (var stmt = conn.createStatement()) {
            stmt.execute(initSql);
        }
        return null;
    });
}

内容的提问来源于stack exchange，提问作者DMcg