java - 停止 JUnit 完成，直到所有其他线程都已停止

Question

在单元测试期间，我想使用 Java FX 绘制一些图形。现在的问题是，一旦单元测试完成，JVM 和 Java FX 就会关闭，我无法检查生成的图(与刚刚从 main 方法启动“测试”的情况不同) .所以我的问题是，有没有办法在特定线程完成之前阻止 JUnit 退出，即直接从 main 方法开始测试时复制行为？是的，我知道绘图很可能不是一般在单元测试期间应该完成的事情。目前我正在做这样的事情:

//@ContextConfiguration(classes = {DummyConfig.class }) 
//@RunWith(SpringJUnit4ClassRunner.class)
public class MainViewTest {

    private boolean fromMain = false;

    // starting the "test" from main does not require explicit waiting for 
    // for the JavaFX thread to finish .. I'd like to replicate this
    // behaviour in JUnit (by configuring JUnit, not my test or application code)
    public static void main(String [] args){
        new MainViewTest(true).test();
    }

    public MainViewTest(){}

    private MainViewTest(boolean fromMain){
        this.fromMain = fromMain;
    }

    @Test
    public void test() {

        //do some tests....

        //plot some results...
        PlotStage.plotStage(new QPMApplication() {
            @Override
            public Stage createStage() {
                Stage stage = new Stage();
                StackPane root = new StackPane();
                Scene scene = new Scene(root, 300, 300);
                stage.setTitle("Stage");
                stage.setScene(scene);
                stage.setOnCloseRequest(new EventHandler<WindowEvent>(){
                    @Override
                    public void handle(WindowEvent event) {
                        Platform.exit();
                    }
                 });
                return stage;
            }
        });

        System.out.println("Stage started");
        // how to get rid of this block (or using a countdownlatch) but
        // still waiting for the threads to finish?
        Set<Thread> threads = Thread.getAllStackTraces().keySet();
        if (!fromMain) {
            System.out.println("checking threads...");
            for (Thread thread : threads) {
                if (thread.getName().contains("JavaFX")) {
                    try {
                        thread.join();
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }
}

问题就在这里，我想摆脱这个讨厌的 block ，等待所有 JavaFX 平台明确退出。我很欣赏关于使用倒计时闩锁而不是明确加入 Java FX 线程的答案。然而，这仍然需要我明确停止当前线程。但是，我宁愿以某种方式“告诉”JUnit 等待 JavaFX 线程完成。

所以基本上我正在寻找的是一种告诉 JUnit 等待特定线程的方法，而我的测试方法中没有任何阻塞代码。

附录:最小运行示例所需的类

public class PlotStage {

    public static boolean toolkitInialized = false;

    public static void plotStage(QPMApplication stageCreator) {
        if (!toolkitInialized) {
            Thread appThread = new Thread(new Runnable() {
                @Override
                public void run() {
                    Application.launch(InitApp.class);
                }
            });
            appThread.start();
        }

        while (!toolkitInialized) {
            try {
                Thread.sleep(100);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        Platform.runLater(new Runnable() {
            @Override
            public void run() {
                Stage stage = stageCreator.createStage();
                stage.show();
            }
        });
    }

    public static class InitApp extends Application {
        @Override
        public void start(final Stage primaryStage) {
            toolkitInialized = true;
        }
    }
}


public interface QPMApplication {
   public abstract  Stage createStage();
}

Answer 1

使用 CountDownLatch为此。

1. 用1初始化。
2. 当 Stage 关闭时，调用 countDown()。
3. 在JUnit Test中，调用await()等待Stage关闭。

例子:

CountDownLatch cdl = new CountDownLatch(1);
// TODO show the stage and do not forget to add cdl.countDown() to your
//   stage.setOnCloseRequest
cdl.await();

---

备选方案#1:

使用 JavaFX Junit Rule直接在 FX 应用程序线程上执行所有操作。

---

备选方案 #2:

使用TestFX ，就我从您更新的描述中读到的内容而言，它最合适。