Fix sign_message for eip191

safe-global · Jun 18, 2024 · 949e84f · 949e84f
1 parent 3911dfd
commit 949e84f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/safe_cli/operators/safe_operator.py b/src/safe_cli/operators/safe_operator.py
@@ -468,7 +468,7 @@ def sign_message(
         else:
             print_formatted_text("EIP191 message to sign:")
             message = get_input()
-            message_bytes = eip191_message.encode("UTF-8")
+            message_bytes = message.encode("UTF-8")
 
         safe_message_hash = self.safe.get_message_hash(message_bytes)
 

diff --git a/tests/test_safe_operator.py b/tests/test_safe_operator.py
@@ -185,7 +185,9 @@ def test_sign_message(self, mock_sign_message_lib_address):
         safe_operator.accounts.add(self.ethereum_test_account)
         safe_operator.default_sender = self.ethereum_test_account
         message_hash = safe_operator.safe.get_message_hash(bytes(message, "utf-8"))
-        safe_operator.sign_message(eip191_message=message)
+        with mock.patch("builtins.input", return_value=message):
+            safe_operator.sign_message(eip191_message=True)
+
         self.assertTrue(safe_operator.safe.retrieve_is_message_signed(message_hash))
         eip712_path = "tests/mocks/mock_eip712.json"
         message = json.load(open(eip712_path, "r"))