From 8eccfcad9b1934b92d193cdc26348ed45c72e8fb Mon Sep 17 00:00:00 2001 From: meherett Date: Thu, 5 Dec 2024 14:55:57 +0300 Subject: [PATCH] Update: BIP's, Electrum V1 and V2 HDs tests --- tests/hdwallet/hds/electrum/test_hds_electrum_v1.py | 4 +++- .../hds/electrum/v2/test_hds_electrum_v2_segwit.py | 5 ++++- .../hds/electrum/v2/test_hds_electrum_v2_segwit_2fa.py | 5 ++++- .../hds/electrum/v2/test_hds_electrum_v2_standard.py | 5 ++++- .../hds/electrum/v2/test_hds_electrum_v2_standard_2fa.py | 5 ++++- tests/hdwallet/hds/test_hds_bip141.py | 2 +- tests/hdwallet/hds/test_hds_bip32.py | 3 ++- tests/hdwallet/hds/test_hds_bip44.py | 2 +- tests/hdwallet/hds/test_hds_bip49.py | 2 +- tests/hdwallet/hds/test_hds_bip84.py | 2 +- tests/hdwallet/hds/test_hds_bip86.py | 7 ++++++- 11 files changed, 31 insertions(+), 11 deletions(-) diff --git a/tests/hdwallet/hds/electrum/test_hds_electrum_v1.py b/tests/hdwallet/hds/electrum/test_hds_electrum_v1.py index e6b1041..abb6982 100644 --- a/tests/hdwallet/hds/electrum/test_hds_electrum_v1.py +++ b/tests/hdwallet/hds/electrum/test_hds_electrum_v1.py @@ -6,6 +6,7 @@ # file COPYING or https://opensource.org/license/mit from hdwallet.hds import ElectrumV1HD +from hdwallet.cryptocurrencies import Bitcoin as Cryptocurrency from hdwallet.derivations import ElectrumDerivation from hdwallet.const import PUBLIC_KEY_TYPES @@ -13,7 +14,8 @@ def test_electrum_v1_hd(data): electrum_v1_hd: ElectrumV1HD = ElectrumV1HD( - public_key_type=PUBLIC_KEY_TYPES.COMPRESSED + public_key_type=PUBLIC_KEY_TYPES.COMPRESSED, + wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX ) electrum_v1_hd.from_seed( diff --git a/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_segwit.py b/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_segwit.py index 58fba94..823abb1 100644 --- a/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_segwit.py +++ b/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_segwit.py @@ -6,6 +6,7 @@ # file COPYING or https://opensource.org/license/mit from hdwallet.hds import ElectrumV2HD +from hdwallet.cryptocurrencies import Bitcoin as Cryptocurrency from hdwallet.derivations import ElectrumDerivation from hdwallet.const import ( PUBLIC_KEY_TYPES, MODES @@ -15,7 +16,9 @@ def test_electrum_v2_segwit_hd(data): electrum_v2_hd: ElectrumV2HD = ElectrumV2HD( - mode=MODES.SEGWIT, public_key_type=PUBLIC_KEY_TYPES.UNCOMPRESSED + mode=MODES.SEGWIT, + public_key_type=PUBLIC_KEY_TYPES.UNCOMPRESSED, + wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX ) electrum_v2_hd.from_seed( diff --git a/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_segwit_2fa.py b/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_segwit_2fa.py index 3866568..9590be4 100644 --- a/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_segwit_2fa.py +++ b/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_segwit_2fa.py @@ -6,6 +6,7 @@ # file COPYING or https://opensource.org/license/mit from hdwallet.hds import ElectrumV2HD +from hdwallet.cryptocurrencies import Bitcoin as Cryptocurrency from hdwallet.derivations import ElectrumDerivation from hdwallet.const import ( PUBLIC_KEY_TYPES, MODES @@ -15,7 +16,9 @@ def test_electrum_v2_segwit_2fa_hd(data): electrum_v2_hd: ElectrumV2HD = ElectrumV2HD( - mode=MODES.SEGWIT, public_key_type=PUBLIC_KEY_TYPES.UNCOMPRESSED + mode=MODES.SEGWIT, + wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX, + public_key_type=PUBLIC_KEY_TYPES.UNCOMPRESSED ) electrum_v2_hd.from_seed( diff --git a/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_standard.py b/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_standard.py index 2ba46f5..90edbb1 100644 --- a/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_standard.py +++ b/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_standard.py @@ -6,6 +6,7 @@ # file COPYING or https://opensource.org/license/mit from hdwallet.hds import ElectrumV2HD +from hdwallet.cryptocurrencies import Bitcoin as Cryptocurrency from hdwallet.derivations import ElectrumDerivation from hdwallet.const import ( PUBLIC_KEY_TYPES, MODES @@ -15,7 +16,9 @@ def test_electrum_v2_standard_hd(data): electrum_v2_hd: ElectrumV2HD = ElectrumV2HD( - mode=MODES.STANDARD, public_key_type=PUBLIC_KEY_TYPES.UNCOMPRESSED + mode=MODES.STANDARD, + wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX, + public_key_type=PUBLIC_KEY_TYPES.UNCOMPRESSED ) electrum_v2_hd.from_seed( diff --git a/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_standard_2fa.py b/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_standard_2fa.py index ee5d573..627d522 100644 --- a/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_standard_2fa.py +++ b/tests/hdwallet/hds/electrum/v2/test_hds_electrum_v2_standard_2fa.py @@ -6,6 +6,7 @@ # file COPYING or https://opensource.org/license/mit from hdwallet.hds import ElectrumV2HD +from hdwallet.cryptocurrencies import Bitcoin as Cryptocurrency from hdwallet.derivations import ElectrumDerivation from hdwallet.const import ( PUBLIC_KEY_TYPES, MODES @@ -15,7 +16,9 @@ def test_electrum_v2_standard_2fa_hd(data): electrum_v2_hd: ElectrumV2HD = ElectrumV2HD( - mode=MODES.STANDARD, public_key_type=PUBLIC_KEY_TYPES.UNCOMPRESSED + mode=MODES.STANDARD, + wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX, + public_key_type=PUBLIC_KEY_TYPES.UNCOMPRESSED ) electrum_v2_hd.from_seed( diff --git a/tests/hdwallet/hds/test_hds_bip141.py b/tests/hdwallet/hds/test_hds_bip141.py index 5c1de47..8e13286 100644 --- a/tests/hdwallet/hds/test_hds_bip141.py +++ b/tests/hdwallet/hds/test_hds_bip141.py @@ -12,7 +12,7 @@ def test_bip141_hd(data): bip141_hd: BIP141HD = BIP141HD( - ecc=Cryptocurrency.ECC, semantic="p2wpkh" + ecc=Cryptocurrency.ECC, wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX, semantic="p2wpkh" ) bip141_hd.from_seed( diff --git a/tests/hdwallet/hds/test_hds_bip32.py b/tests/hdwallet/hds/test_hds_bip32.py index 56205f5..f705cef 100644 --- a/tests/hdwallet/hds/test_hds_bip32.py +++ b/tests/hdwallet/hds/test_hds_bip32.py @@ -12,7 +12,8 @@ def test_bip32_hd(data): bip32_hd: BIP32HD = BIP32HD( - ecc=Cryptocurrency.ECC + ecc=Cryptocurrency.ECC, wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX + ) bip32_hd.from_seed( diff --git a/tests/hdwallet/hds/test_hds_bip44.py b/tests/hdwallet/hds/test_hds_bip44.py index de753ae..5c5db42 100644 --- a/tests/hdwallet/hds/test_hds_bip44.py +++ b/tests/hdwallet/hds/test_hds_bip44.py @@ -12,7 +12,7 @@ def test_bip44_hd(data): bip44_hd: BIP44HD = BIP44HD( - ecc=Cryptocurrency.ECC + ecc=Cryptocurrency.ECC, wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX ) bip44_hd.from_seed( diff --git a/tests/hdwallet/hds/test_hds_bip49.py b/tests/hdwallet/hds/test_hds_bip49.py index ce8c25a..3457f96 100644 --- a/tests/hdwallet/hds/test_hds_bip49.py +++ b/tests/hdwallet/hds/test_hds_bip49.py @@ -12,7 +12,7 @@ def test_bip49_hd(data): bip49_hd: BIP49HD = BIP49HD( - ecc=Cryptocurrency.ECC + ecc=Cryptocurrency.ECC, wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX ) bip49_hd.from_seed( diff --git a/tests/hdwallet/hds/test_hds_bip84.py b/tests/hdwallet/hds/test_hds_bip84.py index 809eef5..55a0b4b 100644 --- a/tests/hdwallet/hds/test_hds_bip84.py +++ b/tests/hdwallet/hds/test_hds_bip84.py @@ -12,7 +12,7 @@ def test_bip84_hd(data): bip84_hd: BIP84HD = BIP84HD( - ecc=Cryptocurrency.ECC + ecc=Cryptocurrency.ECC, wif_prefix = Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX ) bip84_hd.from_seed( diff --git a/tests/hdwallet/hds/test_hds_bip86.py b/tests/hdwallet/hds/test_hds_bip86.py index 3f51014..b28dcef 100644 --- a/tests/hdwallet/hds/test_hds_bip86.py +++ b/tests/hdwallet/hds/test_hds_bip86.py @@ -11,7 +11,12 @@ def test_bip86_hd(data): bip86_hd: BIP86HD = BIP86HD( - ecc=Cryptocurrency.ECC, coin_type=Cryptocurrency.COIN_TYPE, account=0, change="external-chain", address=0 + ecc=Cryptocurrency.ECC, + coin_type=Cryptocurrency.COIN_TYPE, + wif_prefix=Cryptocurrency.NETWORKS.MAINNET.WIF_PREFIX, + account=0, + change="external-chain", + address=0 ) bip86_hd.from_seed(