more tests

This commit is contained in:
newtonick
2021-08-01 23:01:02 -04:00
parent bc614e890b
commit 0f7cc42e4c
+15 -25
View File
@@ -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"