在JUnit5中，我如何从扩展调用测试类方法？

Question

在JUnit5中，我试图获得一个从扩展运行的测试类方法。我正在使用JUnit5扩展接口TestWatcher，并重写testFailed（）方法。

这个扩展的目的是在测试类的Selenium WebDriver浏览器中拍摄失败的屏幕快照，并将其附加到该测试的诱惑报告中。test class方法有实例化的浏览器和注释，用于附加到Allure。并且我的takeScreenshot方法依赖浏览器和来自测试类的testName字符串来正确运行。

package utils;

public class ScreenshotOnFailureExtension implements TestWatcher{
    @Override
    public void testFailed(ExtensionContext context, Throwable cause) {
        try {
            Object clazz = context.getRequiredTestInstance();
            Method takeScreenshot = clazz.getClass().getMethod("takeScreenshot");
            takeScreenshot.setAccessible(true);
            Object test = clazz.getClass().getConstructor().newInstance();
            takeScreenshot.invoke(test);
        } catch (Exception e) {
            e.printStackTrace();
        } 
}

我的测试类中的代码是这样的：

package tests;

@ExtendWith(ScreenshotOnFailureExtension.class)
public class MyTest implements Config {
    public WebDriver driver;
    public String testName;

//bunch of Junit5 annotations with functions to initialize above variables omitted...

    //take a screen shot
    public void takeScreenshot() {
        System.out.println("Taking screenshot.");
        byte[] srcFile=((TakesScreenshot)driver).getScreenshotAs(OutputType.BYTES);
        saveScreenshot(srcFile, testName+ ".png");
    }
    
    //this attaches screenshot to an allure test result
    @Attachment(value = "{testName}", type = "image/png")
    public byte[] saveScreenshot(byte[] screenShot, String testName) {
        System.out.println("Attaching screenshot to Allure report");
        return screenShot;
    }
}

上面的测试类能够在测试方法中从@aftereach调用时正确拍摄屏幕快照。但我只想在失败的时候接受它。

当我运行测试时，它调用takeScreenshot，但在执行时给出一个异常：

正在获取Screenshot.java.lang.Reflect.InvocationTargetException

null

您可以看到我的日志记录语句在该方法中的下一行代码（引用测试实例中的驱动程序）导致的NullPointerException之前输出。有没有正确的方法在上下文中触发现有测试实例的TakesCreenShot()方法？

或

如果有更简单的方法直接在测试的@aftereach方法中拍摄失败时的屏幕快照，请告诉我。似乎是一个非常基本的用例。：）

Answer 1

您不应该在扩展中实例化测试类，框架应该负责所有的事情。

请参考文档中的https://junit.org/junit5/docs/current/user-guide/#extensions 5.9.1，并查看此问答

您可以使用它，也可以修改您的TestWatcher以按照注释中的建议执行屏幕截图。您必须将驱动程序引用保存在ExtensionContext中才能访问它。

Answer 2

IMO问题就在你所描述的流程中。JUnit为每个测试方法创建一个测试类的新实例（尽管这是可以更改的）。

更好的方法是：

1. 使扩展“有状态”，因为它将包含对web驱动程序的引用。
2. 不要在测试中实现TakesCreenShot方法，而是在扩展（私有方法）中实现
3. 在扩展中实现回调，并将WebDriver的实例“注入”（通过反射）到测试中，如果您需要在测试中使用它。这将保证测试使用正确实例化的“状态”（webdriver的实例）运行。
4. 在扩展中实现了“如果测试方法失败，则调用扩展的私有方法TakesCreenShot
”的逻辑