tests: slip39 cardano tests for get_address and get_public_key

python/trezorlib/tests/device_tests/test_msg_cardano_get_address.py

@@ -20,12 +20,12 @@ from trezorlib.cardano import get_address
 from trezorlib.tools import parse_path
 
 from .common import TrezorTest
+from .conftest import setup_client
 
 
 @pytest.mark.cardano
 @pytest.mark.skip_t1  # T1 support is not planned
-class TestMsgCardanoGetAddress(TrezorTest):
+@pytest.mark.parametrize(
-    @pytest.mark.parametrize(
     "path,expected_address",
     [
         (
@@ -41,10 +41,10 @@ class TestMsgCardanoGetAddress(TrezorTest):
             "Ae2tdPwUPEZ3gZD1QeUHvAqadAV59Zid6NP9VCR9BG5LLAja9YtBUgr6ttK",
         ),
     ],
-    )
+)
-    def test_cardano_get_address(self, path, expected_address):
+@setup_client(mnemonic=TrezorTest.mnemonic12)
+def test_cardano_get_address(client, path, expected_address):
     # data from https://iancoleman.io/bip39/#english
-        self.setup_mnemonic_nopin_nopassphrase()
 
-        address = get_address(self.client, parse_path(path))
+    address = get_address(client, parse_path(path))
     assert address == expected_address

python/trezorlib/tests/device_tests/test_msg_cardano_get_address_shamir.py

@@ -0,0 +1,60 @@
+# This file is part of the Trezor project.
+#
+# Copyright (C) 2012-2019 SatoshiLabs and contributors
+#
+# This library is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License version 3
+# as published by the Free Software Foundation.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# 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 pytest
+
+from trezorlib import device
+from trezorlib.cardano import get_address
+from trezorlib.messages.PassphraseSourceType import HOST as PASSPHRASE_ON_HOST
+from trezorlib.tools import parse_path
+
+from .conftest import setup_client
+
+SLIP39_MNEMONIC = [
+    "extra extend academic bishop cricket bundle tofu goat apart victim enlarge program behavior permit course armed jerky faint language modern",
+    "extra extend academic acne away best indicate impact square oasis prospect painting voting guest either argue username racism enemy eclipse",
+    "extra extend academic arcade born dive legal hush gross briefing talent drug much home firefly toxic analysis idea umbrella slice",
+]
+
+
+@pytest.mark.cardano
+@pytest.mark.skip_t1  # T1 support is not planned
+@pytest.mark.parametrize(
+    "path,expected_address",
+    [
+        (
+            "m/44'/1815'/0'/0/0",
+            "Ae2tdPwUPEYxF9NAMNdd3v2LZoMeWp7gCZiDb6bZzFQeeVASzoP7HC4V9s6",
+        ),
+        (
+            "m/44'/1815'/0'/0/1",
+            "Ae2tdPwUPEZ1TjYcvfkWAbiHtGVxv4byEHHZoSyQXjPJ362DifCe1ykgqgy",
+        ),
+        (
+            "m/44'/1815'/0'/0/2",
+            "Ae2tdPwUPEZGXmSbda1kBNfyhRQGRcQxJFdk7mhWZXAGnapyejv2b2U3aRb",
+        ),
+    ],
+)
+@setup_client(mnemonic=SLIP39_MNEMONIC, passphrase=True)
+def test_cardano_get_address(client, path, expected_address):
+    # enter passphrase
+    assert client.debug.read_passphrase_protection() is True
+    device.apply_settings(client, passphrase_source=PASSPHRASE_ON_HOST)
+    client.set_passphrase("TREZOR")
+
+    address = get_address(client, parse_path(path))
+    assert address == expected_address

python/trezorlib/tests/device_tests/test_msg_cardano_get_public_key.py

@@ -19,13 +19,13 @@ import pytest
 from trezorlib.cardano import get_public_key
 from trezorlib.tools import parse_path
 
