1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/*
* Copyright (C) 2007-2010 Júlio Vilmar Gesser.
* Copyright (C) 2011, 2013-2016 The JavaParser Team.
*
* This file is part of JavaParser.
*
* JavaParser can be used either under the terms of
* a) the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* b) the terms of the Apache License
*
* You should have received a copy of both licenses in LICENCE.LGPL and
* LICENCE.APACHE. Please refer to those files for details.
*
* JavaParser is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*/
package com.github.javaparser.ast.visitor;
import com.github.javaparser.JavaParser;
import com.github.javaparser.ast.NodeList;
import com.github.javaparser.ast.body.BodyDeclaration;
import com.github.javaparser.ast.body.VariableDeclarator;
import com.github.javaparser.ast.expr.Expression;
import com.github.javaparser.ast.expr.IntegerLiteralExpr;
import com.github.javaparser.ast.expr.StringLiteralExpr;
import org.junit.Test;
import static com.github.javaparser.JavaParser.parseExpression;
import static com.github.javaparser.utils.Utils.EOL;
import static org.junit.Assert.assertEquals;
public class ModifierVisitorTest {
@Test
public void makeSureParentListsCanBeModified() {
NodeList<StringLiteralExpr> list = new NodeList<>();
list.add(new StringLiteralExpr("t"));
list.add(new StringLiteralExpr("a"));
list.add(new StringLiteralExpr("b"));
list.add(new StringLiteralExpr("c"));
list.accept(new ModifierVisitor<Void>() {
@Override
public Visitable visit(final StringLiteralExpr n, final Void arg) {
String v = n.getValue();
list.addFirst(new StringLiteralExpr("extra " + v));
list.remove(new StringLiteralExpr("t"));
if (v.equals("a")) {
return new StringLiteralExpr("x");
}
if (v.equals("b")) {
return null;
}
return n;
}
}, null);
assertEquals("extra c", list.get(0).getValue());
assertEquals("extra b", list.get(1).getValue());
assertEquals("extra a", list.get(2).getValue());
assertEquals("extra t", list.get(3).getValue());
assertEquals("x", list.get(4).getValue());
assertEquals("c", list.get(5).getValue());
assertEquals(6, list.size());
}
@Test
public void binaryExprReturnsLeftExpressionWhenRightSideIsRemoved() {
Expression expression = parseExpression("1+2");
Visitable result = expression.accept(new ModifierVisitor<Void>() {
public Visitable visit(IntegerLiteralExpr integerLiteralExpr, Void arg) {
if (integerLiteralExpr.getValue().equals("1")) {
return null;
}
return integerLiteralExpr;
}
}, null);
assertEquals("2", result.toString());
}
@Test
public void binaryExprReturnsRightExpressionWhenLeftSideIsRemoved() {
final Expression expression = parseExpression("1+2");
final Visitable result = expression.accept(new ModifierVisitor<Void>() {
public Visitable visit(IntegerLiteralExpr integerLiteralExpr, Void arg) {
if (integerLiteralExpr.getValue().equals("2")) {
return null;
}
return integerLiteralExpr;
}
}, null);
assertEquals("1", result.toString());
}
@Test
public void fieldDeclarationCantSurviveWithoutVariables() {
final BodyDeclaration<?> bodyDeclaration = JavaParser.parseBodyDeclaration("int x=1;");
final Visitable result = bodyDeclaration.accept(new ModifierVisitor<Void>() {
public Visitable visit(VariableDeclarator x, Void arg) {
return null;
}
}, null);
assertEquals(null, result);
}
@Test
public void variableDeclarationCantSurviveWithoutVariables() {
final BodyDeclaration<?> bodyDeclaration = JavaParser.parseBodyDeclaration("void x() {int x=1;}");
final Visitable result = bodyDeclaration.accept(new ModifierVisitor<Void>() {
public Visitable visit(VariableDeclarator x, Void arg) {
return null;
}
}, null);
assertEquals("void x() {" + EOL + "}", result.toString());
}
}
|