Skip to content

Commit

Permalink
test QR viewer standard Seed QR
Browse files Browse the repository at this point in the history
  • Loading branch information
odudex committed Feb 12, 2024
1 parent d2f53ac commit 236531d
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions tests/pages/test_qr_view.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
from ..shared_mocks import mock_context
from .test_home import create_ctx, tdata


def test_load_qr_view(amigo_tft, mocker):
Expand Down Expand Up @@ -27,6 +28,39 @@ def test_load_qr_view(amigo_tft, mocker):
assert ctx.display.draw_qr_code.call_count == 8


def test_load_seed_qr(amigo_tft, mocker, tdata):
from krux.pages.qr_view import SeedQRView
from krux.themes import WHITE
from krux.input import BUTTON_ENTER, BUTTON_PAGE_PREV, SWIPE_LEFT, SWIPE_RIGHT
from krux.wallet import Wallet

BTN_SEQUENCE = [
SWIPE_LEFT, # lines mode
BUTTON_ENTER, # move to line 1
SWIPE_LEFT, # zoomed regions mode
SWIPE_LEFT, # regions mode
SWIPE_LEFT, # grided mode
SWIPE_LEFT, # back to standard mode
SWIPE_LEFT, # lines mode again
SWIPE_RIGHT, # back to standard mode
BUTTON_ENTER, # leave
BUTTON_PAGE_PREV, # move to Back to Main Menu
BUTTON_ENTER, # confirm
]

ctx = create_ctx(mocker, BTN_SEQUENCE, Wallet(tdata.SINGLESIG_12_WORD_KEY), None)
seed_qr_view = SeedQRView(ctx, binary=False)
seed_qr_view.display_qr()
assert ctx.display.draw_qr_code.call_count == 8
ctx.display.draw_qr_code.assert_called_with(
0,
# Standard SeedQR
bytearray(
b"\x7fT\xc8?\xa8P\nvA$\xdd\xae\xdb\xab\xdb%\x84t\x83\xb6\xa3\xe0_U\xf5\x07p\x0e\x00\xdf\xbb\xb7*\xb4\x0ei.Q\x96\xa4\x07H\xcc\xa8\xe4z\xeb.\xb2\x94\xa2s%\xe1\xca\x85\xe6\xc8V\xa6\xb7\xaeGKi\x16A\xd2\xbcX(\x18Pwz\xff\x00\xfa0\xea\xdf%V\n\xb2\xa2H]\xa1\xf7\xaf+\x0bUv\xc5\x16\xbd\xa0'\xfe\xf4\x17zr\x00"
),
)


def test_loop_through_regions(amigo_tft, mocker):
from krux.pages.qr_view import SeedQRView
from krux.input import BUTTON_ENTER, BUTTON_PAGE_PREV, SWIPE_LEFT, SWIPE_RIGHT
Expand Down

0 comments on commit 236531d

Please sign in to comment.