LegalStatementsChecker: refactoring

[ifi-stolz-refaktor.git] / software / no.uio.ifi.refaktor.tests / src / no / uio / ifi / refaktor / tests / LegalStatementsTest.java

package no.uio.ifi.refaktor.tests;

import static no.uio.ifi.refaktor.assertion.RefaktorAssert.assertThat;
import static org.hamcrest.CoreMatchers.is;
import no.uio.ifi.refaktor.analyze.analyzers.LegalStatementsChecker;
import no.uio.ifi.refaktor.examples.manager.ExampleCodeManager;
import no.uio.ifi.refaktor.utils.CompilationUnitTextSelection;
import no.uio.ifi.refaktor.utils.RefaktorHandleUtils;

import org.eclipse.jdt.core.IMethod;
import org.eclipse.jdt.core.JavaModelException;
import org.junit.BeforeClass;
import org.junit.Test;

public class LegalStatementsTest {
	// NOTE: if any tests shall alter any code, the example code
	// must be loaded in setUp method
	
	private static final boolean NOT_LEGAL = false;

	@BeforeClass
	public static void setUp() throws Exception {
		ExampleCodeManager.INSTANCE.loadExampleCode();
	}

	@Test
	public void testCheckCallToSuperConstructor() throws Exception {
		testIfSelectionWithinMethodIsLegal("Main", 18, 8, NOT_LEGAL);
	}

	@Test
	public void testCheckCallToSuperMethod() throws Exception {
		testIfSelectionWithinMethodIsLegal("superMethod", 65, 20, NOT_LEGAL);
	}

	@Test
	public void testCheckAccessToSuperField() throws Exception {
		testIfSelectionWithinMethodIsLegal("iAccessSuperField", 58, 37, NOT_LEGAL);
	}

	private void testIfSelectionWithinMethodIsLegal(String methodName, int methodRelativeOffset, int length, boolean expectedLegalityOfSelection) throws JavaModelException {
		IMethod enclosingMethod = getTestDataMethod(methodName);
		assertThat(enclosingMethod.exists());
		int methodOffset = enclosingMethod.getSourceRange().getOffset();

		CompilationUnitTextSelection selection = new CompilationUnitTextSelection(
				enclosingMethod.getCompilationUnit(), methodOffset + methodRelativeOffset , length);
		LegalStatementsChecker legalStatementsChecker = new LegalStatementsChecker(selection);
		assertThat(legalStatementsChecker.allStatementsAreLegal(), is(expectedLegalityOfSelection));
	}

	private IMethod getTestDataMethod(String methodName, String... parameters) {
		return RefaktorHandleUtils.findMethodHandleChecked(ExampleCodeManager.INSTANCE.getProject(), "legal", "Main", methodName, parameters);
	}

}