mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-29 01:28:10 +00:00
110 lines
4.3 KiB
Python
110 lines
4.3 KiB
Python
from common import *
|
|
from apps.common import HARDENED
|
|
from trezor.messages.CardanoTxInputType import CardanoTxInputType
|
|
|
|
if not utils.BITCOIN_ONLY:
|
|
from apps.cardano.sign_tx import _should_hide_output
|
|
|
|
|
|
@unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
|
|
class TestCardanoSignTransaction(unittest.TestCase):
|
|
def test_should_show_outputs(self):
|
|
outputs_to_hide = [
|
|
# output is from the same address as input
|
|
(
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
[[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0]],
|
|
),
|
|
# output is from the same account but from different addresses
|
|
(
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
[
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 1],
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 2],
|
|
],
|
|
),
|
|
# both output and input are from account 2
|
|
(
|
|
[44 | HARDENED, 1815 | HARDENED, 2 | HARDENED, 0, 0],
|
|
[
|
|
[44 | HARDENED, 1815 | HARDENED, 2 | HARDENED, 0, 0],
|
|
[44 | HARDENED, 1815 | HARDENED, 2 | HARDENED, 0, 1],
|
|
[44 | HARDENED, 1815 | HARDENED, 2 | HARDENED, 0, 2],
|
|
],
|
|
),
|
|
# byron input and shelley output
|
|
(
|
|
[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
[
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
],
|
|
),
|
|
# mixed byron and shelley inputs
|
|
(
|
|
[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
[
|
|
[1852 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
],
|
|
),
|
|
]
|
|
outputs_to_show = [
|
|
# output is from different account
|
|
(
|
|
[44 | HARDENED, 1815 | HARDENED, 2 | HARDENED, 0, 0],
|
|
[[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0]],
|
|
),
|
|
# output path is not complete
|
|
(
|
|
[44 | HARDENED, 1815 | HARDENED],
|
|
[[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0]],
|
|
),
|
|
# output path is not complete
|
|
(
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED],
|
|
[[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0]],
|
|
),
|
|
# one of the inputs has different account than output
|
|
(
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
[
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0],
|
|
[44 | HARDENED, 1815 | HARDENED, 2 | HARDENED, 0, 0],
|
|
],
|
|
),
|
|
# staking output path
|
|
(
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 2, 0,],
|
|
[[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0]],
|
|
),
|
|
# output address too large
|
|
(
|
|
[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 1000001],
|
|
[[44 | HARDENED, 1815 | HARDENED, 0 | HARDENED, 0, 0]],
|
|
),
|
|
# max safe account number exceeded
|
|
(
|
|
[1852 | HARDENED, 1815 | HARDENED, 101 | HARDENED, 0, 0],
|
|
[
|
|
[1852 | HARDENED, 1815 | HARDENED, 101 | HARDENED, 0, 0]
|
|
],
|
|
),
|
|
]
|
|
|
|
for output_path, input_paths in outputs_to_hide:
|
|
inputs = [
|
|
CardanoTxInputType(address_n=input_path, prev_hash=b"", prev_index=0) for input_path in input_paths
|
|
]
|
|
self.assertTrue(_should_hide_output(output_path, inputs))
|
|
|
|
for output_path, input_paths in outputs_to_show:
|
|
inputs = [
|
|
CardanoTxInputType(address_n=input_path, prev_hash=b"", prev_index=0) for input_path in input_paths
|
|
]
|
|
self.assertFalse(_should_hide_output(output_path, inputs))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|