method chains search: fix context extraction for assignment expression

This commit is contained in:
Dmitry Batkovich
2017-04-28 12:44:25 +03:00
parent 4beecd238e
commit 1fc4e42986
@@ -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;