Added simple IDE templates test which checks that folding worked normally.

2012-02-14 14:21:17 +04:00
parent e3fdc5d595
commit 7ce86a138e
2 changed files with 105 additions and 0 deletions
@@ -0,0 +1,42 @@
 fun main(args : Array<String>) {
    if (<#<condition>#>) {
        <#<block>#>
    } else {
        <#<block>#>
    }
    fun <#<name>#>(<#<params>#>) : <#<returnType>#> {
        <#<body>#>
    }
    for (<#<i>#> in <#<elements>#>) {
        <#<body>#>
    }
    when (<#<expression>#>) {
        <#<condition>#> -> <#<value>#>
        else -> <#<elseValue>#>
    }
    var <#<name>#> = <#<value>#>
    class <#<name>#> {
        <#<body>#>
    }
    class <#<name>#> {
        var <#<name>#> : <#<varType>#>
        get() {
            <#<body>#>
        }
        set(value) {
            <#<body>#>
        }
        val <#<name>#> : <#<valType>#>
        get() {
            <#<body>#>
        }
    }
 }
@@ -0,0 +1,63 @@
 package org.jetbrains.jet.plugin.codeInsight;
 import com.intellij.codeInsight.folding.CodeFoldingManager;
 import com.intellij.openapi.editor.FoldRegion;
 import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase;
 import org.jetbrains.jet.plugin.PluginTestCaseBase;
 import java.util.ArrayList;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 /**
 * @author Evgeny Gerashchenko
 * @since 2/14/12
 */
 public class IdeTemplatesTest extends LightCodeInsightFixtureTestCase {
    public void testAll() {
        myFixture.configureByFile(PluginTestCaseBase.getTestDataPathBase() + "/templates/IdeTemplates.kt");
        CodeFoldingManager.getInstance(myFixture.getProject()).buildInitialFoldings(myFixture.getEditor());
        ArrayList<Region> regions = getExpectedRegions();
        assertEquals(regions.toString(), getActualRegions().toString());
    }
    private ArrayList<Region> getExpectedRegions() {
        Pattern regex = Pattern.compile("<#<(\\w+)>#>");
        Matcher matcher = regex.matcher(myFixture.getEditor().getDocument().getText());
        ArrayList<Region> expected = new ArrayList<Region>();
        while (matcher.find()) {
            expected.add(new Region(matcher.start(), matcher.end(), matcher.group(1)));
        }
        return expected;
    }
    private ArrayList<Region> getActualRegions() {
        ArrayList<Region> actual = new ArrayList<Region>();
        for (FoldRegion fr : myFixture.getEditor().getFoldingModel().getAllFoldRegions()) {
            if (fr.shouldNeverExpand()) {
                assertFalse(fr.isExpanded());
                actual.add(new Region(fr.getStartOffset(), fr.getEndOffset(), fr.getPlaceholderText()));
            }
        }
        return actual;
    }
    private static class Region {
        public int start;
        public int end;
        public String group;
        private Region(int start, int end, String group) {
            this.start = start;
            this.end = end;
            this.group = group;
        }
        @Override
        public String toString() {
            return String.format("%d..%d:%s", start, end, group);
        }
    }
 }