UnitTestCase.java

package unit;

import javafx.stage.Stage;
import org.jmock.Mockery;
import org.jmock.integration.junit4.JUnit4Mockery;
import org.jmock.lib.concurrent.Synchroniser;
import org.junit.After;
import org.testfx.api.FxToolkit;
import org.testfx.framework.junit.ApplicationTest;

import java.util.concurrent.TimeoutException;

public abstract class UnitTestCase extends ApplicationTest {
    protected final Mockery context = new UnitTestingMockery();

    @After
    public void assertGoodContext(){
        this.context.assertIsSatisfied();
    }

    private class UnitTestingMockery extends JUnit4Mockery {
        UnitTestingMockery(){
            setThreadingPolicy(new Synchroniser());
        }
    }

    @Override
    public void init() throws TimeoutException {
        FxToolkit.registerStage(Stage::new);
    }

    @Override
    public void start(Stage stage) throws Exception {
        // This is deliberate. Don't start the test
    }

    @Override
    public void stop() throws TimeoutException {
        FxToolkit.hideStage();
    }
}