cast to Number

Activiti · Jan 11, 2017 · a8f2262 · a8f2262
1 parent 98309df
commit a8f2262
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/...es/activiti-engine/src/test/java/org/activiti/standalone/cfg/CustomMybatisMapperTest.java b/...es/activiti-engine/src/test/java/org/activiti/standalone/cfg/CustomMybatisMapperTest.java
@@ -1,6 +1,5 @@
 package org.activiti.standalone.cfg;
 
-import java.math.BigDecimal;
 import java.util.List;
 import java.util.Map;
 
@@ -81,16 +80,10 @@ public List<Map<String, Object>> execute(MyTestMapper customMapper) {
 		List<Map<String, Object>> results = managementService.executeCustomSql(customSqlExecution);
 		assertEquals(5, results.size());
 		for (int i=0; i<5; i++) {
-			Map<String, Object> result = results.get(i);
-			Long id = Long.valueOf((String) getResultObject("TASKID", result));
-			Long variableValue = null;
-			Object variableObjectValue = getResultObject("VARIABLEVALUE", result);
-			if (variableObjectValue instanceof BigDecimal) { // in oracle it's BigDecimal
-			  variableValue = ((BigDecimal)variableObjectValue).longValue();
-			} else {
-			  variableValue = (Long) variableObjectValue;
-			}
-			assertEquals(id * 2, variableValue.longValue());
+		  Map<String, Object> result = results.get(i);
+		  Long id = Long.valueOf((String) getResultObject("TASKID", result));
+		  Long variableValue = ((Number) getResultObject("VARIABLEVALUE", result)).longValue();
+		  assertEquals(id * 2, variableValue.longValue());
 		}
 
 		// Cleanup