Spring Data R2DBC测试中如何替代JPA的@Sql注解实现数据管理？

好问题！确实Spring Data JPA里的@Sql注解用起来特别顺手，但在响应式的R2DBC生态里并没有直接对应的注解。不过别担心，有几个实用的方案能帮你轻松完成测试数据的准备和清理工作：

方案一：结合@BeforeEach/@AfterEach与DatabaseClient执行SQL #

这是最直接也最灵活的方式，利用Spring的测试生命周期注解，配合DatabaseClient来手动执行数据初始化和清理的SQL语句或脚本。我自己做R2DBC集成测试时，最常用的就是这个方案，灵活度拉满。

示例代码如下：

@SpringBootTest
class R2dbcRepositoryTest {

    @Autowired
    private DatabaseClient databaseClient;

    @BeforeEach
    void setupTestData() {
        // 执行单条SQL插入测试数据
        databaseClient.sql("INSERT INTO users (id, name) VALUES (:id, :name)")
                .bind("id", 1L)
                .bind("name", "Test User")
                .fetch()
                .rowsUpdated()
                .block(); // 测试环境下用block()同步执行完全没问题

        // 或者加载外部SQL脚本，和@Sql的脚本逻辑一致
        Resource scriptResource = new ClassPathResource("test-data.sql");
        ScriptUtils.executeSqlScript(databaseClient.getConnectionFactory(), scriptResource);
    }

    @AfterEach
    void cleanUpTestData() {
        databaseClient.sql("DELETE FROM users WHERE id = :id")
                .bind("id", 1L)
                .fetch()
                .rowsUpdated()
                .block();
    }

    // 你的测试方法...
}

方案二：使用Testcontainers配合初始化脚本 #

如果你的测试依赖真实的数据库（比如PostgreSQL、MySQL），Testcontainers可以帮你启动一个临时数据库容器，容器启动时自动执行初始化脚本，测试结束后直接销毁容器，完全隔离测试环境，连手动清理数据的步骤都省了。

示例配置：

@Testcontainers
@SpringBootTest
class R2dbcContainerTest {

    @Container
    static PostgreSQLContainer<?> postgresContainer = new PostgreSQLContainer<>("postgres:15")
            .withInitScript("init-test-data.sql"); // 容器启动时自动执行的初始化脚本

    @DynamicPropertySource
    static void configureProperties(DynamicPropertyRegistry registry) {
        registry.add("spring.r2dbc.url", () -> "r2dbc:postgresql://" + postgresContainer.getHost() + ":" + postgresContainer.getFirstMappedPort() + "/" + postgresContainer.getDatabaseName());
        registry.add("spring.r2dbc.username", postgresContainer::getUsername);
        registry.add("spring.r2dbc.password", postgresContainer::getPassword);
    }

    // 你的测试方法...
}

这种方式适合需要完整数据库环境的复杂集成测试，不用操心数据残留的问题。

方案三：自定义注解封装复用逻辑 #

如果多个测试类都需要重复的初始化/清理逻辑，你可以自定义一个类似@Sql的注解，结合Spring的@TestExecutionListener扩展机制来实现。比如让自定义注解指定SQL脚本路径，然后在监听器里自动执行脚本。这个方案需要写一点封装代码，但能大幅提升复用性。

总结下来，简单测试用方案一最便捷，复杂集成测试选方案二更省心，复用需求高的话可以试试方案三。

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