method chains search: fix context extraction for assignment expression

1fc4e429 · Dmitry Batkovich · 4beecd23 · 1fc4e429
Commit 1fc4e429 authored 8 years ago by Dmitry Batkovich
Hide whitespace changes
Inline Side-by-side

Showing

with 3 additions and 2 deletions
+3 -2
--- a/java/compiler/impl/src/com/intellij/compiler/chainsSearch/completion/MethodsChainsCompletionContributor.java
+++ b/java/compiler/impl/src/com/intellij/compiler/chainsSearch/completion/MethodsChainsCompletionContributor.java
@@ -154,8 +154,9 @@ public class MethodsChainsCompletionContributor extends CompletionContributor {
  @Nullable
  private static ChainCompletionContext extractContextFromAssignment(PsiAssignmentExpression assignmentExpression,
                                                                     CompletionParameters parameters) {
-    if (!(assignmentExpression instanceof PsiReferenceExpression)) return null;
-    PsiElement resolved = ((PsiReferenceExpression)assignmentExpression).resolve();
+    PsiExpression lExpr = assignmentExpression.getLExpression();
+    if (!(lExpr instanceof PsiReferenceExpression)) return null;
+    PsiElement resolved = ((PsiReferenceExpression)lExpr).resolve();
    return resolved instanceof PsiVariable
           ? ChainCompletionContext.createContext(((PsiVariable)resolved).getType(), assignmentExpression, suggestIterators(parameters))
           : null;