plugins/groovy/test/org/jetbrains/plugins/groovy/completion/GroovyCompletionTestBase.groovy

/*
 * 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 org.jetbrains.plugins.groovy.completion
import com.intellij.codeInsight.CodeInsightSettings
import com.intellij.codeInsight.completion.CompletionType
import com.intellij.codeInsight.lookup.LookupManager
import com.intellij.codeInsight.lookup.impl.LookupImpl
import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase

abstract public class GroovyCompletionTestBase extends LightCodeInsightFixtureTestCase {

  protected void doSmartTest() {
    doCompletionTest(CompletionType.SMART)
  }

  protected void doBasicTest(String before =  null, String after = null) {
    doCompletionTest(before, after, CompletionType.BASIC)
  }

  protected void doSmartTest(String before, String after) {
    doCompletionTest(before, after, CompletionType.SMART)
  }

  protected void checkResult() {
    myFixture.checkResultByFile(getTestName(false) + "_after.groovy", true);
  }

  protected void doCompletionTest(String before = null, String after = null, String type = "", CompletionType ct) {
    if (before == null) {
      myFixture.configureByFile(getTestName(false) + ".groovy")
    }
    else {
      myFixture.configureByText(getTestName(false) + ".groovy", before)
    }

    myFixture.complete(ct)
    type.each { myFixture.type(it) }

    assertNull(myFixture.lookupElementStrings as String, myFixture.lookupElements)
    if (after == null) {
      myFixture.checkResultByFile(getTestName(false) + "_after.groovy", true)
    }
    else {
      myFixture.checkResult(after, true)
    }
  }

  protected void doVariantableTest(String before = null, String type = "", CompletionType ct, CompletionResult testType = CompletionResult.equal, int completionCount = 1, String... variants) {
    if (before == null) {
      myFixture.configureByFile(getTestName(false) + ".groovy")
    }
    else {
      myFixture.configureByText(getTestName(false) + ".groovy", before)
    }

    myFixture.complete(ct, completionCount)
    type.each { myFixture.type(it) }

    assertNotNull(myFixture.lookupElements)

    final actual = myFixture.lookupElementStrings
    switch (testType) {
      case CompletionResult.contain:
        assertTrue(myFixture.lookupElementStrings as String, actual.containsAll(variants))
        break
      case CompletionResult.equal:
        assertOrderedEquals(actual, variants)
        break
      case CompletionResult.notContain:
        variants.each {
          assertFalse(myFixture.lookupElementStrings as String, actual.contains(it))
        }
    }
  }


  public void doVariantableTest(String... variants) {
    doVariantableTest(CompletionType.BASIC, variants)
  }

  public void doHasVariantsTest(String... variants) {
    doVariantableTest(null, "", CompletionType.BASIC, CompletionResult.contain, variants)
  }

  public void doSmartCompletion(String... variants) {
    doVariantableTest(CompletionType.SMART, variants)
  }

  public void checkCompletion(String before, String type, String after) {
    doCompletionTest(before, after, type, CompletionType.BASIC)
  }

  public void checkSingleItemCompletion(String before, String after) {
    doCompletionTest(before, after, CompletionType.BASIC)
  }

  public void doNoVariantsTest(String before, String... excludedVariants) {
    doVariantableTest(before, "", CompletionType.BASIC, CompletionResult.notContain, excludedVariants)
  }

  protected static def caseSensitiveNone() {
    CodeInsightSettings.instance.COMPLETION_CASE_SENSITIVE = CodeInsightSettings.NONE
  }

  LookupImpl getLookup() {
    LookupManager.getActiveLookup(myFixture.editor)
  }

}