fix: add error message for nonpayable check (#3466)

vyperlang · May 26, 2023 · 64733b9 · 64733b9
1 parent 056dfd8
commit 64733b9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/vyper/codegen/function_definitions/external_function.py b/vyper/codegen/function_definitions/external_function.py
@@ -200,7 +200,10 @@ def generate_ir_for_external_function(code, func_t, context, skip_nonpayable_che
     if not func_t.is_payable and not skip_nonpayable_check:
         # if the contract contains payable functions, but this is not one of them
         # add an assertion that the value of the call is zero
-        body += [["assert", ["iszero", "callvalue"]]]
+        nonpayable_check = IRnode.from_list(
+            ["assert", ["iszero", "callvalue"]], error_msg="nonpayable check"
+        )
+        body.append(nonpayable_check)
 
     body += nonreentrant_pre
 

diff --git a/vyper/codegen/module.py b/vyper/codegen/module.py
@@ -97,7 +97,10 @@ def _runtime_ir(runtime_functions, global_ctx):
         selector_section.append(func_ir)
 
     if batch_payable_check:
-        selector_section.append(["assert", ["iszero", "callvalue"]])
+        nonpayable_check = IRnode.from_list(
+            ["assert", ["iszero", "callvalue"]], error_msg="nonpayable check"
+        )
+        selector_section.append(nonpayable_check)
 
     for func_ast in nonpayables:
         func_ir = generate_ir_for_function(func_ast, global_ctx, skip_nonpayable_check)