Nikolay Krasko
81 lines
2.6 KiB
Java
81 lines
2.6 KiB
Java
package org.jetbrains.jet.completion;
|
|
|
|
import com.intellij.codeInsight.completion.CodeCompletionHandlerBase;
|
|
import com.intellij.codeInsight.completion.CompletionType;
|
|
import com.intellij.codeInsight.completion.LightCompletionTestCase;
|
|
import com.intellij.codeInsight.lookup.LookupElement;
|
|
import com.intellij.codeInsight.lookup.LookupManager;
|
|
import com.intellij.codeInsight.lookup.impl.LookupImpl;
|
|
import com.intellij.openapi.projectRoots.Sdk;
|
|
import org.jetbrains.annotations.NotNull;
|
|
import org.jetbrains.jet.plugin.PluginTestCaseBase;
|
|
|
|
import java.io.File;
|
|
|
|
/**
|
|
* @author Nikolay.Krasko
|
|
*/
|
|
public abstract class JetCompletionTestBase extends LightCompletionTestCase {
|
|
private final String myPath;
|
|
private final String myName;
|
|
|
|
protected JetCompletionTestBase(@NotNull String path, @NotNull String name) {
|
|
myPath = path;
|
|
myName = name;
|
|
|
|
// Set name explicitly because otherwise there will be "TestCase.fName cannot be null"
|
|
setName("testCompletionExecute");
|
|
}
|
|
|
|
public void testCompletionExecute() throws Exception {
|
|
doTest();
|
|
}
|
|
|
|
@Override
|
|
protected String getTestDataPath() {
|
|
return new File(PluginTestCaseBase.getTestDataPathBase(), myPath).getPath() +
|
|
File.separator;
|
|
}
|
|
|
|
@NotNull
|
|
@Override
|
|
public String getName() {
|
|
return "test" + myName;
|
|
}
|
|
|
|
private CompletionType type;
|
|
|
|
protected void doTest() throws Exception {
|
|
final String testName = getTestName(false);
|
|
|
|
type = (testName.startsWith("Smart")) ? CompletionType.SMART : CompletionType.BASIC;
|
|
|
|
configureByFile(testName + ".kt");
|
|
|
|
final String fileText = getFile().getText();
|
|
final ExpectedCompletionUtils completionUtils = new ExpectedCompletionUtils();
|
|
|
|
assertContainsItems(completionUtils.itemsShouldExist(fileText));
|
|
assertNotContainItems(completionUtils.itemsShouldAbsent(fileText));
|
|
|
|
Integer itemsNumber = completionUtils.getExpectedNumber(fileText);
|
|
if (itemsNumber != null) {
|
|
assertEquals(itemsNumber.intValue(), myItems.length);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected Sdk getProjectJDK() {
|
|
return PluginTestCaseBase.jdkFromIdeaHome();
|
|
}
|
|
|
|
@Override
|
|
protected void complete(final int time) {
|
|
new CodeCompletionHandlerBase(type, false, false, true).invokeCompletion(getProject(), getEditor(), time, false);
|
|
|
|
LookupImpl lookup = (LookupImpl) LookupManager.getActiveLookup(myEditor);
|
|
myItems = lookup == null ? null : lookup.getItems().toArray(LookupElement.EMPTY_ARRAY);
|
|
myPrefix = lookup == null ? null : lookup.itemPattern(lookup.getItems().get(0));
|
|
}
|
|
}
|