plugins/groovy/test/org/jetbrains/plugins/groovy/GroovyActionsTest.groovy

/*
 * Copyright 2000-2009 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.plugins.groovy;


import com.intellij.ide.DataManager
import com.intellij.openapi.actionSystem.IdeActions
import com.intellij.openapi.editor.Editor
import com.intellij.openapi.editor.actionSystem.EditorActionHandler
import com.intellij.openapi.editor.actionSystem.EditorActionManager
import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase
import org.jetbrains.plugins.groovy.util.TestUtils

/**
 * @author peter
 */
public class GroovyActionsTest extends LightCodeInsightFixtureTestCase {

  final String basePath =  TestUtils.testDataPath + 'groovy/actions/';

  public void testSelectWordBeforeMethod() {
    doTestForSelectWord 1;
  }

  public void testSWInGString1() {doTestForSelectWord(1);}
  public void testSWInGString2() {doTestForSelectWord(2);}
  public void testSWInGString3() {doTestForSelectWord(3);}
  public void testSWInGString4() {doTestForSelectWord(4);}
  public void testSWInGString5() {doTestForSelectWord(5);}
  public void testSWInParameterList() {doTestForSelectWord(3);}
  public void testSWInArgLabel1() {doTestForSelectWord(2)}
  public void testSWInArgLabel2() {doTestForSelectWord(2)}
  public void testSWInArgLabel3() {doTestForSelectWord(2)}

  public void testSWListLiteralArgument() {
    doTestForSelectWord 2,
"foo([a<caret>], b)",
"foo(<selection>[a<caret>]</selection>, b)"
  }

  public void testSWMethodParametersBeforeQualifier() {
    doTestForSelectWord 2,
"a.fo<caret>o(b)",
"a.<selection>foo(b)</selection>"
  }

  public void testSWInCodeBlock() {doTestForSelectWord 5}

  public void testElseBranch() {
    doTestForSelectWord (3, '''\
def foo() {
  if (a){
  }
  else <caret>{
  }
}
''', '''\
def foo() {
  if (a){
  }
<selection>  else <caret>{
  }
</selection>}
''')
  }

  void testBlocksOfCode() {
    doTestForSelectWord(8, '''\
this.allOptions = [:];
    confTag.option.each{ opt ->
      def value = opt.'@value';
      if (value == null) {
        value = opt.value ? opt.value[0].'@defaultName' : null;
      }
      this.allOptions[opt.'@name'] = value;
    }

    def moduleNode = confTag.mod<caret>ule[0] ;
    if (moduleNode != null && !"wholeProject".equals(this.allOptions['TEST_SEARCH_SCOPE'])) {
      this.moduleRef = JpsElementFactory.instance.createModuleReference(moduleNode.'@name');
    } else {
      this.moduleRef = null;
    }

    this.macroExpander = macroExpander;
''', '''\
this.allOptions = [:];
    confTag.option.each{ opt ->
      def value = opt.'@value';
      if (value == null) {
        value = opt.value ? opt.value[0].'@defaultName' : null;
      }
      this.allOptions[opt.'@name'] = value;
    }

<selection>    def moduleNode = confTag.mod<caret>ule[0] ;
    if (moduleNode != null && !"wholeProject".equals(this.allOptions['TEST_SEARCH_SCOPE'])) {
      this.moduleRef = JpsElementFactory.instance.createModuleReference(moduleNode.'@name');
    } else {
      this.moduleRef = null;
    }
</selection>
    this.macroExpander = macroExpander;
''')
  }

  void testSWforMemberWithDoc() {
    doTestForSelectWord(4, '''\
class A {
  /**
   * abc
   */
  def fo<caret>o() {}

  def bar(){}
}
''', '''\
class A {
<selection>  /**
   * abc
   */
  def fo<caret>o() {}
</selection>
  def bar(){}
}
''')
  }

  private void doTestForSelectWord(int count, String input, String expected) {
    myFixture.configureByText("a.groovy", input);
    selectWord(count)
    myFixture.checkResult(expected);
  }

  private void doTestForSelectWord(int count) {
    myFixture.configureByFile(getTestName(false) + ".groovy");
    selectWord(count)
    myFixture.checkResultByFile(getTestName(false) + "_after.groovy");
  }

  private def selectWord(int count) {
    myFixture.editor.settings.camelWords = true;
    for (int i = 0; i < count; i++) {
      performEditorAction(IdeActions.ACTION_EDITOR_SELECT_WORD_AT_CARET);
    }
  }

  private void performEditorAction(final String actionId) {
    final EditorActionHandler handler = EditorActionManager.instance.getActionHandler(actionId);
    final Editor editor = myFixture.editor;
    handler.execute(editor, DataManager.instance.getDataContext(editor.contentComponent));
  }

}