Java中如何实现类Pytest风格的Fixture依赖机制？

如果你习惯了pytest中Fixture的链式/图式依赖结构，回到Java可以通过以下方式实现类似功能，同时也有几个同类框架可选：

一、JUnit 5（Jupiter）的原生实现 #

JUnit 5提供了多种机制来分层搭建和销毁测试环境，足以模拟pytest Fixture的依赖逻辑：

1. 嵌套测试类（@Nested）实现层级环境 #

通过嵌套类可以分层管理Fixture的作用域，外层类负责基础环境（比如数据库事务），内层类基于外层环境扩展（比如插入测试数据），测试方法按需使用对应层级的资源。

import org.junit.jupiter.api.*;

class DatabaseFixtureTests {
    // 基础Fixture：开启并自动回滚事务
    @BeforeEach
    void initTransaction() {
        // 实现事务开启逻辑（比如用JDBC或ORM框架的事务API）
    }

    @AfterEach
    void rollbackTransaction() {
        // 实现事务回滚逻辑
    }

    // 第一层扩展Fixture：插入基础测试数据
    @Nested
    class WithBaseTestData {
        private User testUser;

        @BeforeEach
        void insertBaseData() {
            testUser = new User("test_user", "test@example.com");
            // 基于外层事务插入数据到数据库
        }

        @Test
        void testUserCreation() {
            // 使用testUser验证数据库操作
            Assertions.assertEquals("test_user", testUser.getUsername());
        }

        // 第二层扩展Fixture：插入关联数据
        @Nested
        class WithAssociatedOrders {
            private Order testOrder;

            @BeforeEach
            void insertOrderData() {
                testOrder = new Order(testUser.getId(), "TEST_ORDER");
                // 基于上层的testUser插入关联订单
            }

            @Test
            void testOrderAssociation() {
                // 同时使用testUser和testOrder测试关联逻辑
                Assertions.assertEquals(testUser.getId(), testOrder.getUserId());
            }
        }
    }
}

2. 自定义扩展（@ExtendWith）实现Fixture依赖链 #

通过实现JUnit 5的Extension接口，可以封装可复用的Fixture逻辑，并支持依赖传递。比如先定义事务管理扩展，再定义依赖事务的数据插入扩展：

import org.junit.jupiter.api.extension.*;

// 事务管理扩展：负责开启/回滚事务
class TransactionExtension implements BeforeEachCallback, AfterEachCallback {
    @Override
    public void beforeEach(ExtensionContext context) {
        // 开启事务
    }

    @Override
    public void afterEach(ExtensionContext context) {
        // 回滚事务
    }
}

// 测试数据扩展：依赖事务扩展，插入数据并支持注入
class TestDataExtension implements BeforeEachCallback, ParameterResolver {
    private User testUser;

    @Override
    public void beforeEach(ExtensionContext context) {
        // 此时TransactionExtension已执行，可安全插入数据
        testUser = new User("test_user", "test@example.com");
        // 插入到数据库
    }

    @Override
    public boolean supportsParameter(ParameterContext paramCtx, ExtensionContext extCtx) {
        return paramCtx.getParameter().getType() == User.class;
    }

    @Override
    public Object resolveParameter(ParameterContext paramCtx, ExtensionContext extCtx) {
        return testUser;
    }
}

// 使用扩展组合的测试类
@ExtendWith({TransactionExtension.class, TestDataExtension.class})
class FixtureChainDemoTest {
    @Test
    void testWithInjectedData(User testUser) {
        // 直接注入Fixture实例，无需手动创建
        Assertions.assertNotNull(testUser);
    }
}

3. Spring Test + JUnit 5（针对Spring项目） #

如果是Spring应用，Spring Test提供的注解可以快速实现事务回滚、数据插入等Fixture逻辑，结合JUnit的扩展机制，完全贴合pytest的使用习惯：

import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.annotation.Rollback;
import org.springframework.test.context.jdbc.Sql;
import org.springframework.transaction.annotation.Transactional;

@SpringBootTest
@Transactional // 开启事务
@Rollback(true) // 测试后自动回滚
class SpringFixtureTest {
    @Test
    @Sql("/base-test-data.sql") // 插入基础测试数据
    void testBaseData(UserRepository userRepo) {
        Assertions.assertEquals(1, userRepo.count());
    }

    @Test
    @Sql({"/base-test-data.sql", "/associated-orders.sql"}) // 组合多个数据脚本
    void testCombinedData(UserRepository userRepo, OrderRepository orderRepo) {
        Assertions.assertEquals(1, userRepo.count());
        Assertions.assertEquals(2, orderRepo.count());
    }
}

二、同类替代框架 #

如果觉得JUnit的原生写法不够简洁，这些框架更接近pytest的Fixture风格：

1. Spock Framework #

基于Groovy的测试框架，语法简洁，天生支持分层Fixture和依赖注入。setup()/cleanup()方法支持嵌套层级，还可以用@Shared标记全局Fixture，@Stepwise控制测试执行顺序，和pytest的使用体验非常接近。

2. TestNG #

支持更灵活的Fixture层级（@BeforeSuite/@BeforeTest/@BeforeClass/@BeforeMethod），以及方法依赖（dependsOnMethods），可以轻松搭建链式的测试环境，适合复杂场景的测试编排。

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