Tests

Author: sovdeeth

src/main/java/ch/njol/skript/expressions/arithmetic/ExprArithmetic.java

@@ -386,4 +386,20 @@ private Expression<T> simplifyInternal() {
 		return this;
 	}
 
+	/**
+	 * For testing purposes only.
+	 * @return the first expression
+	 */
+	Expression<L> getFirst() {
+		return first;
+	}
+
+	/**
+	 * For testing purposes only.
+	 * @return the second expression
+	 */
+	Expression<R> getSecond() {
+		return second;
+	}
+
 }

src/test/java/ch/njol/skript/expressions/arithmetic/ArithmeticSimplificationTest.java

@@ -0,0 +1,39 @@
+package ch.njol.skript.expressions.arithmetic;
+
+import ch.njol.skript.lang.SkriptParser;
+import ch.njol.skript.lang.Variable;
+import ch.njol.skript.lang.util.ContextlessEvent;
+import ch.njol.skript.test.runner.SkriptJUnitTest;
+import org.junit.Assert;
+import org.junit.Test;
+import org.skriptlang.skript.lang.simplification.SimplifiedLiteral;
+
+/**
+ * Check if arithmetic expressions are simplified correctly.
+ */
+public class ArithmeticSimplificationTest extends SkriptJUnitTest {
+
+	@Test
+	public void test() {
+		//noinspection unchecked
+		var arithmetic = new SkriptParser("5 * 2 - 3 + 4").parseExpression(Number.class);
+		Assert.assertTrue(arithmetic instanceof SimplifiedLiteral<? extends Number>);
+		Assert.assertEquals(5 * 2 - 3 + 4, ((SimplifiedLiteral<? extends Number>) arithmetic).getSingle().intValue());
+
+		//noinspection unchecked
+		arithmetic = new SkriptParser("5 + 4").parseExpression(Number.class);
+		Assert.assertTrue(arithmetic instanceof SimplifiedLiteral<? extends Number>);
+		Assert.assertEquals(5 + 4, ((SimplifiedLiteral<? extends Number>) arithmetic).getSingle().intValue());
+
+		//noinspection unchecked
+		arithmetic = new SkriptParser("5 - 10 + {_loc}").parseExpression(Number.class);
+		Assert.assertTrue(arithmetic instanceof ExprArithmetic<?,?,?>);
+		var leftArith = ((ExprArithmetic<?,?,?>) arithmetic).getFirst();
+		var rightArith = ((ExprArithmetic<?,?,?>) arithmetic).getSecond();
+		Assert.assertTrue(leftArith instanceof SimplifiedLiteral<?>);
+		Assert.assertTrue(rightArith instanceof Variable<?>);
+		//noinspection unchecked,DataFlowIssue
+		Assert.assertEquals(5 - 10, ((ExprArithmetic<?,?,Number>) arithmetic).getSingle(ContextlessEvent.get()).intValue());
+	}
+
+}