platform/platform-tests/testSrc/com/intellij/openapi/editor/impl/FoldingAnchorsOverlayStrategyTest.java - platform/tools/idea - Git at Google

 /*
  * Copyright 2000-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 com.intellij.openapi.editor.impl;

 import com.intellij.codeInsight.folding.CodeFoldingManager;
 import com.intellij.openapi.editor.FoldRegion;
 import com.intellij.openapi.editor.FoldingModel;
 import com.intellij.openapi.fileTypes.FileTypes;
 import com.intellij.openapi.fileTypes.StdFileTypes;
 import com.intellij.testFramework.fixtures.LightPlatformCodeInsightFixtureTestCase;

 import java.util.*;

 import static com.intellij.openapi.editor.impl.DisplayedFoldingAnchor.Type;

 public class FoldingAnchorsOverlayStrategyTest extends LightPlatformCodeInsightFixtureTestCase {
   public void testExpanded() {
     prepareEditor("<body><div>\n" +
                   "</div><div>\n" +
                   "some text\n" +
                   "some text\n" +
                   "some text\n" +
                   "</div><div>\n" +
                   "</div></body>");
     verifyAnchors(null,
                   0, Type.EXPANDED_TOP,
                   1, Type.EXPANDED_BOTTOM,
                   5, Type.EXPANDED_TOP,
                   6, Type.EXPANDED_BOTTOM);
   }

   public void testCollapsed() {
     prepareEditor("<body><div>\n" +
                   "</div><div>\n" +
                   "some text\n" +
                   "some text\n" +
                   "some text\n" +
                   "</div><div>\n" +
                   "</div></body>");
     collapseFoldingRegion(2);
     verifyAnchors(null,
                   0, Type.EXPANDED_TOP,
                   1, Type.COLLAPSED,
                   2, Type.EXPANDED_BOTTOM);
   }

   public void testWithActiveRegion() {
     prepareEditor("<body><div>\n" +
                   "</div><div>\n" +
                   "some text\n" +
                   "some text\n" +
                   "some text\n" +
                   "</div><div>\n" +
                   "</div></body>");
     collapseFoldingRegion(2);
     verifyAnchors(myFixture.getEditor().getFoldingModel().getAllFoldRegions()[1],
                   0, Type.EXPANDED_TOP,
                   1, Type.EXPANDED_BOTTOM,
                   2, Type.EXPANDED_BOTTOM);
   }

   public void testWithEmptyLastLine() {
     myFixture.configureByText(FileTypes.PLAIN_TEXT, "some text\n");
     final FoldingModel foldingModel = myFixture.getEditor().getFoldingModel();
     foldingModel.runBatchFoldingOperation(new Runnable() {
       @Override
       public void run() {
         foldingModel.addFoldRegion(0, 10, "...");
       }
     });
     verifyAnchors(null,
                   0, Type.EXPANDED_TOP,
                   1, Type.EXPANDED_BOTTOM);
   }

   private void prepareEditor(String text) {
     myFixture.configureByText(StdFileTypes.XML, text);
     CodeFoldingManager.getInstance(myFixture.getProject()).updateFoldRegions(myFixture.getEditor());
   }

   private void collapseFoldingRegion(int n) {
     FoldingModel foldingModel = myFixture.getEditor().getFoldingModel();
     final FoldRegion foldRegion = foldingModel.getAllFoldRegions()[n];
     foldingModel.runBatchFoldingOperation(new Runnable() {
       @Override
       public void run() {
         foldRegion.setExpanded(false);
       }
     });
   }

   private void verifyAnchors(FoldRegion activeFoldRegion, Object... expectedAnchorParameters) {
     Collection<DisplayedFoldingAnchor> actualAnchors = new FoldingAnchorsOverlayStrategy((EditorImpl)myFixture.getEditor())
       .getAnchorsToDisplay(0, myFixture.getEditor().getDocument().getTextLength(), activeFoldRegion);
     List<DisplayedFoldingAnchor> sortedActualAnchors = new ArrayList<DisplayedFoldingAnchor>(actualAnchors);
     Collections.sort(sortedActualAnchors, new Comparator<DisplayedFoldingAnchor>() {
       @Override
       public int compare(DisplayedFoldingAnchor o1, DisplayedFoldingAnchor o2) {
         return o1.visualLine - o2.visualLine;
       }
     });

     assertEquals("Wrong number of anchors", expectedAnchorParameters.length / 2, sortedActualAnchors.size());
     int i = 0;
     for (DisplayedFoldingAnchor anchor :  sortedActualAnchors) {
       int expectedVisualLine = (Integer) expectedAnchorParameters[i++];
       assertEquals("Folding anchor at wrong line found", expectedVisualLine, anchor.visualLine);
       Type expectedType = (Type)expectedAnchorParameters[i++];
       assertEquals("Folding anchor of wrong type found at line " + expectedVisualLine, expectedType, anchor.type);
     }
   }
 }
	/*
	* Copyright 2000-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 com.intellij.openapi.editor.impl;

	import com.intellij.codeInsight.folding.CodeFoldingManager;
	import com.intellij.openapi.editor.FoldRegion;
	import com.intellij.openapi.editor.FoldingModel;
	import com.intellij.openapi.fileTypes.FileTypes;
	import com.intellij.openapi.fileTypes.StdFileTypes;
	import com.intellij.testFramework.fixtures.LightPlatformCodeInsightFixtureTestCase;

	import java.util.*;

	import static com.intellij.openapi.editor.impl.DisplayedFoldingAnchor.Type;

	public class FoldingAnchorsOverlayStrategyTest extends LightPlatformCodeInsightFixtureTestCase {
	public void testExpanded() {
	prepareEditor("<body><div>\n" +
	"</div><div>\n" +
	"some text\n" +
	"some text\n" +
	"some text\n" +
	"</div><div>\n" +
	"</div></body>");
	verifyAnchors(null,
	0, Type.EXPANDED_TOP,
	1, Type.EXPANDED_BOTTOM,
	5, Type.EXPANDED_TOP,
	6, Type.EXPANDED_BOTTOM);
	}

	public void testCollapsed() {
	prepareEditor("<body><div>\n" +
	"</div><div>\n" +
	"some text\n" +
	"some text\n" +
	"some text\n" +
	"</div><div>\n" +
	"</div></body>");
	collapseFoldingRegion(2);
	verifyAnchors(null,
	0, Type.EXPANDED_TOP,
	1, Type.COLLAPSED,
	2, Type.EXPANDED_BOTTOM);
	}

	public void testWithActiveRegion() {
	prepareEditor("<body><div>\n" +
	"</div><div>\n" +
	"some text\n" +
	"some text\n" +
	"some text\n" +
	"</div><div>\n" +
	"</div></body>");
	collapseFoldingRegion(2);
	verifyAnchors(myFixture.getEditor().getFoldingModel().getAllFoldRegions()[1],
	0, Type.EXPANDED_TOP,
	1, Type.EXPANDED_BOTTOM,
	2, Type.EXPANDED_BOTTOM);
	}

	public void testWithEmptyLastLine() {
	myFixture.configureByText(FileTypes.PLAIN_TEXT, "some text\n");
	final FoldingModel foldingModel = myFixture.getEditor().getFoldingModel();
	foldingModel.runBatchFoldingOperation(new Runnable() {
	@Override
	public void run() {
	foldingModel.addFoldRegion(0, 10, "...");
	}
	});
	verifyAnchors(null,
	0, Type.EXPANDED_TOP,
	1, Type.EXPANDED_BOTTOM);
	}

	private void prepareEditor(String text) {
	myFixture.configureByText(StdFileTypes.XML, text);
	CodeFoldingManager.getInstance(myFixture.getProject()).updateFoldRegions(myFixture.getEditor());
	}

	private void collapseFoldingRegion(int n) {
	FoldingModel foldingModel = myFixture.getEditor().getFoldingModel();
	final FoldRegion foldRegion = foldingModel.getAllFoldRegions()[n];
	foldingModel.runBatchFoldingOperation(new Runnable() {
	@Override
	public void run() {
	foldRegion.setExpanded(false);
	}
	});
	}

	private void verifyAnchors(FoldRegion activeFoldRegion, Object... expectedAnchorParameters) {
	Collection<DisplayedFoldingAnchor> actualAnchors = new FoldingAnchorsOverlayStrategy((EditorImpl)myFixture.getEditor())
	.getAnchorsToDisplay(0, myFixture.getEditor().getDocument().getTextLength(), activeFoldRegion);
	List<DisplayedFoldingAnchor> sortedActualAnchors = new ArrayList<DisplayedFoldingAnchor>(actualAnchors);
	Collections.sort(sortedActualAnchors, new Comparator<DisplayedFoldingAnchor>() {
	@Override
	public int compare(DisplayedFoldingAnchor o1, DisplayedFoldingAnchor o2) {
	return o1.visualLine - o2.visualLine;
	}
	});

	assertEquals("Wrong number of anchors", expectedAnchorParameters.length / 2, sortedActualAnchors.size());
	int i = 0;
	for (DisplayedFoldingAnchor anchor : sortedActualAnchors) {
	int expectedVisualLine = (Integer) expectedAnchorParameters[i++];
	assertEquals("Folding anchor at wrong line found", expectedVisualLine, anchor.visualLine);
	Type expectedType = (Type)expectedAnchorParameters[i++];
	assertEquals("Folding anchor of wrong type found at line " + expectedVisualLine, expectedType, anchor.type);
	}
	}
	}