org.eclipse.virgo.ide.ui.tests/src/org/eclipse/virgo/ide/ui/tests/util/InSection.java

/*******************************************************************************
 * Copyright (c) 2007, 2009, 2010 SpringSource
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     SpringSource - initial API and implementation
 *******************************************************************************/

package org.eclipse.virgo.ide.ui.tests.util;

import static org.eclipse.swtbot.swt.finder.matchers.WidgetMatcherFactory.withMnemonic;

import org.eclipse.swt.widgets.Widget;
import org.eclipse.swtbot.swt.finder.finders.PathGenerator;
import org.eclipse.swtbot.swt.finder.matchers.AbstractMatcher;
import org.eclipse.swtbot.swt.finder.utils.SWTUtils;
import org.eclipse.swtbot.swt.finder.utils.TreePath;
import org.eclipse.ui.forms.widgets.Section;
import org.hamcrest.Description;
import org.hamcrest.Factory;
import org.hamcrest.Matcher;

/**
 * @author Leo Dos Santos
 */
public class InSection<T extends Widget> extends AbstractMatcher<T> {

    @Factory
    public static <T extends Widget> Matcher<T> inSection(Matcher<?> matcher) {
        return new InSection<T>(matcher);
    }

    @Factory
    public static <T extends Widget> Matcher<T> inSection(String labelText) {
        return new InSection<T>(labelText);
    }

    private final Matcher<?> matcher;

    InSection(Matcher<?> matcher) {
        this.matcher = matcher;
    }

    InSection(String labelText) {
        this.matcher = withMnemonic(labelText);
    }

    public void describeTo(Description description) {
        description.appendText("in section (").appendDescriptionOf(this.matcher).appendText(")"); //$NON-NLS-1$ //$NON-NLS-2$
    }

    @Override
    protected boolean doMatch(Object item) {
        Widget previousWidget = SWTUtils.previousWidget((Widget) item);
        TreePath path = new PathGenerator().getPath((Widget) item);
        int segmentCount = path.getSegmentCount();
        for (int i = 1; i < segmentCount; i++) {
            previousWidget = (Widget) path.getSegment(segmentCount - i - 1);
            if (previousWidget instanceof Section && this.matcher.matches(previousWidget)) {
                return true;
            }
        }
        return false;
    }

}