Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
package org.eclipse.swtbot.jface;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import org.eclipse.jface.wizard.IWizard;
import org.eclipse.jface.wizard.WizardDialog;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swtbot.swt.finder.finders.UIThreadRunnable;
import org.eclipse.swtbot.swt.finder.junit.SWTBotJunit4ClassRunner;
import org.eclipse.swtbot.swt.finder.results.VoidResult;
import org.junit.runner.notification.RunNotifier;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.TestClass;
public class WizardDialogTestRunner extends SWTBotJunit4ClassRunner {
private IWizard newWizard = null;
public WizardDialogTestRunner(Class<?> klass) throws Exception {
super(klass);
}
@Override
protected void runChild(final FrameworkMethod method, final RunNotifier notifier) {
Shell parentShell = new Shell();
try {
newWizard = newWizard();
} catch (Exception e) {
throw new IllegalArgumentException(e);
}
final boolean[] finishPressed = new boolean[1];
final WizardDialog dialog = new WizardDialog(parentShell, newWizard) {
@Override
protected void finishPressed() {
super.finishPressed();
finishPressed[0] = true;
}
};
final Exception[] ex = new Exception[1];
final boolean[] done = new boolean[1];
Runnable r = new Runnable() {
public void run() {
try {
WizardDialogTestRunner.super.runChild(method, notifier);
} catch (Exception e) {
ex[0] = e;
} finally {
if (dialog.getShell() != null && dialog.getShell().isDisposed() == false) {
UIThreadRunnable.syncExec(new VoidResult() {
public void run() {
dialog.close();
}
});
}
done[0] = true;
}
}
};
Thread nonUIThread = new Thread(r);
nonUIThread.setName("Runnning Test Thread");//$NON-NLS-1$
nonUIThread.start();
dialog.open();
waitForNonUIThreadFinished(parentShell, nonUIThread);
if (ex[0] != null) {
throw new IllegalStateException(ex[0]);
}
}
private void waitForNonUIThreadFinished(Shell parentShell, Thread nonUIThread) {
Display display = parentShell.getDisplay();
while (nonUIThread.isAlive()) {
try {
if (!display.readAndDispatch()) {
display.sleep();
}
} catch (Throwable e) {
}
}
}
@Override
protected Object createTest() throws Exception {
Object test = super.createTest();
if (newWizard != null) {
try {
Field field = test.getClass().getField("wizard");
field.set(test, newWizard);
} catch (NoSuchFieldException e) {
}
}
return test;
}
private IWizard newWizard() throws InstantiationException, IllegalAccessException {
TestClass testClass = getTestClass();
Annotation[] annotations = testClass.getAnnotations();
IWizard newWizard = null;
for (Annotation annotation : annotations) {
if (annotation instanceof WithWizard) {
WithWizard wizardAnnotation = (WithWizard) annotation;
Class<? extends IWizard> value = wizardAnnotation.value();
newWizard = value.newInstance();
}
}
return newWizard;
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment