d-bis/app-ethereum
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: erc20information simple test and refacto plugin class

Browse Source
This commit is contained in:
Coline
2022-07-04 10:38:32 +02:00
parent 9b4d6504f1
commit 87eaaeeb5b
6 changed files with 100 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/speculos/ethereum_client/ethereum_cmd.py
View File

@@ -8,7 +8,7 @@ from speculos.client import SpeculosClient, ApduException
from ethereum_client.ethereum_cmd_builder import EthereumCommandBuilder, InsType
from ethereum_client.exception import DeviceException
from ethereum_client.transaction import PersonalTransaction, Transaction
from ethereum_client.plugin import Plugin
from ethereum_client.plugin import ERC20_Information, Plugin
from ethereum_client.utils import parse_sign_response
@@ -42,20 +42,30 @@ class EthereumCommand:
def set_plugin(self, plugin: Plugin):
try:
response = self.client._apdu_exchange(
self.client._apdu_exchange(
self.builder.set_plugin(plugin=plugin)
)
except ApduException as error:
raise DeviceException(error_code=error.sw, ins=InsType.INS_SET_PLUGIN)
def provide_nft_information(self, plugin: Plugin):
try:
response = self.client._apdu_exchange(
self.client._apdu_exchange(
self.builder.provide_nft_information(plugin=plugin)
)
except ApduException as error:
raise DeviceException(error_code=error.sw, ins=InsType.INS_SET_PLUGIN)
raise DeviceException(error_code=error.sw, ins=InsType.INS_PROVIDE_NFT_INFORMATION)
def provide_erc20_token_information(self, info: ERC20_Information):
try:
self.client._apdu_exchange(
self.builder.provide_erc20_token_information(info=info)
)
except ApduException as error:
raise DeviceException(error_code=error.sw, ins=InsType.INS_PROVIDE_ERC20)
@contextmanager