From f890df70d168bd681cd5839be00307c5e92e5819 Mon Sep 17 00:00:00 2001 From: z80 Date: Thu, 21 Sep 2023 16:44:11 -0400 Subject: [PATCH] Revert "fix: add missing return statement" This reverts commit c9d9e12893ba9929506b2a2379b05b13f2d8c3c2. --- src/ape_test/accounts.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/src/ape_test/accounts.py b/src/ape_test/accounts.py index c596b3b165..733c6912de 100644 --- a/src/ape_test/accounts.py +++ b/src/ape_test/accounts.py @@ -101,13 +101,16 @@ def alias(self) -> str: def address(self) -> AddressType: return self.network_manager.ethereum.decode_address(self.address_str) - def sign_message(self, msg: SignableMessage) -> Optional[MessageSignature]: - signed_msg = EthAccount.sign_message(msg, self.private_key) - return MessageSignature( - v=signed_msg.v, - r=to_bytes(signed_msg.r), - s=to_bytes(signed_msg.s), - ) + def sign_message(self, msg: Any, **signer_options) -> Optional[MessageSignature]: + if isinstance(msg, str): + msg = encode_defunct(text=msg) + if isinstance(msg, SignableMessage): + signed_msg = EthAccount.sign_message(msg, self.private_key) + return MessageSignature( + v=signed_msg.v, + r=to_bytes(signed_msg.r), + s=to_bytes(signed_msg.s), + ) def sign_transaction(self, txn: TransactionAPI, **kwargs) -> Optional[TransactionAPI]: # Signs anything that's given to it