JetIntroduceVariableTest.java

/*
 * Copyright 2010-2013 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.jetbrains.jet.plugin.refactoring.introduceVariable;

import com.intellij.ide.DataManager;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.psi.PsiElement;
import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase;
import org.jetbrains.jet.lang.psi.JetFile;
import org.jetbrains.jet.lexer.JetTokens;
import org.jetbrains.jet.plugin.PluginTestCaseBase;

/**
 * User: Alefas
 * Date: 31.01.12
 */
public class JetIntroduceVariableTest extends LightCodeInsightFixtureTestCase {
    public void testClassBody() {
        doTest();
    }

    public void testDoWhileAddBlock() {
        doTest();
    }

    public void testDoWhileAddBlockInner() {
        doTest();
    }

    public void testFewOccurrences() {
        doTest();
    }

    public void testFunctionAddBlock() {
        doTest();
    }

    public void testFunctionAddBlockInner() {
        doTest();
    }

    public void testFunctionLiteral() {
        doTest();
    }

    public void testFunctionLiteralFromExpected() {
        doTest();
    }

    public void testIfCondition() {
        doTest();
    }

    public void testIfElseAddBlock() {
        doTest();
    }

    public void testIfElseAddBlockInner() {
        doTest();
    }

    public void testIfThenAddBlock() {
        doTest();
    }

    public void testIfThenAddBlockInner() {
        doTest();
    }

    public void testIt() {
        doTest();
    }

    public void testLoopRange() {
        doTest();
    }

    public void testManyInnerOccurences() {
        doTest();
    }

    public void testManyOccurrences() {
        doTest();
    }

    public void testReplaceOccurence() {
        doTest();
    }

    public void testSimple() {
        doTest();
    }

    public void testSimpleCreateValue() {
        doTest();
    }

    public void testStringInjection() {
        doTest();
    }

    public void testWhenAddBlock() {
        doTest();
    }

    public void testWhenAddBlockInner() {
        doTest();
    }

    public void testWhenParts() {
        doTest();
    }

    public void testWhileAddBlock() {
        doTest();
    }

    public void testWhileAddBlockInner() {
        doTest();
    }

    public void testWhileCondition() {
        doTest();
    }

    public void testArrayAccessExpr() {
        doTest();
    }

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        myFixture.setTestDataPath(PluginTestCaseBase.getTestDataPathBase() + "/refactoring/introduceVariable");
    }

    private void doTest() {
        myFixture.configureByFile(getTestName(false) + ".kt");
        final JetFile file = (JetFile) myFixture.getFile();
        PsiElement lastChild = file.getLastChild();
        assert lastChild != null;
        String expectedResultText = null;
        if (lastChild.getNode().getElementType().equals(JetTokens.BLOCK_COMMENT)) {
            String lastChildText = lastChild.getText();
            expectedResultText = lastChildText.substring(2, lastChildText.length() - 2).trim();
        }
        else if (lastChild.getNode().getElementType().equals(JetTokens.EOL_COMMENT)) {
            expectedResultText = lastChild.getText().substring(2).trim();
        }
        assert expectedResultText != null;
        ApplicationManager.getApplication().runWriteAction(new Runnable() {
            @Override
            public void run() {
                new JetIntroduceVariableHandler().invoke(getProject(), myFixture.getEditor(), file,
                                                         DataManager.getInstance().getDataContext(myFixture.getEditor().
                                                                 getComponent()));
            }
        });
        int endOffset = file.getLastChild().getTextRange().getStartOffset();
//        System.out.println(file.getText().substring(0, endOffset).trim());
        assertEquals(expectedResultText, file.getText().substring(0, endOffset).trim());
    }
}