diff --git a/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java b/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java index 3528d7c6517d..18dec8438a21 100644 --- a/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java +++ b/python/testSrc/com/jetbrains/python/fixtures/PyTestCase.java @@ -16,6 +16,7 @@ package com.jetbrains.python.fixtures; import com.intellij.codeInsight.intention.IntentionAction; +import com.intellij.codeInsight.lookup.LookupElement; import com.intellij.codeInspection.LocalQuickFix; import com.intellij.codeInspection.ex.QuickFixWrapper; import com.intellij.execution.actions.ConfigurationContext; @@ -71,6 +72,7 @@ import org.jetbrains.annotations.Nullable; import java.io.File; import java.util.ArrayList; import java.util.Collection; +import java.util.List; /** * @author yole @@ -240,6 +242,19 @@ public abstract class PyTestCase extends UsefulTestCase { return result; } + /** + * @return completion strings suggested by {@link com.intellij.testFramework.fixtures.CodeInsightTestFixture#completeBasic()} + */ + @NotNull + protected List getCompletionStrings() { + final LookupElement[] elements = myFixture.completeBasic(); + final List result = new ArrayList(elements.length); + for (final LookupElement element : elements) { + result.add(element.getLookupString()); + } + return result; + } + protected static class PyLightProjectDescriptor implements LightProjectDescriptor { private final String myPythonVersion;