JUnit 5：将Spring组件注入扩展（BeforeAllCallback/AfterAllCallback）

Question

如何在所有测试运行之前将自定义数据提供程序实例化为Spring组件？

有没有一种聪明的方法将Spring组件注入到一个定制的JUnit Jupiter扩展中，该扩展在回调之前实现？beforeAll方法应该在使用@extendedwith（OncePerTestRunExtension.class）执行MyTestClass之前触发一个复杂的进程。

我创建了一个Spring启动应用程序（src/main/java），它为我的测试（src/test/java）提供了必要的数据。为测试准备数据可能需要几个小时。它还为我提供了对一些restendpoint的抽象访问。

在所有测试类的进程之间，数据不会发生变化。所以我只想提取一次数据。

在一个类中编写所有的测试是可行的，但是我认为将测试分为不同的类可以提供更好的概述。

Answer 1

下面是我如何实现这一点，使用dataSourcespringbean在我的数据库中设置一些测试数据，方法是将上面Sam的答案与下面的答案结合起来：https://stackoverflow.com/a/51556718/278800

import javax.sql.DataSource;
import org.junit.jupiter.api.extension.BeforeAllCallback;
import org.junit.jupiter.api.extension.ExtensionContext;
import org.springframework.context.ApplicationContext;
import org.springframework.test.context.junit.jupiter.SpringExtension;

public class TestDataSetup implements BeforeAllCallback, ExtensionContext.Store.CloseableResource {

  private static boolean started = false;

  private DataSource dataSource;

  @Override
  public void beforeAll(ExtensionContext extensionContext) {
    synchronized (TestDataSetup.class) {
      if (!started) {
        started = true;

        // get the dataSource bean from the spring context
        ApplicationContext springContext = SpringExtension.getApplicationContext(extensionContext);
        this.dataSource = springContext.getBean(DataSource.class);

        // TODO: put your one-time db initialization code here

        // register a callback hook for when the root test context is shut down
        extensionContext
            .getRoot()
            .getStore(ExtensionContext.Namespace.GLOBAL)
            .put("TestDataSetup-started", this);
      }
    }
  }

  @Override
  public void close() {
    synchronized (TestDataSetup.class) {
      // TODO: put your db cleanup code here
    }
  }

（我不能100%确定这个线程的安全性，所以我添加了synchronized块只是为了安全。）

要启用这个扩展，您只需要将这个注释添加到需要它的测试类中：

@ExtendWith(TestDataSetup.class)

好的方面是JUnit5允许多个扩展，因此即使您的测试已经用@extendedwith（SpringExtension.class）进行了注释，它也可以工作。

Answer 2

在自定义beforeAll回调的beforeAll（ExtensionContext）方法中，您可以通过SpringExtension访问当前测试类的SpringApplicationContext。getApplicationContext（extensionContext）。

如果您将自定义数据提供程序配置为该Application ationContext中的Spring组件，那么您可以从扩展中的Application ationContext中检索该组件-例如，通过Application ationContext.getBean（MyDataProvider.class）.

如果需要处理数据并在测试之间存储处理后的数据，可以将其存储在rootExtensionContext中。在JUnit Jupiter中存储。请参阅扩展上下文。getRoot（）和getorComputeFabSent（…） 变体。有关详细信息，请存储。