From 3e20852361aee1dd7f19bf2d8e266f75f3ec036c Mon Sep 17 00:00:00 2001
From: "Ilya.Kazakevich" <Ilya.Kazakevich@jetbrains.com>
Date: Tue, 7 Oct 2014 17:53:30 +0400
Subject: [PATCH] PY-13740 Django 1.7 QuerySet.as_manager() code completion
 support

---
 .../com/jetbrains/python/fixtures/PyTestCase.java | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

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<String> getCompletionStrings() {
+    final LookupElement[] elements = myFixture.completeBasic();
+    final List<String> result = new ArrayList<String>(elements.length);
+    for (final LookupElement element : elements) {
+      result.add(element.getLookupString());
+    }
+    return result;
+  }
+
   protected static class PyLightProjectDescriptor implements LightProjectDescriptor {
     private final String myPythonVersion;