tests/org.eclipse.dltk.javascript.core.tests/src/org/eclipse/dltk/javascript/core/tests/typeinfo/RTypeMemberQueryTest.java

/*******************************************************************************
 * Copyright (c) 2013 NumberFour AG
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 * 
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *     NumberFour AG - initial API and Implementation (Alex Panchenko)
 *******************************************************************************/
package org.eclipse.dltk.javascript.core.tests.typeinfo;

import org.eclipse.dltk.internal.javascript.ti.TypeSystemImpl;
import org.eclipse.dltk.javascript.typeinfo.IRMethod;
import org.eclipse.dltk.javascript.typeinfo.IRTypeDeclaration;
import org.eclipse.dltk.javascript.typeinfo.ITypeSystem;
import org.eclipse.dltk.javascript.typeinfo.MemberPredicates;
import org.eclipse.dltk.javascript.typeinfo.RTypeMemberQuery;
import org.eclipse.dltk.javascript.typeinfo.model.Method;
import org.eclipse.dltk.javascript.typeinfo.model.Type;
import org.eclipse.dltk.javascript.typeinfo.model.TypeInfoModelFactory;
import org.junit.Assert;
import org.junit.Test;

@SuppressWarnings("restriction")
public class RTypeMemberQueryTest extends Assert {

	private static final TypeInfoModelFactory TIMF = TypeInfoModelFactory.eINSTANCE;

	private final ITypeSystem typeSystem = new TypeSystemImpl();

	@Test
	public void findSuperMethod() {
		final Type base = TIMF.createType();
		base.setName("Base");
		final Method baseRun = TIMF.createMethod();
		baseRun.setAbstract(true);
		baseRun.setName("run");
		base.getMembers().add(baseRun);

		final Type child = TIMF.createType();
		child.setSuperType(base);
		child.setName("Child");
		final Method childRun = TIMF.createMethod();
		childRun.setAbstract(true);
		childRun.setName("run");
		child.getMembers().add(childRun);

		final IRTypeDeclaration declaration = typeSystem.convert(child);
		final RTypeMemberQuery memberQuery = new RTypeMemberQuery(declaration,
				MemberPredicates.NON_STATIC);
		final IRMethod method = memberQuery.findMethod("run");
		assertNotNull(method);
		assertSame(childRun, method.getSource());

		final IRMethod superMethod = memberQuery.findSuperMethod("run");
		assertNotNull(superMethod);
		assertSame(baseRun, superMethod.getSource());
	}

}