From a13bec1e73c5204b5ddb663657eebe9d0045e367 Mon Sep 17 00:00:00 2001
From: peter <peter@jetbrains.com>
Date: Thu, 7 Jun 2018 06:24:08 +0200
Subject: [PATCH] mad testing completion: use reference under caret, not before
 it

---
 .../testFramework/propertyBased/InvokeCompletion.java        | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/platform/testFramework/src/com/intellij/testFramework/propertyBased/InvokeCompletion.java b/platform/testFramework/src/com/intellij/testFramework/propertyBased/InvokeCompletion.java
index 61619d27c06a..18d00bb77a92 100644
--- a/platform/testFramework/src/com/intellij/testFramework/propertyBased/InvokeCompletion.java
+++ b/platform/testFramework/src/com/intellij/testFramework/propertyBased/InvokeCompletion.java
@@ -107,10 +107,9 @@ public class InvokeCompletion extends ActionOnFile {
                                  char completionChar,
                                  Environment env) {
     int caretOffset = editor.getCaretModel().getOffset();
-    int adjustedOffset = TargetElementUtil.adjustOffset(file, getDocument(), caretOffset);
 
-    PsiElement leaf = file.findElementAt(adjustedOffset);
-    PsiReference ref = file.findReferenceAt(adjustedOffset);
+    PsiElement leaf = file.findElementAt(TargetElementUtil.adjustOffset(file, getDocument(), caretOffset));
+    PsiReference ref = TargetElementUtil.findReference(editor);
 
     String expectedVariant = leaf == null ? null : myPolicy.getExpectedVariant(editor, file, leaf, ref);
     boolean prefixEqualsExpected = isPrefixEqualToExpectedVariant(caretOffset, leaf, ref, expectedVariant);