-from .common import TrezorTest
+from .conftest import setup_client
 
 
 @pytest.mark.cardano
 @pytest.mark.skip_t1  # T1 support is not planned
-class TestMsgCardanoGetPublicKey(TrezorTest):
+@setup_client(mnemonic=" ".join(["all"] * 12))
-    @pytest.mark.parametrize(
+@pytest.mark.parametrize(
     "path,public_key,chain_code",
     [
         (
@@ -49,11 +49,9 @@ class TestMsgCardanoGetPublicKey(TrezorTest):
             "1197ea234f528987cbac9817ebc31344395b837a3bb7c2332f87e095e70550a5",
         ),
     ],
-    )
+)
-    def test_cardano_get_public_key(self, path, public_key, chain_code):
+def test_cardano_get_public_key(client, path, public_key, chain_code):
-        self.setup_mnemonic_allallall()
+    key = get_public_key(client, parse_path(path))
-
-        key = get_public_key(self.client, parse_path(path))
 
     assert key.node.public_key.hex() == public_key
     assert key.node.chain_code.hex() == chain_code

python/trezorlib/tests/device_tests/test_msg_cardano_get_public_key_shamir.py

@@ -0,0 +1,69 @@
+# This file is part of the Trezor project.
+#
+# Copyright (C) 2012-2019 SatoshiLabs and contributors
+#
+# This library is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License version 3
+# as published by the Free Software Foundation.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Lesser General Public License for more details.
+#
+# 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 pytest
+
+from trezorlib import device
+from trezorlib.cardano import get_public_key
+from trezorlib.messages.PassphraseSourceType import HOST as PASSPHRASE_ON_HOST
+from trezorlib.tools import parse_path
+
+from .conftest import setup_client
+
+SLIP39_MNEMONIC = [
+    "extra extend academic bishop cricket bundle tofu goat apart victim "
+    "enlarge program behavior permit course armed jerky faint language modern",
+    "extra extend academic acne away best indicate impact square oasis "
+    "prospect painting voting guest either argue username racism enemy eclipse",
+    "extra extend academic arcade born dive legal hush gross briefing "
+    "talent drug much home firefly toxic analysis idea umbrella slice",
+]
+
+
+@pytest.mark.cardano
+@pytest.mark.skip_t1  # T1 support is not planned
+@setup_client(mnemonic=SLIP39_MNEMONIC, passphrase=True)
+@pytest.mark.parametrize(
+    "path,public_key,chain_code",
+    [
+        (
+            "m/44'/1815'/0'/0/0",
+            "bc043d84b8b891d49890edb6aced6f2d78395f255c5b6aea8878b913f83e8579",
+            "dc3f0d2b5cccb822335ef6213fd133f4ca934151ec44a6000aee43b8a101078c",
+        ),
+        (
+            "m/44'/1815'/0'/0/1",
+            "24c4fe188a39103db88818bc191fd8571eae7b284ebcbdf2462bde97b058a95c",
+            "6f7a744035f4b3ddb8f861c18446169643cc3ae85e271b4b4f0eda05cf84c65b",
+        ),
+        (
+            "m/44'/1815'/0'/0/2",
+            "831a63d381a8dab1e6e1ee991a4300fc70687aae5f97f4fcf92ed1b6c2bd99de",
+            "672d6af4707aba201b7940231e83dd357f92f8851b3dfdc224ef311e1b64cdeb",
+        ),
+    ],
+)
+def test_cardano_get_public_key(client, path, public_key, chain_code):
+    # enter passphrase
+    assert client.debug.read_passphrase_protection() is True
+    device.apply_settings(client, passphrase_source=PASSPHRASE_ON_HOST)
+    client.set_passphrase("TREZOR")
+
+    key = get_public_key(client, parse_path(path))
+
+    assert key.node.public_key.hex() == public_key
+    assert key.node.chain_code.hex() == chain_code
+    assert key.xpub == public_key + chain_code