添加单元测试

Author: Salpadding

wisdom-core/src/main/java/org/wisdom/vm/abi/SafeMath.java

@@ -31,21 +31,38 @@ public static long mul(long x, long y) {
     }
 
     public static void main(String[] args) {
+        // 正常加法
+        assertTrue(add(Long.MAX_VALUE, 0) == Long.MAX_VALUE);
+        assertTrue(add(0, Long.MAX_VALUE) == Long.MAX_VALUE);
+        assertTrue(add(Long.MAX_VALUE - 1, 1) == Long.MAX_VALUE);
+        assertTrue(add(0, 0) == 0);
+        assertTrue(add(0, 1) ==  1);
+        assertTrue(add(1, 0) == 1);
+        assertTrue(add(1, 1) == 2);
+
         // 加法溢出
-        assertException(() -> {
-            add(1, Long.MAX_VALUE);
-        });
-        assertException(() -> {
-            add(Long.MAX_VALUE, 1);
-        });
+        assertException(() -> add(1, Long.MAX_VALUE));
+        assertException(() -> add(Long.MAX_VALUE, 1));
+        assertException(() -> add(-1, 0));
+
+        // 普通减法
+        assertTrue(sub(1, 0) == 1);
+        assertException(() -> sub(0, 1));
+        assertTrue(sub(Long.MAX_VALUE, 1) == Long.MAX_VALUE - 1);
+        assertTrue(sub(Long.MAX_VALUE, 0) == Long.MAX_VALUE);
+
         // 减法溢出
-        assertException(() -> {
-            sub(1, 2);
-        });
+        assertException(() -> sub(1, 2));
+
+        assertException(() -> sub(-1, 0));
+
+        assertTrue(mul(1, 1) == 1);
+        assertTrue(mul(1, Long.MAX_VALUE) == Long.MAX_VALUE);
+        assertTrue(mul(0, Long.MAX_VALUE) == 0);
+
         // 乘法溢出
-        assertException(() -> {
-            mul(Long.MAX_VALUE/2 + 1, 2);
-        });
+        assertException(() -> mul(Long.MAX_VALUE/2 + 1, 2));
+        assertException(() -> mul(-1, 1));
     }
 
     public static void assertException(Runnable r){
@@ -58,4 +75,9 @@ public static void assertException(Runnable r){
         if(e0 == null)
             throw new RuntimeException("assert failed");
     }
+
+    public static void assertTrue(boolean b){
+        if(!b)
+            throw new RuntimeException("assert failed");
+    }
 }

wisdom-core/src/main/java/org/wisdom/vm/abi/Uint256.java

@@ -290,21 +290,39 @@ public long longValue(){
     }
 
     public static void main(String[] args) {
+        // 正常加法
+        assertTrue(MAX_U256.safeAdd(Uint256.ZERO).equals(MAX_U256));
+        assertTrue(Uint256.ZERO.safeAdd(MAX_U256).equals(MAX_U256));
+        assertTrue(MAX_U256.safeSub(Uint256.ONE).safeAdd(Uint256.ONE).equals(MAX_U256));
+        assertTrue(Uint256.ZERO.safeAdd(Uint256.ZERO).equals(Uint256.ZERO));
+        assertTrue(Uint256.ZERO.safeAdd(Uint256.ONE).equals(Uint256.ONE));
+        assertTrue(Uint256.ONE.safeAdd(Uint256.ZERO).equals(Uint256.ONE));
+        assertTrue(Uint256.ONE.safeAdd(Uint256.ONE).equals(Uint256.of(2)));
+
         // 加法溢出
-        assertException(() -> {
-            MAX_U256.safeAdd(ONE);
-        });
-        assertException(() -> {
-            ONE.safeAdd(MAX_U256);
-        });
+        assertException(() -> Uint256.ONE.safeAdd(Uint256.MAX_U256));
+        assertException(() -> Uint256.MAX_U256.safeAdd(Uint256.ONE));
+        assertException(() -> Uint256.MAX_U256.safeSub(Uint256.ONE).safeAdd(Uint256.of(2)));
+
+        // 普通减法
+        assertTrue(Uint256.ONE.safeSub(Uint256.ZERO).equals(Uint256.ONE));
+        assertException(() -> Uint256.ZERO.safeSub(Uint256.ONE));
+        assertTrue(Uint256.MAX_U256.safeSub(Uint256.ONE).equals(Uint256.of("fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffe", 16)));
+        assertTrue(Uint256.MAX_U256.safeSub(Uint256.ZERO).equals(Uint256.MAX_U256));
+
         // 减法溢出
-        assertException(() -> {
-            ONE.safeSub(Uint256.of(2));
-        });
+        assertException(() -> Uint256.ONE.safeSub(Uint256.of(2)));
+
+
+        assertTrue(Uint256.ONE.safeMul(Uint256.ONE).equals(Uint256.ONE));
+        assertTrue(Uint256.MAX_U256.safeMul(Uint256.ONE).equals(Uint256.MAX_U256));
+        assertTrue(Uint256.MAX_U256.safeMul(Uint256.ZERO).equals(Uint256.ZERO));
+
         // 乘法溢出
-        assertException(() -> {
-            MAX_U256.safeMul(Uint256.of(2));
-        });
+        assertException(() -> Uint256.MAX_U256.div(Uint256.of(2)).add(Uint256.ONE).safeMul(Uint256.of(2)));
+
+        // 除法溢出
+        assertException(() -> Uint256.ONE.div(Uint256.ZERO));
     }
 
     public static void assertException(Runnable r){
@@ -317,4 +335,9 @@ public static void assertException(Runnable r){
         if(e0 == null)
             throw new RuntimeException("assert failed");
     }
+
+    public static void assertTrue(boolean b){
+        if(!b)
+            throw new RuntimeException("assert failed");
+    }
 }