IteratorDeclaration: support parentheses in declaration

This commit is contained in:
Tagir Valeev
2018-07-10 14:13:42 +07:00
parent b892f42553
commit c8eb903c46
2 changed files with 2 additions and 2 deletions
@@ -102,7 +102,7 @@ public class IteratorDeclaration {
PsiElement element = declaration.getDeclaredElements()[0];
if (!(element instanceof PsiLocalVariable)) return null;
PsiLocalVariable variable = (PsiLocalVariable)element;
PsiExpression initializer = variable.getInitializer();
PsiExpression initializer = PsiUtil.skipParenthesizedExprDown(variable.getInitializer());
if (!(initializer instanceof PsiMethodCallExpression)) return null;
PsiMethodCallExpression call = (PsiMethodCallExpression)initializer;
if (!call.getArgumentList().isEmpty()) return null;
@@ -4,7 +4,7 @@ import java.util.List;
public class Main {
public void testIterator(List<List<String>> data, boolean b) {
for(Ite<caret>rator<List<String>> iter = data.iterator(); iter.hasNext();) {
for(Ite<caret>rator<List<String>> iter = (data.iterator()); iter.hasNext();) {
if(iter.next().isEmpty() && /* also check the flag */ b) {
iter.remove();
}