如何创建自定义 JUnit5 扩展

Question

是否可以像在 JUnit4 中创建 @Rule 一样创建自定义扩展？

public class MockMetadataServiceRule extends ExternalResource {   
    @Override
    protected void before() throws Throwable {
        //some setup
    }

    @Override
    protected void after() {
       // some teardown
    }
}

然后，我可以使用 JUnit5

在测试类中执行此操作

@BeforeAll
public static void init() {
    MetadataServiceFactory.setService(MockitoMockMetadataService.getMock());
}

@AfterAll
public static void teardown() {
    MetadataServiceFactory.setService(null);
}

但我想你现在可以@ExtendWith(MyExtension.class) 吗？

编辑： 我的扩展示例以防@nullpointer 链接消失。谢谢。

public class MockMetadataServiceExtension implements BeforeAllCallback, AfterAllCallback {

    @Override
    public void afterAll(ExtensionContext extensionContext) throws Exception {
        MetadataServiceFactory.setService(null);
    }

    @Override
    public void beforeAll(ExtensionContext extensionContext) throws Exception {
        MetadataServiceFactory.setService(MockitoMockMetadataService.getMock());
    }
}

Answer 1

是的。在 Junit5 中可以创建custom extensions。事实上，这正是 @ExtendWith 的用途：

为特定的所有测试注册一个自定义YourExtension 类及其子类，您可以将测试类注释为 跟随。

@ExtendWith(YourExtension.class)
class YourTests {
    // ...
}

---

BeforeAllCallback、AfterAllCallback 的示例可以在spring-framework. 找到

---

多个扩展可以像这样一起注册：

@ExtendWith({ YourExtension1.class, YourExtension2.class })
class YourTests {