tests: Add device tests for generation of proofs of ownership.

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 <https://www.gnu.org/licenses/lgpl-3.0.html>.
+
+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"
+    )

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",