From a351614e3a3b1c2960790666f7d2a304074ad6df Mon Sep 17 00:00:00 2001
From: obrusvit <vit.obrusnik@gmail.com>
Date: Fri, 26 Jan 2024 15:39:39 +0100
Subject: [PATCH] chore(tests): fix CI failures

Disable some tests of unsupported coins for TS3.

[no changelog]
---
 core/tests/test_apps.bitcoin.signtx_decred.py | 108 +++++++++---------
 core/tests/test_apps.common.coins.py          |   9 +-
 core/tests/test_apps.nem.hdnode.py            |   3 +-
 3 files changed, 65 insertions(+), 55 deletions(-)

diff --git a/core/tests/test_apps.bitcoin.signtx_decred.py b/core/tests/test_apps.bitcoin.signtx_decred.py
index 1766ee5df1..001141b015 100644
--- a/core/tests/test_apps.bitcoin.signtx_decred.py
+++ b/core/tests/test_apps.bitcoin.signtx_decred.py
@@ -1,60 +1,63 @@
 from common import *  # isort:skip
 
-from trezor.crypto import bip39
-from trezor.enums import AmountUnit, OutputScriptType
-from trezor.enums.RequestType import TXFINISHED, TXINPUT, TXMETA, TXOUTPUT
-from trezor.messages import (
-    PrevInput,
-    PrevOutput,
-    PrevTx,
-    SignTx,
-    TxAckInput,
-    TxAckInputWrapper,
-    TxAckOutput,
-    TxAckOutputWrapper,
-    TxAckPrevInput,
-    TxAckPrevInputWrapper,
-    TxAckPrevMeta,
-    TxAckPrevOutput,
-    TxAckPrevOutputWrapper,
-    TxInput,
-    TxOutput,
-    TxRequest,
-    TxRequestDetailsType,
-    TxRequestSerializedType,
-)
-from trezor.utils import chunks
+if not utils.MODEL_IS_T2B1:
+    from trezor.crypto import bip39
+    from trezor.enums import AmountUnit, OutputScriptType
+    from trezor.enums.RequestType import TXFINISHED, TXINPUT, TXMETA, TXOUTPUT
+    from trezor.messages import (
+        PrevInput,
+        PrevOutput,
+        PrevTx,
+        SignTx,
+        TxAckInput,
+        TxAckInputWrapper,
+        TxAckOutput,
+        TxAckOutputWrapper,
+        TxAckPrevInput,
+        TxAckPrevInputWrapper,
+        TxAckPrevMeta,
+        TxAckPrevOutput,
+        TxAckPrevOutputWrapper,
+        TxInput,
+        TxOutput,
+        TxRequest,
+        TxRequestDetailsType,
+        TxRequestSerializedType,
+    )
+    from trezor.utils import chunks
 
-from apps.bitcoin.keychain import _get_schemas_for_coin
-from apps.bitcoin.sign_tx import decred, helpers
-from apps.common import coins
-from apps.common.keychain import Keychain
+    from apps.bitcoin.keychain import _get_schemas_for_coin
+    from apps.bitcoin.sign_tx import decred, helpers
+    from apps.common import coins
+    from apps.common.keychain import Keychain
 
-EMPTY_SERIALIZED = TxRequestSerializedType(serialized_tx=bytearray())
+    EMPTY_SERIALIZED = TxRequestSerializedType(serialized_tx=bytearray())
 
-coin_decred = coins.by_name("Decred Testnet")
+    coin_decred = coins.by_name("Decred Testnet")
 
-ptx1 = PrevTx(version=1, lock_time=0, inputs_count=1, outputs_count=2, extra_data_len=0)
-pinp1 = PrevInput(
-    script_sig=unhexlify(
-        "47304402207d127d59a44187952d9d0de94ad34a19dd9a84beb124fd8a3fb439c862544d3202206618f321385c30bda96fb01ce03f70a269d78a301c0b0c2e3e3689dfae3f4733012102ae1f6b51086bd753f072f94eb8ffe6806d3570c088a3ede46c678b6ea47d1675"
-    ),
-    prev_hash=unhexlify(
-        "21012b08c5077036460e8f75bbc57beb11d7bc30e7ad224ad5e67d15bd086500"
-    ),
-    prev_index=2,
-    sequence=0xFFFF_FFFF,
-)
-pout1 = PrevOutput(
-    script_pubkey=unhexlify("76a914e4111051ae0349ab5589cf2b7e125c6da694a1a188ac"),
-    amount=153_185_001,
-    decred_script_version=0,
-)
-pout2 = PrevOutput(
-    script_pubkey=unhexlify("76a914dc1a98d791735eb9a8715a2a219c23680edcedad88ac"),
-    amount=200_000_000,
-    decred_script_version=0,
-)
+    ptx1 = PrevTx(
+        version=1, lock_time=0, inputs_count=1, outputs_count=2, extra_data_len=0
+    )
+    pinp1 = PrevInput(
+        script_sig=unhexlify(
+            "47304402207d127d59a44187952d9d0de94ad34a19dd9a84beb124fd8a3fb439c862544d3202206618f321385c30bda96fb01ce03f70a269d78a301c0b0c2e3e3689dfae3f4733012102ae1f6b51086bd753f072f94eb8ffe6806d3570c088a3ede46c678b6ea47d1675"
+        ),
+        prev_hash=unhexlify(
+            "21012b08c5077036460e8f75bbc57beb11d7bc30e7ad224ad5e67d15bd086500"
+        ),
+        prev_index=2,
+        sequence=0xFFFF_FFFF,
+    )
+    pout1 = PrevOutput(
+        script_pubkey=unhexlify("76a914e4111051ae0349ab5589cf2b7e125c6da694a1a188ac"),
+        amount=153_185_001,
+        decred_script_version=0,
+    )
+    pout2 = PrevOutput(
+        script_pubkey=unhexlify("76a914dc1a98d791735eb9a8715a2a219c23680edcedad88ac"),
+        amount=200_000_000,
+        decred_script_version=0,
+    )
 
 
 @unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
@@ -402,4 +405,5 @@ class TestSignTxDecred(unittest.TestCase):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    if not utils.MODEL_IS_T2B1:
+        unittest.main()
diff --git a/core/tests/test_apps.common.coins.py b/core/tests/test_apps.common.coins.py
index 553a2566f0..3c92b39c51 100644
--- a/core/tests/test_apps.common.coins.py
+++ b/core/tests/test_apps.common.coins.py
@@ -18,12 +18,17 @@ class TestCoins(unittest.TestCase):
     @unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
     def test_altcoins(self):
         ref = [
-            ("NMC", "Namecoin", 52),
             ("LTC", "Litecoin", 48),
-            ("DASH", "Dash", 76),
             ("ZEC", "Zcash", 7352),
             ("TAZ", "Zcash Testnet", 7461),
         ]
+        if not utils.MODEL_IS_T2B1:
+            ref.extend(
+                [
+                    ("NMC", "Namecoin", 52),
+                    ("DASH", "Dash", 76),
+                ]
+            )
         for s, n, a in ref:
             c = coins.by_name(n)
             self.assertEqual(c.address_type, a)
diff --git a/core/tests/test_apps.nem.hdnode.py b/core/tests/test_apps.nem.hdnode.py
index 7e304fbe37..37fabae59b 100644
--- a/core/tests/test_apps.nem.hdnode.py
+++ b/core/tests/test_apps.nem.hdnode.py
@@ -319,4 +319,5 @@ class TestNemHDNode(unittest.TestCase):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    if not utils.MODEL_IS_T2B1:
+        unittest.main()