added slice parser tests

lmittmann · Feb 23, 2022 · 1eb62a6 · 1eb62a6
1 parent 920ea66
commit 1eb62a6
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/internal/abi/parser_test.go b/internal/abi/parser_test.go
@@ -10,8 +10,8 @@ import (
 )
 
 var (
-	typeUint256, _ = abi.NewType("uint256", "", nil)
-	typeAddress, _ = abi.NewType("address", "", nil)
+	typeUint256 = abi.Type{T: abi.UintTy, Size: 256}
+	typeAddress = abi.Type{T: abi.AddressTy, Size: 20}
 )
 
 func TestParser(t *testing.T) {
@@ -38,6 +38,28 @@ func TestParser(t *testing.T) {
 			Input:    "address recipient, uint256 amount",
 			WantArgs: abi.Arguments{{Type: typeAddress, Name: "recipient"}, {Type: typeUint256, Name: "amount"}},
 		},
+		{
+			Input:    "uint256[]",
+			WantArgs: abi.Arguments{{Type: abi.Type{Elem: &typeUint256, T: abi.SliceTy}}},
+		},
+		{
+			Input: "uint256[][]",
+			WantArgs: abi.Arguments{{
+				Type: abi.Type{
+					Elem: &abi.Type{Elem: &typeUint256, T: abi.SliceTy},
+					T:    abi.SliceTy,
+				},
+			}},
+		},
+		{
+			Input: "uint256[3][]",
+			WantArgs: abi.Arguments{{
+				Type: abi.Type{
+					Elem: &abi.Type{Elem: &typeUint256, T: abi.ArrayTy, Size: 3},
+					T:    abi.SliceTy,
+				},
+			}},
+		},
 		{
 			Input:    "transfer(address,uint256)",
 			WantArgs: abi.Arguments{{Type: typeAddress}, {Type: typeUint256}},