JavaSelectionTestUtils.java

/**
 * Copyright (c) 2011 Sebastian Proksch.
 * 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:
 *     Sebastian Proksch - initial API and implementation
 */
package org.eclipse.recommenders.tests.apidocs;

import static org.eclipse.recommenders.rcp.JavaElementSelectionEvent.JavaElementSelectionLocation.*;
import static org.mockito.Mockito.*;

import org.eclipse.jdt.core.IAnnotation;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.IMethod;
import org.eclipse.jdt.core.IType;
import org.eclipse.recommenders.rcp.JavaElementSelectionEvent;
import org.eclipse.recommenders.rcp.JavaElementSelectionEvent.JavaElementSelectionLocation;

public class JavaSelectionTestUtils {

    public static final JavaElementSelectionEvent TYPE_IN_TYPE_DECLARATION = mockJavaSelection(IType.class,
            TYPE_DECLARATION);
    public static final JavaElementSelectionEvent TYPE_IN_TYPE_DECLARATION_EXTENDS = mockJavaSelection(IType.class,
            TYPE_DECLARATION_EXTENDS);
    public static final JavaElementSelectionEvent TYPE_IN_TYPE_DECLARATION_IMPLEMENTS = mockJavaSelection(IType.class,
            TYPE_DECLARATION_IMPLEMENTS);
    public static final JavaElementSelectionEvent TYPE_IN_METHOD_BODY = mockJavaSelection(IType.class, METHOD_BODY);
    public static final JavaElementSelectionEvent TYPE_IN_METHOD_DECLARATION_THROWS = mockJavaSelection(IType.class,
            METHOD_DECLARATION_THROWS);
    public static final JavaElementSelectionEvent TYPE_IN_METHOD_DECLARATION_PARAMS = mockJavaSelection(IType.class,
            JavaElementSelectionLocation.METHOD_DECLARATION_PARAMETER);
    public static final JavaElementSelectionEvent ANNOTATION_IN_METHOD_DECLARATION = mockJavaSelection(
            IAnnotation.class, METHOD_DECLARATION);
    public static final JavaElementSelectionEvent METHOD_IN_METHOD_DECLARATION = mockJavaSelection(IMethod.class,
            METHOD_DECLARATION);
    public static final JavaElementSelectionEvent METHOD_IN_METHOD_BODY = mockJavaSelection(IMethod.class, METHOD_BODY);

    public static JavaElementSelectionEvent mockJavaSelection(Class<? extends IJavaElement> clazz,
            JavaElementSelectionLocation location) {
        IJavaElement element = mock(clazz);
        JavaElementSelectionEvent selection = mock(JavaElementSelectionEvent.class);
        when(selection.getElement()).thenReturn(element);
        when(selection.getLocation()).thenReturn(location);
        return selection;
    }

}