From 93382a10f5b66cdfe133c0e2bbfd531859dacd1f Mon Sep 17 00:00:00 2001
From: Roman Zeyde <me@romanzey.de>
Date: Sat, 8 Jun 2019 17:39:12 +0300
Subject: [PATCH] tests: test Elements explicit addresses (p2pkh, p2sh and
 bech32)

---
 tests/device_tests/test_msg_getaddress.py             |  7 +++++++
 tests/device_tests/test_msg_getaddress_segwit.py      | 11 +++++++++++
 .../device_tests/test_msg_getaddress_segwit_native.py | 11 +++++++++++
 3 files changed, 29 insertions(+)

diff --git a/tests/device_tests/test_msg_getaddress.py b/tests/device_tests/test_msg_getaddress.py
index 374bdf545..3002c2096 100644
--- a/tests/device_tests/test_msg_getaddress.py
+++ b/tests/device_tests/test_msg_getaddress.py
@@ -115,6 +115,13 @@ class TestMsgGetaddress(TrezorTest):
             == "Fmhtxeh7YdCBkyQF7AQG4QnY8y3rJg89di"
         )
 
+    def test_elements(self):
+        self.setup_mnemonic_allallall()
+        assert (
+            btc.get_address(self.client, "Elements", parse_path("m/44'/1'/0'/0/0"))
+            == "2dpWh6jbhAowNsQ5agtFzi7j6nKscj6UnEr"
+        )
+
     def test_multisig(self):
         self.setup_mnemonic_allallall()
         xpubs = []
diff --git a/tests/device_tests/test_msg_getaddress_segwit.py b/tests/device_tests/test_msg_getaddress_segwit.py
index 7a1d5a41a..61670c378 100644
--- a/tests/device_tests/test_msg_getaddress_segwit.py
+++ b/tests/device_tests/test_msg_getaddress_segwit.py
@@ -78,6 +78,17 @@ class TestMsgGetaddressSegwit(TrezorTest):
             )
             == "2N4Q5FhU2497BryFfUgbqkAJE87aKDv3V3e"
         )
+        assert (
+            btc.get_address(
+                self.client,
+                "Elements",
+                parse_path("m/49'/1'/0'/0/0"),
+                False,
+                None,
+                script_type=proto.InputScriptType.SPENDP2SHWITNESS,
+            )
+            == "XNW67ZQA9K3AuXPBWvJH4zN2y5QBDTwy2Z"
+        )
 
     def test_show_multisig_3(self):
         self.setup_mnemonic_allallall()
diff --git a/tests/device_tests/test_msg_getaddress_segwit_native.py b/tests/device_tests/test_msg_getaddress_segwit_native.py
index 73a68d10d..13a4f24b5 100644
--- a/tests/device_tests/test_msg_getaddress_segwit_native.py
+++ b/tests/device_tests/test_msg_getaddress_segwit_native.py
@@ -78,6 +78,17 @@ class TestMsgGetaddressSegwitNative(TrezorTest):
             )
             == "grs1qw4teyraux2s77nhjdwh9ar8rl9dt7zww8r6lne"
         )
+        assert (
+            btc.get_address(
+                self.client,
+                "Elements",
+                parse_path("84'/1'/0'/0/0"),
+                False,
+                None,
+                script_type=proto.InputScriptType.SPENDWITNESS,
+            )
+            == "ert1qkvwu9g3k2pdxewfqr7syz89r3gj557l3xp9k2v"
+        )
 
     def test_show_multisig_3(self):
         self.setup_mnemonic_allallall()