SKALE-2468 Add sign_hash method to sgx wallet

setup.py

@@ -42,7 +42,7 @@
         "web3==5.5.1",
         "asyncio==3.4.3",
         "pyyaml==5.3.1",
-        "sgx.py==0.4.dev6",
+        "sgx.py==0.5dev0",
     ],
 
     python_requires='>=3.6,<4',

skale/wallets/ledger_wallet.py

@@ -139,6 +139,9 @@ def sign_and_send(self, tx) -> str:
         signed_tx = self.sign(tx)
         return self._web3.eth.sendRawTransaction(signed_tx.rawTransaction).hex()
 
+    def sign_hash(self, unsigned_hash: str):
+        raise NotImplementedError('sign_hash is not implemented for hardware wallet')
+
     @classmethod
     def parse_derive_result(cls, exchange_result):
         pk_len = exchange_result[0]

skale/wallets/sgx_wallet.py

@@ -41,6 +41,9 @@ def sign_and_send(self, tx_dict) -> str:
         signed_tx = self.sign(tx_dict)
         return self._web3.eth.sendRawTransaction(signed_tx.rawTransaction).hex()
 
+    def sign_hash(self, unsigned_hash: str):
+        return self.sgx_client.sign_hash(unsigned_hash, self._key_name, self._web3.eth.chainId)
+
     @property
     def address(self):
         return self._address

tests/wallets/sgx_test.py

@@ -43,6 +43,15 @@ def sign(self, transaction_dict, key_name):
             'v': 37,
         })
 
+    def sign_hash(self, message, key_name, chain_id):
+        return AttributeDict({
+            'messageHash': HexBytes('0x0'),
+            'r': 123,
+            's': 123,
+            'v': 27,
+            'signature': HexBytes('0x0')
+        })
+
 
 def test_sgx_sign():
     with mock.patch('skale.wallets.sgx_wallet.SgxClient',
@@ -113,6 +122,16 @@ def test_sgx_sign_with_key():
         wallet.sign(tx_dict)
 
 
+def test_sgx_sign_hash():
+    with mock.patch('skale.wallets.sgx_wallet.SgxClient',
+                    new=SgxClient):
+        web3 = init_web3(ENDPOINT)
+        wallet = SgxWallet('TEST_ENDPOINT', web3, key_name='TEST_KEY')
+        unsigned_hash = '0x0'
+        signed_message = wallet.sign_hash(unsigned_hash)
+        assert signed_message.signature == HexBytes('0x0')
+
+
 def test_sgx_key_init():
     with mock.patch('skale.wallets.sgx_wallet.SgxClient',
                     new=SgxClient):