java/java-tests/testSrc/com/intellij/codeInsight/completion/VariablesCompletionTest.groovy

package com.intellij.codeInsight.completion
import com.intellij.JavaTestUtil
import com.intellij.ide.highlighter.JavaFileType
import com.intellij.psi.codeStyle.CodeStyleSettings
import com.intellij.psi.codeStyle.CodeStyleSettingsManager

public class VariablesCompletionTest extends LightFixtureCompletionTestCase {
  public static final String FILE_PREFIX = "/codeInsight/completion/variables/";

  @Override
  protected String getBasePath() {
    return JavaTestUtil.getRelativeJavaTestDataPath();
  }

  public void testObjectVariable() throws Exception {
    configureByFile(FILE_PREFIX + "locals/" + getTestName(false) + ".java");
    checkResultByFile(FILE_PREFIX + "locals/" + getTestName(false) + "_after.java");
  }

  public void testStringVariable() throws Exception {
    configureByFile(FILE_PREFIX + "locals/" + getTestName(false) + ".java");
    checkResultByFile(FILE_PREFIX + "locals/" + getTestName(false) + "_after.java");
  }

  public void testInputMethodEventVariable() throws Exception {
    myFixture.addClass("package java.awt.event; public interface InputMethodEvent {}");

    configureByFile(FILE_PREFIX + "locals/" + getTestName(false) + ".java");
    checkResultByFile(FILE_PREFIX + "locals/" + getTestName(false) + "_after.java");
  }

  public void testLocals1() throws Exception {
    doSelectTest("TestSource1.java", "TestResult1.java");
  }

  public void testInterfaceMethod() throws Exception {
    configureByFile(FILE_PREFIX + "locals/" + "InterfaceMethod.java");
    assertStringItems("calcGooBarDoo", "calcBarDoo", "calcDoo");
  }

  public void testLocals2() throws Exception {
    configureByFile(FILE_PREFIX + "locals/" + "TestSource2.java");
    myFixture.assertPreferredCompletionItems 0, 'abc', 'aaa'
    checkResultByFile(FILE_PREFIX + "locals/" + "TestResult2.java");
  }

  public void testLocals3() throws Exception {
    doSelectTest("TestSource3.java", "TestResult3.java");
  }

  public void testLocals4() throws Exception {
    doSelectTest("TestSource4.java", "TestResult4.java");
  }

  public void testLocals5() throws Exception {
    doSelectTest("TestSource5.java", "TestResult5.java");
  }

  public void testLocals6() throws Exception {
    doSelectTest("TestSource6.java", "TestResult6.java");
  }

  public void testLocals7() throws Exception {
    doTest("TestSource7.java", "TestResult7.java");
  }

  public void testLocalReserved() throws Exception {
    doTest("LocalReserved.java", "LocalReserved_after.java");
  }

  public void testLocalReserved2() throws Exception {
    configureByFile(FILE_PREFIX + "locals/" + "LocalReserved2.java")
    checkResultByFile(FILE_PREFIX + "locals/" + "LocalReserved2.java");
    assert !myFixture.lookupElementStrings
  }

  public void testUniqueNameInFor() throws Exception {
    doTest(getTestName(false) + ".java", getTestName(false) + "_after.java");
  }
  public void testWithBuilderParameter() throws Exception {
    doTest(getTestName(false) + ".java", getTestName(false) + "_after.java");
  }

  private void doTest(String before, String after) throws Exception {
    configureByFile(FILE_PREFIX + "locals/" + before);
    checkResultByFile(FILE_PREFIX + "locals/" + after);
  }

  private void doSelectTest(String before, String after) throws Exception {
    configureByFile(FILE_PREFIX + "locals/" + before);
    myFixture.type('\n')
    checkResultByFile(FILE_PREFIX + "locals/" + after);
  }

  public void testLocals8() throws Exception {
    doTest("TestSource8.java", "TestResult8.java");
  }

  public void testUnresolvedReference() throws Exception {
    configureByFile(FILE_PREFIX + "locals/" + getTestName(false) + ".java");
    assertStringItems("o", "psiClass");
  }

  public void testFieldNameCompletion1() throws Exception {
    CodeStyleSettings settings = CodeStyleSettingsManager.getSettings(getProject());
    String oldPrefix = settings.FIELD_NAME_PREFIX;
    settings.FIELD_NAME_PREFIX = "my";
    try {
      doSelectTest("FieldNameCompletion1.java", "FieldNameCompletion1-result.java");
    }
    finally {
      settings.FIELD_NAME_PREFIX = oldPrefix;
    }
  }

