arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-08-02 03:48:58 +00:00
Code Issues Releases Wiki Activity

fix(tests): fix extra ETH unknown token dialog for T1

Browse Source
This commit is contained in:
Tomas Susanka 2020-11-20 17:49:46 +01:00
parent fe55f0c1f0
commit bd0b91a229
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/device_tests/test_msg_ethereum_signtx.py
View File

@ -117,15 +117,19 @@ class TestMsgEthereumSigntx:
@pytest.mark.setup_client(mnemonic=MNEMONIC12) @pytest.mark.setup_client(mnemonic=MNEMONIC12)
def test_ethereum_signtx_unknown_erc20_token(self, client): def test_ethereum_signtx_unknown_erc20_token(self, client):
with client: with client:
client.set_expected_responses( expected_responses = [
[
messages.ButtonRequest(code=messages.ButtonRequestType.SignTx), messages.ButtonRequest(code=messages.ButtonRequestType.SignTx),
messages.ButtonRequest(code=messages.ButtonRequestType.SignTx), messages.ButtonRequest(code=messages.ButtonRequestType.SignTx),
messages.ButtonRequest(code=messages.ButtonRequestType.SignTx),
messages.EthereumTxRequest(data_length=None),
] ]
# TT asks for contract address confirmation
if client.features.model == "T":
expected_responses.append(
messages.ButtonRequest(code=messages.ButtonRequestType.SignTx)
) )
expected_responses.append(messages.EthereumTxRequest(data_length=None))
client.set_expected_responses(expected_responses)
data = bytearray() data = bytearray()
# method id signalizing `transfer(address _to, uint256 _value)` function # method id signalizing `transfer(address _to, uint256 _value)` function
data.extend(bytes.fromhex("a9059cbb")) data.extend(bytes.fromhex("a9059cbb"))