SwingRunner.java

package org.osmorc;

import com.intellij.util.ui.UIUtil;
import org.junit.runner.notification.RunNotifier;
import org.junit.runners.BlockJUnit4ClassRunner;

public class SwingRunner extends BlockJUnit4ClassRunner {
  public SwingRunner(Class<?> klass) throws org.junit.runners.model.InitializationError {
    super(klass);
  }

  @Override
  public void run(final RunNotifier arg0) {
    UIUtil.invokeAndWaitIfNeeded(new Runnable() {
      @Override
      public void run() {
        SwingRunner.super.run(arg0);
      }
    });
  }
}