platform/platform-tests/testSrc/com/intellij/openapi/editor/actions/SelectUnselectOccurrenceActionsTest.java

/*
 * Copyright 2000-2014 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.actions;

import com.intellij.codeInsight.hint.EditorHintListener;
import com.intellij.openapi.actionSystem.IdeActions;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.editor.Document;
import com.intellij.openapi.editor.FoldRegion;
import com.intellij.openapi.editor.FoldingModel;
import com.intellij.openapi.fileTypes.FileTypes;
import com.intellij.openapi.project.Project;
import com.intellij.testFramework.fixtures.LightPlatformCodeInsightFixtureTestCase;
import com.intellij.ui.LightweightHint;

public class SelectUnselectOccurrenceActionsTest extends LightPlatformCodeInsightFixtureTestCase {
  private int hintCount;

  @Override
  public void setUp() throws Exception {
    super.setUp();
    EditorHintListener listener = new EditorHintListener() {
      @Override
      public void hintShown(Project project, LightweightHint hint, int flags) {
        hintCount++;
      }
    };
    ApplicationManager.getApplication().getMessageBus().connect(myTestRootDisposable).subscribe(EditorHintListener.TOPIC, listener);
  }

  public void testAllWithoutInitialSelection() throws Exception {
    init("some t<caret>ext\n" +
         "some texts\n" +
         "another text here"
    );
    executeSelectAllAction();
    checkResult("some <selection>t<caret>ext</selection>\n" +
                "some texts\n" +
                "another <selection>t<caret>ext</selection> here");
  }

  public void testAllWithInitialWholeWordSelection() throws Exception {
    init("some <selection>t<caret>ext</selection>\n" +
         "some texts\n" +
         "some texts\n" +
         "another text here");
    executeSelectAllAction();
    checkResult("some <selection>t<caret>ext</selection>\n" +
                "some <selection>t<caret>ext</selection>s\n" +
                "some <selection>t<caret>ext</selection>s\n" +
                "another <selection>t<caret>ext</selection> here");
    assertEquals(0, hintCount);
  }

  public void testNoInitialSelection() throws Exception {
    init("some t<caret>ext\n" +
         "some texts\n" +
         "another text here"
    );
    executeAction();
    checkResult("some <selection>t<caret>ext</selection>\n" +
                "some texts\n" +
                "another text here");
    executeAction();
    checkResult("some <selection>t<caret>ext</selection>\n" +
                "some texts\n" +
                "another <selection>t<caret>ext</selection> here");
    assertEquals(0, hintCount);
  }

  public void testInitialWholeWordSelection() throws Exception {
    init("some <selection>t<caret>ext</selection>\n" +
         "some texts\n" +
         "another text here");
    executeAction();
    checkResult("some <selection>t<caret>ext</selection>\n" +
                "some <selection>t<caret>ext</selection>s\n" +
                "another text here");
    assertEquals(0, hintCount);
  }

  public void testShowingHint() throws Exception {
    init("some <selection>t<caret>ext</selection>\n" +
         "another <selection>t<caret>ext</selection> here");
    executeAction();
    assertEquals(1, hintCount);
    checkResult("some <selection>t<caret>ext</selection>\n" +
                "another <selection>t<caret>ext</selection> here");
    executeAction();
    assertEquals(1, hintCount);
    checkResult("some <selection>t<caret>ext</selection>\n" +
                "another <selection>t<caret>ext</selection> here");
  }

  public void testRevert() throws Exception {
    init("some <selection>t<caret>ext</selection>\n" +
         "another <selection>t<caret>ext</selection> here");
    executeReverseAction();
    checkResult("some <selection>t<caret>ext</selection>\n" +
                "another text here");
    assertEquals(0, hintCount);
  }

  public void testRevertSingleSelection() throws Exception {
    init("some <selection>t<caret>ext</selection>\n" +
         "some texts\n" +
         "another text here");
    executeReverseAction();
    checkResult("some t<caret>ext\n" +
                "some texts\n" +
                "another text here");
    assertEquals(0, hintCount);
  }

  public void testSelectAfterHint() throws Exception {
    init("some text\n" +
         "some texts\n" +
         "another <selection>t<caret>ext</selection> here");
    executeAction();
    checkResult("some text\n" +
                "some texts\n" +
                "another <selection>t<caret>ext</selection> here");
    assertEquals(1, hintCount);
    executeAction();
    checkResult("some <selection>t<caret>ext</selection>\n" +
                "some texts\n" +
                "another <selection>t<caret>ext</selection> here");
    assertEquals(1, hintCount);
  }

  public void testInitialNonWholeWordSelection() throws Exception {
    init("some <selection>t<caret>ex</selection>t\n" +
         "some texts\n" +
         "another text here");
    executeAction();
    checkResult("some <selection>t<caret>ex</selection>t\n" +
                "some <selection>t<caret>ex</selection>ts\n" +
                "another text here");
    executeAction();
    checkResult("some <selection>t<caret>ex</selection>t\n" +
                "some <selection>t<caret>ex</selection>ts\n" +
                "another <selection>t<caret>ex</selection>t here");
    assertEquals(0, hintCount);
  }

  public void testOccurrenceInCollapsedRegion() throws Exception {
    init("normal <selection><caret>line</selection>\n" +
         "collapsed line");
    final FoldingModel foldingModel = myFixture.getEditor().getFoldingModel();
    final Document document = myFixture.getEditor().getDocument();
    foldingModel.runBatchFoldingOperation(new Runnable() {
      @Override
      public void run() {
        FoldRegion foldRegion = foldingModel.addFoldRegion(document.getLineStartOffset(1), document.getLineEndOffset(1), "...");
        assertNotNull(foldRegion);
        foldRegion.setExpanded(false);
      }
    });
    executeAction();
    checkResult("normal <selection><caret>line</selection>\n" +
                "collapsed <selection><caret>line</selection>");
    FoldRegion[] foldRegions = foldingModel.getAllFoldRegions();
    assertEquals(1, foldRegions.length);
    assertTrue(foldRegions[0].isExpanded());
  }

  public void testSelectAfterNotFoundAndUnselect() throws Exception {
    init("text <selection><caret>text</selection> <selection><caret>text</selection>");
    executeAction();
    executeReverseAction();
    executeAction();
    checkResult("text <selection><caret>text</selection> <selection><caret>text</selection>");
  }

  public void testEscapeReturnsToInitialPosition() throws Exception {
    init("l<caret>ine\n" +
         "another line");
    executeAction();
    myFixture.performEditorAction("EditorEscape");
    checkResult("l<caret>ine\n" +
                "another line");
  }

  public void testSelectingAdjacentFragments() throws Exception {
    init("fragment<selection>fragment<caret></selection>");
    executeAction();
    executeAction();
    checkResult("<selection>fragment<caret></selection><selection>fragment<caret></selection>");
  }

  public void testSkippingOccurrence() throws Exception {
    init("fr<caret>uit\n" +
         "fruits\n" +
         "fruit\n" +
         "fruits\n" +
         "fruit");
    executeAction();
    executeAction();
    executeFindNext();
    checkResult("<selection>fr<caret>uit</selection>\n" +
                "fruits\n" +
                "fruit\n" +
                "fruits\n" +
                "<selection>fr<caret>uit</selection>");
  }

  public void testMovingSelectionBackAndForth() throws Exception {
    init("fr<caret>uit\n" +
         "fruits\n" +
         "fruit\n" +
         "fruits\n" +
         "fruit");
    executeAction();
    executeAction();
    executeFindNext();
    executeFindPrevious();
    executeAction();
    checkResult("<selection>fr<caret>uit</selection>\n" +
                "fruits\n" +
                "<selection>fr<caret>uit</selection>\n" +
                "fruits\n" +
                "<selection>fr<caret>uit</selection>");
  }

  public void testSkipDoesNotRemovePreviousSelections() throws Exception {
    init("<caret>fruit\n" +
         "fruit\n" +
         "fruit");
    executeAction();
    executeAction();
    executeFindNext();
    executeFindNext();
    assertEquals(1, hintCount);
    executeFindNext();
    assertEquals(1, hintCount);
    checkResult("<selection><caret>fruit</selection>\n" +
                "fruit\n" +
                "<selection><caret>fruit</selection>");
  }

  private void init(String text) {
    myFixture.configureByText(FileTypes.PLAIN_TEXT, text);
  }

  private void checkResult(String text) {
    myFixture.checkResult(text);
  }

  private void executeAction() {
    myFixture.performEditorAction(IdeActions.ACTION_SELECT_NEXT_OCCURENCE);
  }

  private void executeFindNext() {
    myFixture.performEditorAction(IdeActions.ACTION_FIND_NEXT);
  }

  private void executeFindPrevious() {
    myFixture.performEditorAction(IdeActions.ACTION_FIND_PREVIOUS);
  }

  private void executeReverseAction() {
    myFixture.performEditorAction(IdeActions.ACTION_UNSELECT_PREVIOUS_OCCURENCE);
  }

  private void executeSelectAllAction() {
    myFixture.performEditorAction(IdeActions.ACTION_SELECT_ALL_OCCURRENCES);
  }
}