Assert that rootTokenToChildToken exists in the provided predicate ABI

0xPolygon · Jun 25, 2023 · b27d940 · b27d940
1 parent 99acd6c
commit b27d940
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/e2e-polybft/e2e/helpers_test.go b/e2e-polybft/e2e/helpers_test.go
@@ -337,7 +337,8 @@ func getChildToken(t *testing.T, predicateABI *abi.ABI, predicateAddr types.Addr
 	rootToken types.Address, relayer txrelayer.TxRelayer) types.Address {
 	t.Helper()
 
-	rootToChildTokenFn := predicateABI.Methods["rootTokenToChildToken"]
+	rootToChildTokenFn, exists := predicateABI.Methods["rootTokenToChildToken"]
+	require.True(t, exists, "rootTokenToChildToken function is not found in the provided predicate ABI definition")
 	input, err := rootToChildTokenFn.Encode([]interface{}{rootToken})
 	require.NoError(t, err)