edu.illinois.compositerefactorings.steps.tests.StepTest.java Source code

Java tutorial

Introduction

Here is the source code for edu.illinois.compositerefactorings.steps.tests.StepTest.java

Source

/**
 * 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"));
    }

}