fix: adds support to for the NEG Unary Operator in VisitorPartialEval…

…uator (#3193)
INRIA · Dec 15, 2019 · 5e23a7e · 5e23a7e
1 parent 9db3f61
commit 5e23a7e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/main/java/spoon/support/reflect/eval/VisitorPartialEvaluator.java b/src/main/java/spoon/support/reflect/eval/VisitorPartialEvaluator.java
@@ -524,6 +524,10 @@ public <T> void visitCtUnaryOperator(CtUnaryOperator<T> operator) {
 			case NOT:
 				res.setValue(!(Boolean) object);
 				break;
+			case NEG:
+				res.setValue(convert(operator.getType(),
+					-1 * ((Number) object).longValue()));
+				break;
 			default:
 				throw new RuntimeException("unsupported operator " + operator.getKind());
 			}

diff --git a/src/test/java/spoon/test/eval/EvalTest.java b/src/test/java/spoon/test/eval/EvalTest.java
@@ -187,6 +187,19 @@ public void testVisitorPartialEvaluator_binary() {
 		}
 	}
 
+	@Test
+	public void testVisitorPartialEvaluator_unary() {
+
+    	{ // NEG urnary operator
+      		Launcher launcher = new Launcher();
+      		CtCodeElement el =
+          		launcher.getFactory().Code().createCodeSnippetExpression("-(100+1)").compile();
+      		VisitorPartialEvaluator eval = new VisitorPartialEvaluator();
+      		CtElement element = eval.evaluate(el);
+      		assertEquals("-101", element.toString());
+    	}
+	}
+
 	@Test
 	public void testVisitorPartialEvaluator_if() {
 		Launcher launcher = new Launcher();