在 Extensions 中保存状态
通常,一个扩展只实例化一次。 因此,问题变得相关了:如何将状态从一个扩展调用保存到下一个扩展?ExtensionContext API为此目的提供了一个存储。 扩展可以将值放入存储中以便以后检索。
请参阅 TimingExtension,了解如何使用带有方法级范围的存储。 重要的是要记住,在测试执行期间存储在 ExtensionContext 中的值在周围的 ExtensionContext 中不可用。 由于可嵌套扩展上下文,内部上下文的范围也可能是有限的。有关通过存储存储存储和检索值的可用方法的详细信息,请参阅相应的JavaDoc。
使用带有方法级范围的存储
import java.lang.reflect.Method;
import java.util.logging.Logger;
import org.junit.jupiter.api.extension.AfterTestExecutionCallback;
import org.junit.jupiter.api.extension.BeforeTestExecutionCallback;
import org.junit.jupiter.api.extension.ExtensionContext;
import org.junit.jupiter.api.extension.ExtensionContext.Namespace;
import org.junit.jupiter.api.extension.ExtensionContext.Store;
public class TimingExtension implements BeforeTestExecutionCallback, AfterTestExecutionCallback {
private static final Logger logger = Logger.getLogger(TimingExtension.class.getName());
private static final String START_TIME = "start time";
@Override
public void beforeTestExecution(ExtensionContext context) throws Exception {
getStore(context).put(START_TIME, System.currentTimeMillis());
}
@Override
public void afterTestExecution(ExtensionContext context) throws Exception {
Method testMethod = context.getRequiredTestMethod();
long startTime = getStore(context).remove(START_TIME, long.class);
long duration = System.currentTimeMillis() - startTime;
logger.info(() -> String.format("Method [%s] took %s ms.", testMethod.getName(), duration));
}
private Store getStore(ExtensionContext context) {
return context.getStore(Namespace.create(getClass(), context.getRequiredTestMethod()));
}
}
Store API
ExtensionContext.Store.CloseableResource
扩展上下文存储绑定到其扩展上下文生命周期。
当扩展上下文生命周期结束时,它关闭关联的存储。通过调用close()方法通知属于CloseableResource实例的所有存储值。