fix(ape-test): add support for signing eip712 messages (#1965)

* test: update test for eip712 signing in test account to fail * fix(ape-test): add support for signing eip712 messages to TestAccount
ApeWorX · Mar 14, 2024 · e36aea4 · e36aea4
1 parent ee4a3a6
commit e36aea4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/ape_test/accounts.py b/src/ape_test/accounts.py
@@ -1,5 +1,6 @@
 from typing import Any, Iterator, List, Optional
 
+from eip712.messages import EIP712Message
 from eth_account import Account as EthAccount
 from eth_account.messages import SignableMessage, encode_defunct
 from eth_utils import to_bytes
@@ -110,6 +111,9 @@ def sign_message(self, msg: Any, **signer_options) -> Optional[MessageSignature]
         elif isinstance(msg, int):
             msg = HexBytes(msg).hex()
             msg = encode_defunct(hexstr=msg)
+        elif isinstance(msg, EIP712Message):
+            # Convert EIP712Message to SignableMessage for handling below
+            msg = msg.signable_message
 
         # Process SignableMessage
         if isinstance(msg, SignableMessage):

diff --git a/tests/functional/test_accounts.py b/tests/functional/test_accounts.py
@@ -95,9 +95,8 @@ def test_recover_signer(signer, message):
 
 def test_sign_eip712_message(signer):
     foo = Foo(signer.address)  # type: ignore[call-arg]
-    message = foo.signable_message
-    signature = signer.sign_message(message)
-    assert signer.check_signature(message, signature)
+    signature = signer.sign_message(foo)
+    assert signer.check_signature(foo, signature)
 
 
 def test_sign_message_with_prompts(runner, keyfile_account, message):