  public void testFieldNameCompletion2() throws Exception {
    CodeStyleSettings settings = CodeStyleSettingsManager.getSettings(project);
    String oldPrefix = settings.FIELD_NAME_PREFIX;
    settings.FIELD_NAME_PREFIX = "my";
    configureByFile(FILE_PREFIX + "locals/" + "FieldNameCompletion2.java");
    settings.FIELD_NAME_PREFIX = oldPrefix;
    checkResultByFile(FILE_PREFIX + "locals/" + "FieldNameCompletion2-result.java");
  }

  public void testFieldNameCompletion3() throws Exception {
    CodeStyleSettings settings = CodeStyleSettingsManager.getSettings(project);
    String oldPrefix = settings.FIELD_NAME_PREFIX;
    settings.FIELD_NAME_PREFIX = "my";
    configureByFile(FILE_PREFIX + "locals/" + "FieldNameCompletion3.java");
    complete();
    settings.FIELD_NAME_PREFIX = oldPrefix;
    checkResultByFile(FILE_PREFIX + "locals/" + "FieldNameCompletion3-result.java");
  }

  public void testLocals9() throws Exception {
    doSelectTest("TestSource9.java", "TestResult9.java");
  }

  public void testFieldOutOfAnonymous() throws Exception {
    doTest("TestFieldOutOfAnonymous.java", "TestFieldOutOfAnonymousResult.java");
  }

  public void testUnresolvedMethodName() throws Exception {
    configureByFile(FILE_PREFIX + "locals/" + "UnresolvedMethodName.java");
    complete();
    checkResultByFile(FILE_PREFIX + "locals/" + "UnresolvedMethodName_after.java");
  }

  public void testArrayMethodName() throws Throwable {
    doTest("ArrayMethodName.java", "ArrayMethodName-result.java");
  }

  public void testInitializerMatters() throws Exception {
    myFixture.configureByText(JavaFileType.INSTANCE, "class Foo {{ String f<caret>x = getFoo(); }; String getFoo() {}; }");
    complete();
    assertStringItems("foo");
  }

  public void testFieldInitializerMatters() throws Exception {
    myFixture.configureByText(JavaFileType.INSTANCE, "class Foo { String f<caret>x = getFoo(); String getFoo() {}; }");
    complete();
    assertStringItems("foo", "fString");
  }

  public void testNoKeywordsInForLoopVariableName() throws Throwable {
    configure()
    assertStringItems("stringBuffer", "buffer");
  }

  public void testDuplicateSuggestionsFromUsage() {
    configure();
    assertStringItems("preferencePolicy", "policy", "aPreferencePolicy");
  }

  public void configure() {
    configureByFile(FILE_PREFIX + getTestName(false) + ".java")
  }

  public void testAnnotationValue() {
    configure()
    checkResultByFile(FILE_PREFIX + getTestName(false) + "_after.java")
  }

  public void testConstructorParameterName() {
    configure()
    assertStringItems("color");
  }

  public void testConstructorParameterNameWithPrefix() {
    CodeStyleSettings settings = CodeStyleSettingsManager.getSettings(project);
    String oldField = settings.FIELD_NAME_PREFIX;
    String oldParam = settings.PARAMETER_NAME_PREFIX;
    settings.FIELD_NAME_PREFIX = "my";
    settings.PARAMETER_NAME_PREFIX = "p";

    configure()

    settings.FIELD_NAME_PREFIX = oldField
    settings.PARAMETER_NAME_PREFIX = oldParam

    assertStringItems("pColor");
  }

  public void "test finish with ="() {
    myFixture.configureByText 'a.java', '''
class FooFoo {
  FooFoo f<caret>
}
'''
    myFixture.completeBasic()
    myFixture.assertPreferredCompletionItems 0, 'fooFoo', 'foo'
    myFixture.type '='
    myFixture.checkResult '''
class FooFoo {
  FooFoo fooFoo = <caret>
}
'''
  }

  public void "test suggest variable names by non-getter initializer call"() {
    myFixture.configureByText 'a.java', '''
class FooFoo {
  { long <caret>x = System.nanoTime(); }
}
'''
    myFixture.completeBasic()
    myFixture.assertPreferredCompletionItems 0, 'l', 'nanoTime', 'time'
  }

  public void "test use superclass for inner class variable name suggestion"() {
    myFixture.configureByText 'a.java', '''
class FooFoo {
  { Rectangle2D.Double <caret>x }
}
class Rectangle2D {
  static class Double extends Rectangle2D {}
}
'''
    myFixture.completeBasic()
    myFixture.assertPreferredCompletionItems 0, 'aDouble', 'rectangle2D'
  }

}