mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-23 07:58:09 +00:00
cardano: clean up test case
This commit is contained in:
matejcik
parent
81c55c1c5f
commit
ca345e9766
@ -14,203 +14,100 @@
|
||||
# You should have received a copy of the License along with this library.
|
||||
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
|
||||
|
||||
import time
|
||||
|
||||
import pytest
|
||||
|
||||
from trezorlib import messages
|
||||
from trezorlib.cardano import create_input, create_output
|
||||
from trezorlib import cardano, messages
|
||||
|
||||
from .common import TrezorTest
|
||||
from .conftest import setup_client
|
||||
|
||||
VECTORS = [
|
||||
# Mainnet transaction
|
||||
(
|
||||
# network
|
||||
2,
|
||||
# inputs
|
||||
[
|
||||
{
|
||||
"path": "m/44'/1815'/0'/0/1",
|
||||
"prev_hash": "1af8fa0b754ff99253d983894e63a2b09cbb56c833ba18c3384210163f63dcfc",
|
||||
"prev_index": 0,
|
||||
"type": 0,
|
||||
}
|
||||
],
|
||||
# outputs
|
||||
[
|
||||
{
|
||||
"address": "Ae2tdPwUPEZCanmBz5g2GEwFqKTKpNJcGYPKfDxoNeKZ8bRHr8366kseiK2",
|
||||
"amount": "3003112",
|
||||
}
|
||||
],
|
||||
# transactions
|
||||
[
|
||||
"839f8200d818582482582008abb575fac4c39d5bf80683f7f0c37e48f4e3d96e37d1f6611919a7241b456600ff9f8282d818582183581cda4da43db3fca93695e71dab839e72271204d28b9d964d306b8800a8a0001a7a6916a51a00305becffa0"
|
||||
],
|
||||
# tx hash
|
||||
"799c65e8a2c0b1dc4232611728c09d3f3eb0d811c077f8e9798f84605ef1b23d",
|
||||
# tx body
|
||||
"82839f8200d81858248258201af8fa0b754ff99253d983894e63a2b09cbb56c833ba18c3384210163f63dcfc00ff9f8282d818582183581c9e1c71de652ec8b85fec296f0685ca3988781c94a2e1a5d89d92f45fa0001a0d0c25611a002dd2e8ffa0818200d818588582584089053545a6c254b0d9b1464e48d2b5fcf91d4e25c128afb1fcfc61d0843338ea26308151516f3b0e02bb1638142747863c520273ce9bd3e5cd91e1d46fe2a6355840312c01c27317415b0b8acc86aa789da877fe7e15c65b7ea4c4565d8739117f5f6d9d38bf5d058f7be809b2b9b06c1d79fc6b20f9a4d76d8c89bae333edf5680c",
|
||||
),
|
||||
# Testnet transaction
|
||||
(
|
||||
# network
|
||||
1,
|
||||
# inputs
|
||||
[
|
||||
{
|
||||
"path": "m/44'/1815'/0'/0/1",
|
||||
"prev_hash": "1af8fa0b754ff99253d983894e63a2b09cbb56c833ba18c3384210163f63dcfc",
|
||||
"prev_index": 0,
|
||||
"type": 0,
|
||||
}
|
||||
],
|
||||
# outputs
|
||||
[
|
||||
{
|
||||
"address": "Ae2tdPwUPEZCanmBz5g2GEwFqKTKpNJcGYPKfDxoNeKZ8bRHr8366kseiK2",
|
||||
"amount": "3003112",
|
||||
}
|
||||
],
|
||||
# transactions
|
||||
[
|
||||
"839f8200d818582482582008abb575fac4c39d5bf80683f7f0c37e48f4e3d96e37d1f6611919a7241b456600ff9f8282d818582183581cda4da43db3fca93695e71dab839e72271204d28b9d964d306b8800a8a0001a7a6916a51a00305becffa0"
|
||||
],
|
||||
# tx hash
|
||||
"799c65e8a2c0b1dc4232611728c09d3f3eb0d811c077f8e9798f84605ef1b23d",
|
||||
# tx body
|
||||
"82839f8200d81858248258201af8fa0b754ff99253d983894e63a2b09cbb56c833ba18c3384210163f63dcfc00ff9f8282d818582183581c9e1c71de652ec8b85fec296f0685ca3988781c94a2e1a5d89d92f45fa0001a0d0c25611a002dd2e8ffa0818200d818588582584089053545a6c254b0d9b1464e48d2b5fcf91d4e25c128afb1fcfc61d0843338ea26308151516f3b0e02bb1638142747863c520273ce9bd3e5cd91e1d46fe2a63558403594ee7e2bfe4c84f886a8336cecb7c42983ce9a057345ebb6294a436087d8db93ca78cf514c7c48edff4c8435f690a5817951e2b55d2db729875ee7cc0f7d08",
|
||||
),
|
||||
]
|
||||
|
||||
|
||||
@pytest.mark.cardano
|
||||
@pytest.mark.skip_t1 # T1 support is not planned
|
||||
class TestMsgCardanoSignTx(TrezorTest):
|
||||
def test_cardano_sign_tx_mainnet(self):
|
||||
self.setup_mnemonic_allallall()
|
||||
@setup_client()
|
||||
@pytest.mark.parametrize("network,inputs,outputs,transactions,tx_hash,tx_body", VECTORS)
|
||||
def test_cardano_sign_tx(
|
||||
client, network, inputs, outputs, transactions, tx_hash, tx_body
|
||||
):
|
||||
inputs = [cardano.create_input(i) for i in inputs]
|
||||
outputs = [cardano.create_output(o) for o in outputs]
|
||||
|
||||
transaction = {
|
||||
"inputs": [
|
||||
{
|
||||
"path": "m/44'/1815'/0'/0/1",
|
||||
"prev_hash": "1af8fa0b754ff99253d983894e63a2b09cbb56c833ba18c3384210163f63dcfc",
|
||||
"prev_index": 0,
|
||||
"type": 0,
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"address": "Ae2tdPwUPEZCanmBz5g2GEwFqKTKpNJcGYPKfDxoNeKZ8bRHr8366kseiK2",
|
||||
"amount": "3003112",
|
||||
}
|
||||
],
|
||||
"transactions": [
|
||||
"839f8200d818582482582008abb575fac4c39d5bf80683f7f0c37e48f4e3d96e37d1f6611919a7241b456600ff9f8282d818582183581cda4da43db3fca93695e71dab839e72271204d28b9d964d306b8800a8a0001a7a6916a51a00305becffa0"
|
||||
],
|
||||
}
|
||||
expected_responses = [
|
||||
messages.CardanoTxRequest(tx_index=i) for i in range(len(transactions))
|
||||
]
|
||||
expected_responses += [
|
||||
messages.ButtonRequest(code=messages.ButtonRequestType.Other),
|
||||
messages.ButtonRequest(code=messages.ButtonRequestType.Other),
|
||||
messages.ButtonRequest(code=messages.ButtonRequestType.Other),
|
||||
messages.ButtonRequest(code=messages.ButtonRequestType.Other),
|
||||
messages.CardanoSignedTx(),
|
||||
]
|
||||
|
||||
inputs = [create_input(input) for input in transaction["inputs"]]
|
||||
outputs = [create_output(output) for output in transaction["outputs"]]
|
||||
transactions = transaction["transactions"]
|
||||
|
||||
self.client.transport.write(
|
||||
messages.CardanoSignTx(
|
||||
inputs=inputs,
|
||||
outputs=outputs,
|
||||
transactions_count=len(transactions),
|
||||
network=2,
|
||||
)
|
||||
)
|
||||
response = self.client.transport.read()
|
||||
|
||||
assert isinstance(response, messages.CardanoTxRequest)
|
||||
assert response.tx_index == 0
|
||||
|
||||
# Upload first transaction
|
||||
transaction_data = bytes.fromhex(transactions[0])
|
||||
ack_message = messages.CardanoTxAck(transaction=transaction_data)
|
||||
self.client.transport.write(ack_message)
|
||||
|
||||
# Confirm fee
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.ButtonRequest)
|
||||
assert response.code == messages.ButtonRequestType.Other
|
||||
|
||||
self.client.debug.press_yes()
|
||||
self.client.transport.write(messages.ButtonAck())
|
||||
time.sleep(1)
|
||||
|
||||
# Confirm Network
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.ButtonRequest)
|
||||
assert response.code == messages.ButtonRequestType.Other
|
||||
|
||||
self.client.debug.press_yes()
|
||||
self.client.transport.write(messages.ButtonAck())
|
||||
time.sleep(1)
|
||||
|
||||
# Confirm Output
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.ButtonRequest)
|
||||
assert response.code == messages.ButtonRequestType.Other
|
||||
|
||||
self.client.debug.press_yes()
|
||||
self.client.transport.write(messages.ButtonAck())
|
||||
time.sleep(1)
|
||||
self.client.debug.swipe_down()
|
||||
time.sleep(1)
|
||||
|
||||
# Confirm amount
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.ButtonRequest)
|
||||
assert response.code == messages.ButtonRequestType.Other
|
||||
|
||||
self.client.debug.press_yes()
|
||||
self.client.transport.write(messages.ButtonAck())
|
||||
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.CardanoSignedTx)
|
||||
|
||||
assert (
|
||||
response.tx_hash.hex()
|
||||
== "799c65e8a2c0b1dc4232611728c09d3f3eb0d811c077f8e9798f84605ef1b23d"
|
||||
)
|
||||
assert (
|
||||
response.tx_body.hex()
|
||||
== "82839f8200d81858248258201af8fa0b754ff99253d983894e63a2b09cbb56c833ba18c3384210163f63dcfc00ff9f8282d818582183581c9e1c71de652ec8b85fec296f0685ca3988781c94a2e1a5d89d92f45fa0001a0d0c25611a002dd2e8ffa0818200d818588582584089053545a6c254b0d9b1464e48d2b5fcf91d4e25c128afb1fcfc61d0843338ea26308151516f3b0e02bb1638142747863c520273ce9bd3e5cd91e1d46fe2a6355840312c01c27317415b0b8acc86aa789da877fe7e15c65b7ea4c4565d8739117f5f6d9d38bf5d058f7be809b2b9b06c1d79fc6b20f9a4d76d8c89bae333edf5680c"
|
||||
)
|
||||
|
||||
def test_cardano_sign_tx_testnet(self):
|
||||
self.setup_mnemonic_allallall()
|
||||
|
||||
transaction = {
|
||||
"inputs": [
|
||||
{
|
||||
"path": "m/44'/1815'/0'/0/1",
|
||||
"prev_hash": "1af8fa0b754ff99253d983894e63a2b09cbb56c833ba18c3384210163f63dcfc",
|
||||
"prev_index": 0,
|
||||
"type": 0,
|
||||
}
|
||||
],
|
||||
"outputs": [
|
||||
{
|
||||
"address": "Ae2tdPwUPEZCanmBz5g2GEwFqKTKpNJcGYPKfDxoNeKZ8bRHr8366kseiK2",
|
||||
"amount": "3003112",
|
||||
}
|
||||
],
|
||||
"transactions": [
|
||||
"839f8200d818582482582008abb575fac4c39d5bf80683f7f0c37e48f4e3d96e37d1f6611919a7241b456600ff9f8282d818582183581cda4da43db3fca93695e71dab839e72271204d28b9d964d306b8800a8a0001a7a6916a51a00305becffa0"
|
||||
],
|
||||
}
|
||||
|
||||
inputs = [create_input(input) for input in transaction["inputs"]]
|
||||
outputs = [create_output(output) for output in transaction["outputs"]]
|
||||
transactions = transaction["transactions"]
|
||||
|
||||
self.client.transport.write(
|
||||
messages.CardanoSignTx(
|
||||
inputs=inputs,
|
||||
outputs=outputs,
|
||||
transactions_count=len(transactions),
|
||||
network=1,
|
||||
)
|
||||
)
|
||||
response = self.client.transport.read()
|
||||
|
||||
assert isinstance(response, messages.CardanoTxRequest)
|
||||
assert response.tx_index == 0
|
||||
|
||||
# Upload first transaction
|
||||
transaction_data = bytes.fromhex(transactions[0])
|
||||
ack_message = messages.CardanoTxAck(transaction=transaction_data)
|
||||
self.client.transport.write(ack_message)
|
||||
|
||||
# Confirm fee
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.ButtonRequest)
|
||||
assert response.code == messages.ButtonRequestType.Other
|
||||
|
||||
self.client.debug.press_yes()
|
||||
self.client.transport.write(messages.ButtonAck())
|
||||
time.sleep(1)
|
||||
|
||||
# Confirm Network
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.ButtonRequest)
|
||||
assert response.code == messages.ButtonRequestType.Other
|
||||
|
||||
self.client.debug.press_yes()
|
||||
self.client.transport.write(messages.ButtonAck())
|
||||
time.sleep(1)
|
||||
|
||||
# Confirm Output
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.ButtonRequest)
|
||||
assert response.code == messages.ButtonRequestType.Other
|
||||
|
||||
self.client.debug.press_yes()
|
||||
self.client.transport.write(messages.ButtonAck())
|
||||
time.sleep(1)
|
||||
self.client.debug.swipe_down()
|
||||
time.sleep(1)
|
||||
|
||||
# Confirm amount
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.ButtonRequest)
|
||||
assert response.code == messages.ButtonRequestType.Other
|
||||
|
||||
self.client.debug.press_yes()
|
||||
self.client.transport.write(messages.ButtonAck())
|
||||
|
||||
response = self.client.transport.read()
|
||||
assert isinstance(response, messages.CardanoSignedTx)
|
||||
|
||||
assert (
|
||||
response.tx_hash.hex()
|
||||
== "799c65e8a2c0b1dc4232611728c09d3f3eb0d811c077f8e9798f84605ef1b23d"
|
||||
)
|
||||
assert (
|
||||
response.tx_body.hex()
|
||||
== "82839f8200d81858248258201af8fa0b754ff99253d983894e63a2b09cbb56c833ba18c3384210163f63dcfc00ff9f8282d818582183581c9e1c71de652ec8b85fec296f0685ca3988781c94a2e1a5d89d92f45fa0001a0d0c25611a002dd2e8ffa0818200d818588582584089053545a6c254b0d9b1464e48d2b5fcf91d4e25c128afb1fcfc61d0843338ea26308151516f3b0e02bb1638142747863c520273ce9bd3e5cd91e1d46fe2a63558403594ee7e2bfe4c84f886a8336cecb7c42983ce9a057345ebb6294a436087d8db93ca78cf514c7c48edff4c8435f690a5817951e2b55d2db729875ee7cc0f7d08"
|
||||
with client:
|
||||
client.set_expected_responses(expected_responses)
|
||||
response = cardano.sign_tx(
|
||||
client, inputs, outputs, transactions, network=network
|
||||
)
|
||||
assert response.tx_hash.hex() == tx_hash
|
||||
assert response.tx_body.hex() == tx_body
|
||||
|
||||
Loading…
Reference in New Issue
Block a user