diff --git a/tests/device_tests/test_msg_getownershipproof.py b/tests/device_tests/test_msg_getownershipproof.py new file mode 100644 index 0000000000..5757e2865f --- /dev/null +++ b/tests/device_tests/test_msg_getownershipproof.py @@ -0,0 +1,96 @@ +# 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 . + +import pytest + +from trezorlib import btc, messages +from trezorlib.exceptions import TrezorFailure +from trezorlib.tools import parse_path + +pytestmark = pytest.mark.skip_t1 + + +@pytest.mark.skip_ui +def test_ownership_id(client): + ownership_id = btc.get_ownership_id( + client, + "Bitcoin", + parse_path("m/84'/0'/0'/1/0"), + script_type=messages.InputScriptType.SPENDWITNESS, + ) + assert ( + ownership_id.hex() + == "a122407efc198211c81af4450f40b235d54775efd934d16b9e31c6ce9bad5707" + ) + + +@pytest.mark.skip_ui +def test_p2wpkh_ownership_proof(client): + ownership_proof, _ = btc.get_ownership_proof( + client, + "Bitcoin", + parse_path("m/84'/0'/0'/1/0"), + script_type=messages.InputScriptType.SPENDWITNESS, + ) + assert ( + ownership_proof.hex() + == "534c00190001a122407efc198211c81af4450f40b235d54775efd934d16b9e31c6ce9bad57070002483045022100e5eaf2cb0a473b4545115c7b85323809e75cb106175ace38129fd62323d73df30220363dbc7acb7afcda022b1f8d97acb8f47c42043cfe0595583aa26e30bc8b3bb50121032ef68318c8f6aaa0adec0199c69901f0db7d3485eb38d9ad235221dc3d61154b" + ) + + +@pytest.mark.skip_ui +def test_fake_ownership_id(client): + with pytest.raises(TrezorFailure, match="Invalid ownership identifier"): + btc.get_ownership_proof( + client, + "Bitcoin", + parse_path("m/84'/0'/0'/1/0"), + script_type=messages.InputScriptType.SPENDWITNESS, + ownership_ids=[ + b"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f" + ], + ) + + +def test_confirm_ownership_proof(client): + ownership_proof, _ = btc.get_ownership_proof( + client, + "Bitcoin", + parse_path("m/84'/0'/0'/1/0"), + script_type=messages.InputScriptType.SPENDWITNESS, + user_confirmation=True, + ) + + assert ( + ownership_proof.hex() + == "534c00190101a122407efc198211c81af4450f40b235d54775efd934d16b9e31c6ce9bad57070002483045022100fa4561d261464360f18af9668c19e1b07d58f76814623aa8e5c9b2d85bc211d002207f364096183f461be75f763d2f970df6ac9b30a5a39556a07cab9c6f4d7883a80121032ef68318c8f6aaa0adec0199c69901f0db7d3485eb38d9ad235221dc3d61154b" + ) + + +def test_confirm_ownership_proof_with_data(client): + ownership_proof, _ = btc.get_ownership_proof( + client, + "Bitcoin", + parse_path("m/84'/0'/0'/1/0"), + script_type=messages.InputScriptType.SPENDWITNESS, + user_confirmation=True, + commitment_data=b"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f", + ) + + assert ( + ownership_proof.hex() + == "534c00190101a122407efc198211c81af4450f40b235d54775efd934d16b9e31c6ce9bad57070002483045022100ac67fde3f0c8443c2708b3e405b17c4c8a51e510132b4f35aa6d6782713a53280220616192365f6202ee3f050d4e0e13c38198156024ca978fbd2b8c89c8823bb3dd0121032ef68318c8f6aaa0adec0199c69901f0db7d3485eb38d9ad235221dc3d61154b" + ) diff --git a/tests/ui_tests/fixtures.json b/tests/ui_tests/fixtures.json index ea5a3ede78..897538e865 100644 --- a/tests/ui_tests/fixtures.json +++ b/tests/ui_tests/fixtures.json @@ -122,6 +122,8 @@ "test_msg_getentropy.py::test_entropy[65]": "a722fa2048fa3102889ec05558d25f837a364ef2a118e85975683e10a56f1356", "test_msg_getentropy.py::test_entropy[8]": "a722fa2048fa3102889ec05558d25f837a364ef2a118e85975683e10a56f1356", "test_msg_getentropy.py::test_entropy[9]": "a722fa2048fa3102889ec05558d25f837a364ef2a118e85975683e10a56f1356", +"test_msg_getownershipproof.py::test_confirm_ownership_proof": "7724ef59fee121da564b935b5880479c9518d97dc286e2949f13a8a8bdf6fa4a", +"test_msg_getownershipproof.py::test_confirm_ownership_proof_with_data": "1043f54be594bd3180c59c8474ef7ef46be35f99ad5f6890949c38680c097e52", "test_msg_lisk_getaddress.py-test_lisk_getaddress": "0063ceb48d21aecd1ddabdb083c8afd2042cdf577e4751fa3f57b2b80f619084", "test_msg_lisk_getpublickey.py-test_lisk_get_public_key": "e4cb8c7430c240e27a2211391ab5eba848be4f50136cf9f693142c2677a939d7", "test_msg_lisk_signmessage.py-test_sign": "c47a6ec147137c75903cff19da6607eaef5a1fc03ace1f840d2952744342b568",