diff --git a/tests/test_encodepsbtqr.py b/tests/test_encodepsbtqr.py index 00a2ba94..a9bddb46 100644 --- a/tests/test_encodepsbtqr.py +++ b/tests/test_encodepsbtqr.py @@ -1,7 +1,7 @@ import pytest from mock import MagicMock from seedsigner.models.decode_qr import QRType -from seedsigner.models.encode_psbt_qr import EncodePSBTQR, EncodePSBTQRDensity +from seedsigner.models.encode_qr import EncodeQR, EncodeQRDensity from embit import psbt from binascii import a2b_base64 @@ -11,33 +11,11 @@ def test_ur_qr_encode(): tx = psbt.PSBT.parse(a2b_base64(base64_psbt)) - e = EncodePSBTQR(tx, QRType.PSBTUR2) + e = EncodeQR(tx, QRType.PSBTUR2) cnt = 0 while cnt <= 10: fragment = e.nextPart() - if (cnt+1) == 1: - assert fragment == "UR:CRYPTO-PSBT/1-114/LPADCSJPCFCMCPCYBBRSPTSKHDEYHKCMCTJOJKIDJYZMADAELDAOAEAEAEADOEVWFXYKGYENNSKTCXHPISFEEOLFNNPMPAKENSCXZEKBTTYKJOJYGTEEVDMDLBTYCAAETPMUCEHE" - elif (cnt+1) == 2: - assert fragment == "UR:CRYPTO-PSBT/2-114/LPAOCSJPCFCMCPCYBBRSPTSKHDEYAEAEAEZCZMZMZMAOUOTIAXAEAEAEAEAECPAECXBDAHGMLSLACKTBGYLGCTFNRKWDEYSNVSIEMHINDAIOMOTAVAEEDYJPRKZTDKNEQZPYMYUY" - elif (cnt+1) == 3: - assert fragment == "UR:CRYPTO-PSBT/3-114/LPAXCSJPCFCMCPCYBBRSPTSKHDEYPAGDSRAEAEAEAEAEAECPAECXNSLPIEMSRKSTBBCNHYNYAAAMCMZTMKLDCYRKTEOYOYPFTDWSCMTPSWJYFDLKMDBGAEAEAEAEGWADBATDSAVY" - elif (cnt+1) == 4: - assert fragment == "UR:CRYPTO-PSBT/4-114/LPAACSJPCFCMCPCYBBRSPTSKHDEYAALOPRCKAAAHLFRLTNLAAEAEAOOTZTPMISSTBZBEIYGSFELEURTIAHLGUEMTOTHPDYTLGSIHLALUTTESGYTYETKOVOAXFTUEZSUEGLECPFTL" - elif (cnt+1) == 5: - assert fragment == "UR:CRYPTO-PSBT/5-114/LPAHCSJPCFCMCPCYBBRSPTSKHDEYRESFWDLEBZHLJSHYTITKRHFLBEFGCASFYLJZVYTTMTPYWZPFSBFNMOMHBBKBURNSHKDYAEAELAAEAEAELAAEAEAELAAOAEAELAGWMTENHGPA" - elif (cnt+1) == 6: - assert fragment == "UR:CRYPTO-PSBT/6-114/LPAMCSJPCFCMCPCYBBRSPTSKHDEYADAALOPRCKAAGMCXWSVDLAAEAEAORHRFBNKEMUHPSBKEURNETNDYLRHGVTMSYKUEZTISLTJPCNBNSTBEKIRSPKRKAHIOAORDREKIEMLBCHFP" - elif (cnt+1) == 7: - assert fragment == "UR:CRYPTO-PSBT/7-114/LPATCSJPCFCMCPCYBBRSPTSKHDEYPSLBLEWZVDBZGUKEESTNIOAARPTIVWKTFMVELTMSBGTEBEJNHTTOWTNLENBBVTLYCWJEDYAEAELAAEAEAELAAEAEAELAAOAEAELACPBEBYGH" - elif (cnt+1) == 8: - assert fragment == "UR:CRYPTO-PSBT/8-114/LPAYCSJPCFCMCPCYBBRSPTSKHDEYGWADAALOPRCKAAHEWSAMDMLAAEAEAOWSGEHKUONSIOCLGHDISGWYTAWFCEYKHDJOBBHHJPGTASCFBDPSOXHSINJOIAZEHSAOAEHLPKJOSEPE" - elif (cnt+1) == 9: - assert fragment == "UR:CRYPTO-PSBT/9-114/LPASCSJPCFCMCPCYBBRSPTSKHDEYLSLSJESRSPCHKSGLFTKPCKHNKSIHRSLSTIYTFGFWIADTTIAYWDBKLYGMGYLNBBCSGTATWMDYAEAELAAEAEAELAAEAEAELAAOAEAESFFDRNZO" - elif (cnt+1) == 10: - assert fragment == "UR:CRYPTO-PSBT/10-114/LPBKCSJPCFCMCPCYBBRSPTSKHDEYLAGWADAALOPRCKAAHNMHAHGOLAAEAEAOSPBDDRHETNLNBGSTRYDLYLBKTPKBWKKTHYTSIAATDKWYGWFHSSFYDRTBTBMOOTLGAOWNFEFYBNBE" - elif (cnt+1) == 11: - assert fragment == "UR:CRYPTO-PSBT/11-114/LPBDCSJPCFCMCPCYBBRSPTSKHDEYBWPDLDGSFSIDLBLBHPLYRYCXLNCMJLQDSFENDINSFZJNROAMCNIYDKDNTOFHIYBBQZEOVTMDDYAEAELAAEAEAELAAEAEAELAAOAEWNSBHTIY" img = e.part2Image(fragment) cnt += 1 @@ -48,7 +26,7 @@ def test_specter_qr_encode(): tx = psbt.PSBT.parse(a2b_base64(base64_psbt)) - e = EncodePSBTQR(tx, QRType.PSBTSPECTER) + e = EncodeQR(tx, QRType.PSBTSPECTER) cnt = 0 while cnt <= 10: @@ -79,3 +57,15 @@ def test_specter_qr_encode(): img = e.part2Image(fragment) cnt += 1 +def test_seedsigner_qr(): + + mnemonic = "obscure bone gas open exotic abuse virus bunker shuffle nasty ship dash" + + e = EncodeQR(mnemonic.split(" "), QRType.SEEDSSQR) + + print(e.nextPart()) + + assert e.nextPart() == "121802020768124106400009195602431595117715840445" + + +