如何在Quarkus测试中高效加载基于应用实体的测试数据？

当然有几种高效且可维护的方案，直接基于你已定义的实体来加载测试数据，比写SQL脚本灵活得多：

1. 结合@TestTransaction与EntityManager直接持久化实体 #

Quarkus提供的@TestTransaction注解可以让测试方法在事务中执行，测试完成后自动回滚（默认行为），避免污染测试环境。你可以直接注入EntityManager，创建实体对象并持久化：

import io.quarkus.test.junit.QuarkusTest;
import io.quarkus.test.junit.TestTransaction;
import jakarta.persistence.EntityManager;
import jakarta.inject.Inject;
import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.*;

@QuarkusTest
public class UserResourceTest {

    @Inject
    EntityManager em;

    @Test
    @TestTransaction
    void testUserLookup() {
        // 创建实体对象
        User testUser = new User();
        testUser.setUsername("test_user");
        testUser.setEmail("test@example.com");
        
        // 持久化到数据库
        em.persist(testUser);
        
        // 执行测试逻辑，比如调用API或查询验证
        User foundUser = em.find(User.class, testUser.getId());
        assertNotNull(foundUser);
        assertEquals("test_user", foundUser.getUsername());
    }
}

2. 封装测试数据工厂类 #

如果多个测试需要复用相同的实体数据，建议封装一个静态工厂类，统一生成预定义的实体对象，后续实体字段变更时只需修改工厂类，不用逐个调整测试代码：

public class TestDataFactory {
    public static User createDefaultUser() {
        User user = new User();
        user.setUsername("default_user");
        user.setEmail("default@test.com");
        user.setPasswordHash("hashed_pass");
        return user;
    }

    public static Order createSampleOrder(User user) {
        Order order = new Order();
        order.setUser(user);
        order.setTotalAmount(99.99);
        order.setStatus(OrderStatus.PENDING);
        return order;
    }
}

在测试中直接调用：

@Test
@TestTransaction
void testOrderCreation() {
    User user = TestDataFactory.createDefaultUser();
    em.persist(user);
    
    Order order = TestDataFactory.createSampleOrder(user);
    em.persist(order);
    
    // 验证逻辑...
}

3. 基于Quarkus Panache简化操作（若使用Panache） #

如果你的实体继承了PanacheEntity或PanacheEntityBase，可以直接使用Panache提供的简化API，无需手动注入EntityManager，代码更简洁：

@QuarkusTest
public class PanacheUserTest {

    @Test
    @TestTransaction
    void testPanachePersist() {
        User user = new User();
        user.username = "panache_user";
        user.email = "panache@test.com";
        
        // 直接调用Panache的persist方法
        user.persist();
        
        // 查询验证
        assertTrue(User.count("username", "panache_user") > 0);
    }
}

4. 用@BeforeEach初始化共享测试数据 #

如果多个测试方法需要依赖相同的基础数据，可以在@BeforeEach方法中完成实体的持久化，确保每个测试都有干净的初始数据：

@QuarkusTest
public class SharedDataTest {

    @Inject
    EntityManager em;

    @BeforeEach
    @TestTransaction
    void setupTestData() {
        // 初始化基础用户数据
        User adminUser = new User();
        adminUser.setUsername("admin");
        adminUser.setEmail("admin@test.com");
        em.persist(adminUser);
    }

    @Test
    @TestTransaction
    void testAdminAccess() {
        // 这里可以直接使用setup中创建的admin用户
        User admin = em.find(User.class, 1L);
        assertNotNull(admin);
        assertEquals("admin", admin.getUsername());
    }
}

这些方案的核心优势是完全基于业务实体对象，实体字段变更时无需手动修改SQL脚本，类型安全且可维护性更高，和你的业务代码逻辑保持一致，避免了SQL与实体结构不一致的问题。

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