/** * Copyright (c) 2008-2012 University of Illinois at Urbana-Champaign. * 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 */ package edu.illinois.compositerefactorings.steps.tests; import java.text.MessageFormat; import java.util.Hashtable; import java.util.List; import org.eclipse.core.runtime.Preferences; import org.eclipse.jdt.core.ICompilationUnit; import org.eclipse.jdt.core.IJavaProject; import org.eclipse.jdt.core.IPackageFragment; import org.eclipse.jdt.core.IPackageFragmentRoot; import org.eclipse.jdt.core.JavaCore; import org.eclipse.jdt.core.formatter.DefaultCodeFormatterConstants; import org.eclipse.jdt.internal.corext.codemanipulation.StubUtility; import org.eclipse.jdt.internal.corext.template.java.CodeTemplateContextType; import org.eclipse.jdt.internal.ui.JavaPlugin; import org.eclipse.jdt.internal.ui.text.correction.AssistContext; import org.eclipse.jdt.testplugin.JavaProjectHelper; import org.eclipse.jdt.testplugin.TestOptions; import org.eclipse.jdt.ui.PreferenceConstants; import org.eclipse.jdt.ui.tests.core.ProjectTestSetup; import org.eclipse.jface.preference.IPreferenceStore; import org.junit.After; import org.junit.Before; import org.junit.BeforeClass; import org.junit.Test; import edu.illinois.compositerefactorings.messages.CompositeRefactoringsMessages; @SuppressWarnings({ "restriction", "deprecation" }) public class StepTest { private IJavaProject fJProject1; private IPackageFragmentRoot fSourceFolder; @BeforeClass public static void setUpTests() throws Exception { new ProjectTestSetupUtilities().setUp(); } @Before public void setUp() throws Exception { @SuppressWarnings("unchecked") Hashtable<String, String> options= TestOptions.getDefaultOptions(); options.put(DefaultCodeFormatterConstants.FORMATTER_TAB_CHAR, JavaCore.SPACE); options.put(DefaultCodeFormatterConstants.FORMATTER_TAB_SIZE, "4"); JavaCore.setOptions(options); IPreferenceStore store= JavaPlugin.getDefault().getPreferenceStore(); store.setValue(PreferenceConstants.CODEGEN_ADD_COMMENTS, false); store.setValue(PreferenceConstants.CODEGEN_KEYWORD_THIS, false); StubUtility.setCodeTemplate(CodeTemplateContextType.METHODSTUB_ID, "//TODO\n${body_statement}", null); Preferences corePrefs= JavaPlugin.getJavaCorePluginPreferences(); corePrefs.setValue(JavaCore.CODEASSIST_FIELD_PREFIXES, ""); corePrefs.setValue(JavaCore.CODEASSIST_STATIC_FIELD_PREFIXES, ""); corePrefs.setValue(JavaCore.CODEASSIST_FIELD_SUFFIXES, ""); corePrefs.setValue(JavaCore.CODEASSIST_STATIC_FIELD_SUFFIXES, ""); fJProject1= ProjectTestSetup.getProject(); fSourceFolder= JavaProjectHelper.addSourceContainer(fJProject1, "src"); } @After public void tearDown() throws Exception { JavaProjectHelper.clear(fJProject1, ProjectTestSetup.getDefaultClasspath()); } //@Test public void testCreateNewSuperclass1() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("package test1;\n"); sb.append("public class C {\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("}\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), false, null); int offset= sb.toString().indexOf("C"); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); sb= new StringBuilder(); sb.append("package test1;\n"); sb.append("public class C extends SuperC {\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("}\n"); sb.append("\n"); sb.append("class SuperC {\n"); sb.append("}\n"); String expected1= sb.toString(); StepTestUtilities.assertExpectedExistInProposals(proposals, new String[] { expected1 }); } @Test public void testCreateNewTopLevelSuperclass1() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("package test1;\n"); sb.append("public class C {\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("}\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), false, null); int offset= sb.toString().indexOf("C"); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.CreateNewTopLevelSuperClass_description, "C")); } @Test public void testMoveToImmediateSuperclass1() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("package test1;\n"); sb.append("public class C extends D {\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("}\n"); sb.append("\n"); sb.append("class D {\n"); sb.append("}\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), false, null); int offset= sb.toString().indexOf("m()"); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.MoveToImmediateSuperclass_description, "m", "D")); } @Test public void testMoveTypeToNewFile1() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("package test1;\n"); sb.append("public class C extends D {\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("}\n"); sb.append("\n"); sb.append("class D {\n"); sb.append("}\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), false, null); int offset= sb.toString().indexOf("class D") + "class ".length(); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.MoveTypeToNewFile_description, "D")); } //@Test public void testReplaceTypeBySupertype1() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("public class C {\n"); sb.append(" public void m(E o) {\n"); sb.append(" if (o instanceof E) {\n"); sb.append(" o.m();\n"); sb.append(" }\n"); sb.append(" }\n"); sb.append("}\n"); sb.append("\n"); sb.append("class D {\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("}\n"); sb.append("\n"); sb.append("class E extends D {\n"); sb.append("}\n"); sb.append("\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), false, null); int offset= sb.toString().indexOf("class E") + "class ".length(); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInVariableDeclarations_description, "E", "D")); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInInstanceOf_description, "E", "D")); } //@Test public void testReplaceTypeBySupertype2() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("public class C {\n"); sb.append(" public void m(F o) {\n"); sb.append(" if (o instanceof F) {\n"); sb.append(" o.m();\n"); sb.append(" }\n"); sb.append(" }\n"); sb.append("}\n"); sb.append("\n"); sb.append("class D {\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("}\n"); sb.append("\n"); sb.append("class E extends D {\n"); sb.append("}\n"); sb.append("\n"); sb.append("class F extends E {\n"); sb.append("\n"); sb.append("}\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), false, null); int offset= sb.toString().indexOf("class F") + "class ".length(); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInVariableDeclarations_description, "F", "E")); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInVariableDeclarations_description, "F", "D")); StepTestUtilities.assertProposalDoesNotExist(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInVariableDeclarations_description, "F", "Object")); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInInstanceOf_description, "F", "E")); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInInstanceOf_description, "F", "D")); StepTestUtilities.assertProposalDoesNotExist(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInInstanceOf_description, "F", "Object")); } //@Test public void testReplaceTypeBySupertype3() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("interface I {\n"); sb.append("\n"); sb.append(" void m();\n"); sb.append("\n"); sb.append("}\n"); sb.append("\n"); sb.append("class C implements I {\n"); sb.append("\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("\n"); sb.append(" void test() {\n"); sb.append(" C o = new C();\n"); sb.append(" if (o instanceof C) {\n"); sb.append(" ((C) o).m();\n"); sb.append(" }\n"); sb.append(" }\n"); sb.append(" \n"); sb.append("}\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), false, null); int offset= sb.toString().indexOf("class C") + "class ".length(); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInVariableDeclarations_description, "C", "I")); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInVariableDeclarations_description, "C", "Object")); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInInstanceOf_description, "C", "I")); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.ReplaceTypeBySupertypeInInstanceOf_description, "C", "Object")); } @Test public void testAddMethodParameterForExpression1() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("public class C {\n"); sb.append(" \n"); sb.append(" int m1() {\n"); sb.append(" return m2();\n"); sb.append(" }\n"); sb.append(" \n"); sb.append(" int m2() {\n"); sb.append(" return 1 + 2;\n"); sb.append(" }\n"); sb.append("\n"); sb.append("}\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), false, null); int offset= sb.toString().indexOf("return 1") + "return ".length(); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.AddMethodParameterForExpression_description, "m2")); } @Test public void testCreateAndImplementNewInterface() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("package test1;\n"); sb.append("public class C {\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("}\n"); sb.append("\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), false, null); int offset= sb.toString().indexOf("C {"); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.CreateNewTopLevelInterface_description, "C")); } @Test public void testCopyMemberToSubtype() throws Exception { IPackageFragment pack1= fSourceFolder.createPackageFragment("test1", false, null); StringBuilder sb= new StringBuilder(); sb.append("package test1;\n"); sb.append("class C {\n"); sb.append(" public void m() {\n"); sb.append(" }\n"); sb.append("}\n"); sb.append("\n"); sb.append("class D extends C {\n"); sb.append("}\n"); ICompilationUnit cu= pack1.createCompilationUnit("C.java", sb.toString(), true, null); int offset= sb.toString().indexOf("m()"); AssistContext context= StepTestUtilities.getCorrectionContext(cu, offset, 0); List<?> proposals= StepTestUtilities.doCollectAssists(context, false); StepTestUtilities.assertCorrectLabels(proposals); StepTestUtilities.assertProposalExists(proposals, MessageFormat.format(CompositeRefactoringsMessages.CopyMemberToSubstype_description, "m", "D")); } }