org.eclipse.jdt.junit/src/org/eclipse/jdt/internal/junit/wizards/NewTestCaseCreationWizardPage2.java - jdt/eclipse.jdt.ui - Git at Google

 /*******************************************************************************
  * Copyright (c) 2000, 2003 IBM Corporation and others.
  * All rights reserved. This program and the accompanying materials
  * are made available under the terms of the Common Public License v1.0
  * which accompanies this distribution, and is available at
  * http://www.eclipse.org/legal/cpl-v10.html
  *
  * Contributors:
  *     IBM Corporation - initial API and implementation
  *******************************************************************************/
 package org.eclipse.jdt.internal.junit.wizards;

 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Vector;

 import org.eclipse.jdt.core.Flags;
 import org.eclipse.jdt.core.IMethod;
 import org.eclipse.jdt.core.IType;
 import org.eclipse.jdt.core.ITypeHierarchy;
 import org.eclipse.jdt.core.JavaModelException;
 import org.eclipse.jdt.internal.junit.ui.IJUnitHelpContextIds;
 import org.eclipse.jdt.internal.junit.ui.JUnitPlugin;
 import org.eclipse.jdt.internal.ui.viewsupport.AppearanceAwareLabelProvider;

 import org.eclipse.jface.dialogs.Dialog;
 import org.eclipse.jface.dialogs.IDialogSettings;
 import org.eclipse.jface.viewers.CheckStateChangedEvent;
 import org.eclipse.jface.viewers.ICheckStateListener;
 import org.eclipse.jface.viewers.ITreeContentProvider;
 import org.eclipse.jface.viewers.StructuredSelection;
 import org.eclipse.jface.viewers.Viewer;
 import org.eclipse.jface.viewers.ViewerFilter;
 import org.eclipse.jface.wizard.WizardPage;
 import org.eclipse.swt.SWT;
 import org.eclipse.swt.events.SelectionAdapter;
 import org.eclipse.swt.events.SelectionEvent;
 import org.eclipse.swt.layout.GridData;
 import org.eclipse.swt.layout.GridLayout;
 import org.eclipse.swt.widgets.Button;
 import org.eclipse.swt.widgets.Composite;
 import org.eclipse.swt.widgets.Label;
 import org.eclipse.ui.help.WorkbenchHelp;
 import org.eclipse.ui.internal.dialogs.ContainerCheckedTreeViewer;

 /**
  * Wizard page to select the methods from a class under test.
  */
 public class NewTestCaseCreationWizardPage2 extends WizardPage {

 	private final static String PAGE_NAME= "NewTestCaseCreationWizardPage2"; //$NON-NLS-1$
 	private final static String STORE_USE_TASKMARKER= PAGE_NAME + ".USE_TASKMARKER"; //$NON-NLS-1$
 	private final static String STORE_CREATE_FINAL_METHOD_STUBS= PAGE_NAME + ".CREATE_FINAL_METHOD_STUBS"; //$NON-NLS-1$
 	public final static String PREFIX= "test"; //$NON-NLS-1$

 	private NewTestCaseCreationWizardPage fFirstPage;
 	private IType fClassToTest;

 	private Button fCreateFinalMethodStubsButton;
 	private Button fCreateTasksButton;
 	private ContainerCheckedTreeViewer fMethodsTree;
 	private Button fSelectAllButton;
 	private Button fDeselectAllButton;
 	private Label fSelectedMethodsLabel;

 	/**
 	 * Constructor for NewTestCaseCreationWizardPage2.
 	 */
 	protected NewTestCaseCreationWizardPage2(NewTestCaseCreationWizardPage firstPage) {
 		super(PAGE_NAME);
 		fFirstPage= firstPage;
 		setTitle(WizardMessages.getString("NewTestClassWizPage2.title")); //$NON-NLS-1$
 		setDescription(WizardMessages.getString("NewTestClassWizPage2.description")); //$NON-NLS-1$
 	}

 	/**
 	 * @see org.eclipse.jface.dialogs.IDialogPage#createControl(Composite)
 	 */
 	public void createControl(Composite parent) {
 		Composite container= new Composite(parent, SWT.NONE);
 		GridLayout layout= new GridLayout();
 		layout.numColumns= 2;
 		container.setLayout(layout);

 		createMethodsTreeControls(container);
 		createSpacer(container);
 		createButtonChoices(container);
 		setControl(container);
 		restoreWidgetValues();
 		Dialog.applyDialogFont(container);
 		WorkbenchHelp.setHelp(container, IJUnitHelpContextIds.NEW_TESTCASE_WIZARD_PAGE2);
 	}

 	protected void createButtonChoices(Composite container) {
 		GridLayout layout;
 		GridData gd;
 		Composite prefixContainer= new Composite(container, SWT.NONE);
 		gd= new GridData();
 		gd.horizontalAlignment = GridData.FILL;
 		gd.horizontalSpan = 1;
 		prefixContainer.setLayoutData(gd);

 		layout = new GridLayout();
 		layout.numColumns = 1;
 		layout.marginWidth = 0;
 		layout.marginHeight = 0;
 		prefixContainer.setLayout(layout);

 		Button buttons[] = {null, null};

 		String buttonNames[] = {
 			WizardMessages.getString("NewTestClassWizPage2.create_final_method_stubs.text"), //$NON-NLS-1$
 			WizardMessages.getString("NewTestClassWizPage2.create_tasks.text") //$NON-NLS-1$
 		};

 		for (int i=0; i < buttons.length; i++) {
 			buttons[i]= new Button(prefixContainer, SWT.CHECK | SWT.LEFT);
 			buttons[i].setText(buttonNames[i]); //$NON-NLS-1$
 			buttons[i].setEnabled(true);
 			buttons[i].setSelection(true);
 			gd= new GridData();
 			gd.horizontalAlignment= GridData.FILL;
 			gd.horizontalSpan= 1;
 			buttons[i].setLayoutData(gd);
 		}
 		fCreateFinalMethodStubsButton= buttons[0];
 		fCreateTasksButton= buttons[1];
 	}

 	protected void createMethodsTreeControls(Composite container) {
 		Label label= new Label(container, SWT.LEFT | SWT.WRAP);
 		label.setFont(container.getFont());
 		label.setText(WizardMessages.getString("NewTestClassWizPage2.methods_tree.label")); //$NON-NLS-1$
 		GridData gd = new GridData();
 		gd.horizontalSpan = 2;
 		label.setLayoutData(gd);

 		fMethodsTree= new ContainerCheckedTreeViewer(container, SWT.BORDER);
 		gd= new GridData(GridData.FILL_BOTH | GridData.GRAB_HORIZONTAL | GridData.GRAB_VERTICAL);
 		gd.heightHint= 180;
 		fMethodsTree.getTree().setLayoutData(gd);

 		fMethodsTree.setLabelProvider(new AppearanceAwareLabelProvider());
 		fMethodsTree.setAutoExpandLevel(2);
 		fMethodsTree.addCheckStateListener(new ICheckStateListener() {
 			public void checkStateChanged(CheckStateChangedEvent event) {
 				updateSelectedMethodsLabel();
 			}
 		});
 		fMethodsTree.addFilter(new ViewerFilter() {
 			public boolean select(Viewer viewer, Object parentElement, Object element) {
 				if (element instanceof IMethod) {
 					IMethod method = (IMethod) element;
 					return !method.getElementName().equals("<clinit>"); //$NON-NLS-1$
 				}
 				return true;
 			}
 		});

 		Composite buttonContainer= new Composite(container, SWT.NONE);
 		gd= new GridData(GridData.FILL_VERTICAL);
 		buttonContainer.setLayoutData(gd);
 		GridLayout buttonLayout= new GridLayout();
 		buttonLayout.marginWidth= 0;
 		buttonLayout.marginHeight= 0;
 		buttonContainer.setLayout(buttonLayout);

 		fSelectAllButton= new Button(buttonContainer, SWT.PUSH);
 		fSelectAllButton.setText(WizardMessages.getString("NewTestClassWizPage2.selectAll")); //$NON-NLS-1$
 		gd= new GridData(GridData.FILL_HORIZONTAL | GridData.VERTICAL_ALIGN_BEGINNING);
 		fSelectAllButton.setLayoutData(gd);
 		fSelectAllButton.addSelectionListener(new SelectionAdapter() {
 			public void widgetSelected(SelectionEvent e) {
 				fMethodsTree.setCheckedElements((Object[]) fMethodsTree.getInput());
 				updateSelectedMethodsLabel();
 			}
 		});

 		fDeselectAllButton= new Button(buttonContainer, SWT.PUSH);
 		fDeselectAllButton.setText(WizardMessages.getString("NewTestClassWizPage2.deselectAll")); //$NON-NLS-1$
 		gd= new GridData(GridData.FILL_HORIZONTAL | GridData.VERTICAL_ALIGN_BEGINNING);
 		fDeselectAllButton.setLayoutData(gd);
 		fDeselectAllButton.addSelectionListener(new SelectionAdapter() {
 			public void widgetSelected(SelectionEvent e) {
 				fMethodsTree.setCheckedElements(new Object[0]);
 				updateSelectedMethodsLabel();
 			}
 		});

 		/* No of selected methods label */
 		fSelectedMethodsLabel= new Label(container, SWT.LEFT);
 		fSelectedMethodsLabel.setFont(container.getFont());
 		updateSelectedMethodsLabel();
 		gd= new GridData(GridData.FILL_HORIZONTAL);
 		gd.horizontalSpan= 1;
 		fSelectedMethodsLabel.setLayoutData(gd);

 		Label emptyLabel= new Label(container, SWT.LEFT);
 		gd= new GridData();
 		gd.horizontalSpan= 1;
 		emptyLabel.setLayoutData(gd);
 	}

 	protected void createSpacer(Composite container) {
 		Label spacer= new Label(container, SWT.NONE);
 		GridData data= new GridData();
 		data.horizontalSpan= 2;
 		data.horizontalAlignment= GridData.FILL;
 		data.verticalAlignment= GridData.BEGINNING;
 		data.heightHint= 4;
 		spacer.setLayoutData(data);
 	}

 	/**
 	 * @see org.eclipse.jface.dialogs.DialogPage#setVisible(boolean)
 	 */
 	public void setVisible(boolean visible) {
 		super.setVisible(visible);
 		if (visible) {
 			fClassToTest= fFirstPage.getClassToTest();
 			IType currType= fClassToTest;
 			ArrayList types= null;
 			try {
 				ITypeHierarchy hierarchy= currType.newSupertypeHierarchy(null);
 				IType[] superTypes;
 				if (currType.isClass())
 					superTypes= hierarchy.getAllSuperclasses(currType);
 				else if (currType.isInterface())
 					superTypes= hierarchy.getAllSuperInterfaces(currType);
 				else
 					superTypes= new IType[0];
 				types= new ArrayList(superTypes.length+1);
 				types.add(currType);
 				types.addAll(Arrays.asList(superTypes));
 			} catch(JavaModelException e) {
 				JUnitPlugin.log(e);
 			}
 			fMethodsTree.setContentProvider(new MethodsTreeContentProvider(types.toArray()));
 			if (types == null)
 				types= new ArrayList();
 			fMethodsTree.setInput(types.toArray());
 			fMethodsTree.setSelection(new StructuredSelection(currType), true);
 			updateSelectedMethodsLabel();
 			setFocus();
 		}
 	}

 	/**
 	 * Returns all checked methods in the Methods tree.
 	 */
 	public IMethod[] getCheckedMethods() {
 		Object[] checkedObjects= fMethodsTree.getCheckedElements();
 		int methodCount= 0;
 		for (int i = 0; i < checkedObjects.length; i++) {
 			if (checkedObjects[i] instanceof IMethod)
 				methodCount++;
 		}
 		IMethod[] checkedMethods= new IMethod[methodCount];
 		int j= 0;
 		for (int i = 0; i < checkedObjects.length; i++) {
 			if (checkedObjects[i] instanceof IMethod) {
 				checkedMethods[j]= (IMethod)checkedObjects[i];
 				j++;
 			}
 		}
 		return checkedMethods;
 	}

 	private static class MethodsTreeContentProvider implements ITreeContentProvider {
 		private Object[] fTypes;
 		private IMethod[] fMethods;
 		private final Object[] fEmpty= new Object[0];

 		public MethodsTreeContentProvider(Object[] types) {
 			fTypes= types;
 			Vector methods= new Vector();
 			for (int i = types.length-1; i > -1; i--) {
 				Object object = types[i];
 				if (object instanceof IType) {
 					IType type = (IType) object;
 					try {
 						IMethod[] currMethods= type.getMethods();
 						for_currMethods:
 						for (int j = 0; j < currMethods.length; j++) {
 							IMethod currMethod = currMethods[j];
 							int flags= currMethod.getFlags();
 							if (!Flags.isPrivate(flags)) {
 								for (int k = 0; k < methods.size(); k++) {
 									IMethod m= ((IMethod)methods.get(k));
 									if (m.getElementName().equals(currMethod.getElementName())
 										&& m.getSignature().equals(currMethod.getSignature())) {
 										methods.set(k,currMethod);
 										continue for_currMethods;
 									}
 								}
 								methods.add(currMethod);
 							}
 						}
 					} catch (JavaModelException e) {
 						JUnitPlugin.log(e);
 					}
 				}
 			}
 			fMethods= new IMethod[methods.size()];
 			methods.copyInto(fMethods);
 		}

 		/*
 		 * @see ITreeContentProvider#getChildren(Object)
 		 */
 		public Object[] getChildren(Object parentElement) {
 			if (parentElement instanceof IType) {
 				IType parentType= (IType)parentElement;
 				ArrayList result= new ArrayList(fMethods.length);
 				for (int i= 0; i < fMethods.length; i++) {
 					if (fMethods[i].getDeclaringType().equals(parentType)) {
 						result.add(fMethods[i]);
 					}
 				}
 				return result.toArray();
 			}
 			return fEmpty;
 		}

 		/*
 		 * @see ITreeContentProvider#getParent(Object)
 		 */
 		public Object getParent(Object element) {
 			if (element instanceof IMethod)
 				return ((IMethod)element).getDeclaringType();
 			return null;
 		}

 		/*
 		 * @see ITreeContentProvider#hasChildren(Object)
 		 */
 		public boolean hasChildren(Object element) {
 			return getChildren(element).length > 0;
 		}

 		/*
 		 * @see IStructuredContentProvider#getElements(Object)
 		 */
 		public Object[] getElements(Object inputElement) {
 			return fTypes;
 		}

 		/*
 		 * @see IContentProvider#dispose()
 		 */
 		public void dispose() {
 		}

 		/*
 		 * @see IContentProvider#inputChanged(Viewer, Object, Object)
 		 */
 		public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
 		}

 		public IMethod[] getAllMethods() {
 			return fMethods;
 		}
 	}

 	/**
 	 * Returns true if the checkbox for creating tasks is checked.
 	 */
 	public boolean getCreateTasksButtonSelection() {
 		return fCreateTasksButton.getSelection();
 	}

 	/**
 	 * Returns true if the checkbox for final method stubs is checked.
 	 */
 	public boolean getCreateFinalMethodStubsButtonSelection() {
 		return fCreateFinalMethodStubsButton.getSelection();
 	}

 	private void updateSelectedMethodsLabel() {
 		Object[] checked= fMethodsTree.getCheckedElements();
 		int checkedMethodCount= 0;
 		for (int i= 0; i < checked.length; i++) {
 			if (checked[i] instanceof IMethod)
 				checkedMethodCount++;
 		}
 		String label= ""; //$NON-NLS-1$
 		if (checkedMethodCount == 1)
 			label= WizardMessages.getFormattedString("NewTestClassWizPage2.selected_methods.label_one", new Integer(checkedMethodCount)); //$NON-NLS-1$
 		else
 			label= WizardMessages.getFormattedString("NewTestClassWizPage2.selected_methods.label_many", new Integer(checkedMethodCount)); //$NON-NLS-1$
 		fSelectedMethodsLabel.setText(label);
 	}

 	/**
 	 * Returns all the methods in the Methods tree.
 	 */
 	public IMethod[] getAllMethods() {
 		return ((MethodsTreeContentProvider)fMethodsTree.getContentProvider()).getAllMethods();
 	}

 	/**
 	 * Sets the focus on the type name.
 	 */
 	protected void setFocus() {
 		fMethodsTree.getControl().setFocus();
 	}

 	/**
 	 *	Use the dialog store to restore widget values to the values that they held
 	 *	last time this wizard was used to completion
 	 */
 	private void restoreWidgetValues() {
 		IDialogSettings settings= getDialogSettings();
 		if (settings != null) {
 			fCreateTasksButton.setSelection(settings.getBoolean(STORE_USE_TASKMARKER));
 			fCreateFinalMethodStubsButton.setSelection(settings.getBoolean(STORE_CREATE_FINAL_METHOD_STUBS));
 		}
 	}

 	/**
 	 * 	Since Finish was pressed, write widget values to the dialog store so that they
 	 *	will persist into the next invocation of this wizard page
 	 */
 	void saveWidgetValues() {
 		IDialogSettings settings= getDialogSettings();
 		if (settings != null) {
 			settings.put(STORE_USE_TASKMARKER, fCreateTasksButton.getSelection());
 			settings.put(STORE_CREATE_FINAL_METHOD_STUBS, fCreateFinalMethodStubsButton.getSelection());
 		}
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2000, 2003 IBM Corporation and others.
	* All rights reserved. This program and the accompanying materials
	* are made available under the terms of the Common Public License v1.0
	* which accompanies this distribution, and is available at
	* http://www.eclipse.org/legal/cpl-v10.html
	*
	* Contributors:
	* IBM Corporation - initial API and implementation
	*******************************************************************************/
	package org.eclipse.jdt.internal.junit.wizards;

	import java.util.ArrayList;
	import java.util.Arrays;
	import java.util.Vector;

	import org.eclipse.jdt.core.Flags;
	import org.eclipse.jdt.core.IMethod;
	import org.eclipse.jdt.core.IType;
	import org.eclipse.jdt.core.ITypeHierarchy;
	import org.eclipse.jdt.core.JavaModelException;
	import org.eclipse.jdt.internal.junit.ui.IJUnitHelpContextIds;
	import org.eclipse.jdt.internal.junit.ui.JUnitPlugin;
	import org.eclipse.jdt.internal.ui.viewsupport.AppearanceAwareLabelProvider;

	import org.eclipse.jface.dialogs.Dialog;
	import org.eclipse.jface.dialogs.IDialogSettings;
	import org.eclipse.jface.viewers.CheckStateChangedEvent;
	import org.eclipse.jface.viewers.ICheckStateListener;
	import org.eclipse.jface.viewers.ITreeContentProvider;
	import org.eclipse.jface.viewers.StructuredSelection;
	import org.eclipse.jface.viewers.Viewer;
	import org.eclipse.jface.viewers.ViewerFilter;
	import org.eclipse.jface.wizard.WizardPage;
	import org.eclipse.swt.SWT;
	import org.eclipse.swt.events.SelectionAdapter;
	import org.eclipse.swt.events.SelectionEvent;
	import org.eclipse.swt.layout.GridData;
	import org.eclipse.swt.layout.GridLayout;
	import org.eclipse.swt.widgets.Button;
	import org.eclipse.swt.widgets.Composite;
	import org.eclipse.swt.widgets.Label;
	import org.eclipse.ui.help.WorkbenchHelp;
	import org.eclipse.ui.internal.dialogs.ContainerCheckedTreeViewer;

	/**
	* Wizard page to select the methods from a class under test.
	*/
	public class NewTestCaseCreationWizardPage2 extends WizardPage {

	private final static String PAGE_NAME= "NewTestCaseCreationWizardPage2"; //$NON-NLS-1$
	private final static String STORE_USE_TASKMARKER= PAGE_NAME + ".USE_TASKMARKER"; //$NON-NLS-1$
	private final static String STORE_CREATE_FINAL_METHOD_STUBS= PAGE_NAME + ".CREATE_FINAL_METHOD_STUBS"; //$NON-NLS-1$
	public final static String PREFIX= "test"; //$NON-NLS-1$

	private NewTestCaseCreationWizardPage fFirstPage;
	private IType fClassToTest;

	private Button fCreateFinalMethodStubsButton;
	private Button fCreateTasksButton;
	private ContainerCheckedTreeViewer fMethodsTree;
	private Button fSelectAllButton;
	private Button fDeselectAllButton;
	private Label fSelectedMethodsLabel;

	/**
	* Constructor for NewTestCaseCreationWizardPage2.
	*/
	protected NewTestCaseCreationWizardPage2(NewTestCaseCreationWizardPage firstPage) {
	super(PAGE_NAME);
	fFirstPage= firstPage;
	setTitle(WizardMessages.getString("NewTestClassWizPage2.title")); //$NON-NLS-1$
	setDescription(WizardMessages.getString("NewTestClassWizPage2.description")); //$NON-NLS-1$
	}

	/**
	* @see org.eclipse.jface.dialogs.IDialogPage#createControl(Composite)
	*/
	public void createControl(Composite parent) {
	Composite container= new Composite(parent, SWT.NONE);
	GridLayout layout= new GridLayout();
	layout.numColumns= 2;
	container.setLayout(layout);

	createMethodsTreeControls(container);
	createSpacer(container);
	createButtonChoices(container);
	setControl(container);
	restoreWidgetValues();
	Dialog.applyDialogFont(container);
	WorkbenchHelp.setHelp(container, IJUnitHelpContextIds.NEW_TESTCASE_WIZARD_PAGE2);
	}

	protected void createButtonChoices(Composite container) {
	GridLayout layout;
	GridData gd;
	Composite prefixContainer= new Composite(container, SWT.NONE);
	gd= new GridData();
	gd.horizontalAlignment = GridData.FILL;
	gd.horizontalSpan = 1;
	prefixContainer.setLayoutData(gd);

	layout = new GridLayout();
	layout.numColumns = 1;
	layout.marginWidth = 0;
	layout.marginHeight = 0;
	prefixContainer.setLayout(layout);

	Button buttons[] = {null, null};

	String buttonNames[] = {
	WizardMessages.getString("NewTestClassWizPage2.create_final_method_stubs.text"), //$NON-NLS-1$
	WizardMessages.getString("NewTestClassWizPage2.create_tasks.text") //$NON-NLS-1$
	};

	for (int i=0; i < buttons.length; i++) {
	buttons[i]= new Button(prefixContainer, SWT.CHECK \| SWT.LEFT);
	buttons[i].setText(buttonNames[i]); //$NON-NLS-1$
	buttons[i].setEnabled(true);
	buttons[i].setSelection(true);
	gd= new GridData();
	gd.horizontalAlignment= GridData.FILL;
	gd.horizontalSpan= 1;
	buttons[i].setLayoutData(gd);
	}
	fCreateFinalMethodStubsButton= buttons[0];
	fCreateTasksButton= buttons[1];
	}

	protected void createMethodsTreeControls(Composite container) {
	Label label= new Label(container, SWT.LEFT \| SWT.WRAP);
	label.setFont(container.getFont());
	label.setText(WizardMessages.getString("NewTestClassWizPage2.methods_tree.label")); //$NON-NLS-1$
	GridData gd = new GridData();
	gd.horizontalSpan = 2;
	label.setLayoutData(gd);

	fMethodsTree= new ContainerCheckedTreeViewer(container, SWT.BORDER);
	gd= new GridData(GridData.FILL_BOTH \| GridData.GRAB_HORIZONTAL \| GridData.GRAB_VERTICAL);
	gd.heightHint= 180;
	fMethodsTree.getTree().setLayoutData(gd);

	fMethodsTree.setLabelProvider(new AppearanceAwareLabelProvider());
	fMethodsTree.setAutoExpandLevel(2);
	fMethodsTree.addCheckStateListener(new ICheckStateListener() {
	public void checkStateChanged(CheckStateChangedEvent event) {
	updateSelectedMethodsLabel();
	}
	});
	fMethodsTree.addFilter(new ViewerFilter() {
	public boolean select(Viewer viewer, Object parentElement, Object element) {
	if (element instanceof IMethod) {
	IMethod method = (IMethod) element;
	return !method.getElementName().equals("<clinit>"); //$NON-NLS-1$
	}
	return true;
	}
	});

	Composite buttonContainer= new Composite(container, SWT.NONE);
	gd= new GridData(GridData.FILL_VERTICAL);
	buttonContainer.setLayoutData(gd);
	GridLayout buttonLayout= new GridLayout();
	buttonLayout.marginWidth= 0;
	buttonLayout.marginHeight= 0;
	buttonContainer.setLayout(buttonLayout);

	fSelectAllButton= new Button(buttonContainer, SWT.PUSH);
	fSelectAllButton.setText(WizardMessages.getString("NewTestClassWizPage2.selectAll")); //$NON-NLS-1$
	gd= new GridData(GridData.FILL_HORIZONTAL \| GridData.VERTICAL_ALIGN_BEGINNING);
	fSelectAllButton.setLayoutData(gd);
	fSelectAllButton.addSelectionListener(new SelectionAdapter() {
	public void widgetSelected(SelectionEvent e) {
	fMethodsTree.setCheckedElements((Object[]) fMethodsTree.getInput());
	updateSelectedMethodsLabel();
	}
	});

	fDeselectAllButton= new Button(buttonContainer, SWT.PUSH);
	fDeselectAllButton.setText(WizardMessages.getString("NewTestClassWizPage2.deselectAll")); //$NON-NLS-1$
	gd= new GridData(GridData.FILL_HORIZONTAL \| GridData.VERTICAL_ALIGN_BEGINNING);
	fDeselectAllButton.setLayoutData(gd);
	fDeselectAllButton.addSelectionListener(new SelectionAdapter() {
	public void widgetSelected(SelectionEvent e) {
	fMethodsTree.setCheckedElements(new Object[0]);
	updateSelectedMethodsLabel();
	}
	});

	/* No of selected methods label */
	fSelectedMethodsLabel= new Label(container, SWT.LEFT);
	fSelectedMethodsLabel.setFont(container.getFont());
	updateSelectedMethodsLabel();
	gd= new GridData(GridData.FILL_HORIZONTAL);
	gd.horizontalSpan= 1;
	fSelectedMethodsLabel.setLayoutData(gd);

	Label emptyLabel= new Label(container, SWT.LEFT);
	gd= new GridData();
	gd.horizontalSpan= 1;
	emptyLabel.setLayoutData(gd);
	}

	protected void createSpacer(Composite container) {
	Label spacer= new Label(container, SWT.NONE);
	GridData data= new GridData();
	data.horizontalSpan= 2;
	data.horizontalAlignment= GridData.FILL;
	data.verticalAlignment= GridData.BEGINNING;
	data.heightHint= 4;
	spacer.setLayoutData(data);
	}

	/**
	* @see org.eclipse.jface.dialogs.DialogPage#setVisible(boolean)
	*/
	public void setVisible(boolean visible) {
	super.setVisible(visible);
	if (visible) {
	fClassToTest= fFirstPage.getClassToTest();
	IType currType= fClassToTest;
	ArrayList types= null;
	try {
	ITypeHierarchy hierarchy= currType.newSupertypeHierarchy(null);
	IType[] superTypes;
	if (currType.isClass())
	superTypes= hierarchy.getAllSuperclasses(currType);
	else if (currType.isInterface())
	superTypes= hierarchy.getAllSuperInterfaces(currType);
	else
	superTypes= new IType[0];
	types= new ArrayList(superTypes.length+1);
	types.add(currType);
	types.addAll(Arrays.asList(superTypes));
	} catch(JavaModelException e) {
	JUnitPlugin.log(e);
	}
	fMethodsTree.setContentProvider(new MethodsTreeContentProvider(types.toArray()));
	if (types == null)
	types= new ArrayList();
	fMethodsTree.setInput(types.toArray());
	fMethodsTree.setSelection(new StructuredSelection(currType), true);
	updateSelectedMethodsLabel();
	setFocus();
	}
	}

	/**
	* Returns all checked methods in the Methods tree.
	*/
	public IMethod[] getCheckedMethods() {
	Object[] checkedObjects= fMethodsTree.getCheckedElements();
	int methodCount= 0;
	for (int i = 0; i < checkedObjects.length; i++) {
	if (checkedObjects[i] instanceof IMethod)
	methodCount++;
	}
	IMethod[] checkedMethods= new IMethod[methodCount];
	int j= 0;
	for (int i = 0; i < checkedObjects.length; i++) {
	if (checkedObjects[i] instanceof IMethod) {
	checkedMethods[j]= (IMethod)checkedObjects[i];
	j++;
	}
	}
	return checkedMethods;
	}

	private static class MethodsTreeContentProvider implements ITreeContentProvider {
	private Object[] fTypes;
	private IMethod[] fMethods;
	private final Object[] fEmpty= new Object[0];

	public MethodsTreeContentProvider(Object[] types) {
	fTypes= types;
	Vector methods= new Vector();
	for (int i = types.length-1; i > -1; i--) {
	Object object = types[i];
	if (object instanceof IType) {
	IType type = (IType) object;
	try {
	IMethod[] currMethods= type.getMethods();
	for_currMethods:
	for (int j = 0; j < currMethods.length; j++) {
	IMethod currMethod = currMethods[j];
	int flags= currMethod.getFlags();
	if (!Flags.isPrivate(flags)) {
	for (int k = 0; k < methods.size(); k++) {
	IMethod m= ((IMethod)methods.get(k));
	if (m.getElementName().equals(currMethod.getElementName())
	&& m.getSignature().equals(currMethod.getSignature())) {
	methods.set(k,currMethod);
	continue for_currMethods;
	}
	}
	methods.add(currMethod);
	}
	}
	} catch (JavaModelException e) {
	JUnitPlugin.log(e);
	}
	}
	}
	fMethods= new IMethod[methods.size()];
	methods.copyInto(fMethods);
	}

	/*
	* @see ITreeContentProvider#getChildren(Object)
	*/
	public Object[] getChildren(Object parentElement) {
	if (parentElement instanceof IType) {
	IType parentType= (IType)parentElement;
	ArrayList result= new ArrayList(fMethods.length);
	for (int i= 0; i < fMethods.length; i++) {
	if (fMethods[i].getDeclaringType().equals(parentType)) {
	result.add(fMethods[i]);
	}
	}
	return result.toArray();
	}
	return fEmpty;
	}

	/*
	* @see ITreeContentProvider#getParent(Object)
	*/
	public Object getParent(Object element) {
	if (element instanceof IMethod)
	return ((IMethod)element).getDeclaringType();
	return null;
	}

	/*
	* @see ITreeContentProvider#hasChildren(Object)
	*/
	public boolean hasChildren(Object element) {
	return getChildren(element).length > 0;
	}

	/*
	* @see IStructuredContentProvider#getElements(Object)
	*/
	public Object[] getElements(Object inputElement) {
	return fTypes;
	}

	/*
	* @see IContentProvider#dispose()
	*/
	public void dispose() {
	}

	/*
	* @see IContentProvider#inputChanged(Viewer, Object, Object)
	*/
	public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
	}

	public IMethod[] getAllMethods() {
	return fMethods;
	}
	}

	/**
	* Returns true if the checkbox for creating tasks is checked.
	*/
	public boolean getCreateTasksButtonSelection() {
	return fCreateTasksButton.getSelection();
	}

	/**
	* Returns true if the checkbox for final method stubs is checked.
	*/
	public boolean getCreateFinalMethodStubsButtonSelection() {
	return fCreateFinalMethodStubsButton.getSelection();
	}

	private void updateSelectedMethodsLabel() {
	Object[] checked= fMethodsTree.getCheckedElements();
	int checkedMethodCount= 0;
	for (int i= 0; i < checked.length; i++) {
	if (checked[i] instanceof IMethod)
	checkedMethodCount++;
	}
	String label= ""; //$NON-NLS-1$
	if (checkedMethodCount == 1)
	label= WizardMessages.getFormattedString("NewTestClassWizPage2.selected_methods.label_one", new Integer(checkedMethodCount)); //$NON-NLS-1$
	else
	label= WizardMessages.getFormattedString("NewTestClassWizPage2.selected_methods.label_many", new Integer(checkedMethodCount)); //$NON-NLS-1$
	fSelectedMethodsLabel.setText(label);
	}

	/**
	* Returns all the methods in the Methods tree.
	*/
	public IMethod[] getAllMethods() {
	return ((MethodsTreeContentProvider)fMethodsTree.getContentProvider()).getAllMethods();
	}

	/**
	* Sets the focus on the type name.
	*/
	protected void setFocus() {
	fMethodsTree.getControl().setFocus();
	}

	/**
	* Use the dialog store to restore widget values to the values that they held
	* last time this wizard was used to completion
	*/
	private void restoreWidgetValues() {
	IDialogSettings settings= getDialogSettings();
	if (settings != null) {
	fCreateTasksButton.setSelection(settings.getBoolean(STORE_USE_TASKMARKER));
	fCreateFinalMethodStubsButton.setSelection(settings.getBoolean(STORE_CREATE_FINAL_METHOD_STUBS));
	}
	}

	/**
	* Since Finish was pressed, write widget values to the dialog store so that they
	* will persist into the next invocation of this wizard page
	*/
	void saveWidgetValues() {
	IDialogSettings settings= getDialogSettings();
	if (settings != null) {
	settings.put(STORE_USE_TASKMARKER, fCreateTasksButton.getSelection());
	settings.put(STORE_CREATE_FINAL_METHOD_STUBS, fCreateFinalMethodStubsButton.getSelection());
	}
	}
	}