mirror of
https://github.com/SeedSigner/seedsigner.git
synced 2026-10-05 23:18:25 +00:00
99 lines
3.5 KiB
Python
99 lines
3.5 KiB
Python
import pytest
|
|
|
|
# Must import test base before the Controller
|
|
from base import FlowTest, FlowStep
|
|
|
|
from seedsigner.controller import Controller, FlowBasedTestUnexpectedViewError
|
|
from seedsigner.gui.screens.screen import RET_CODE__BACK_BUTTON, RET_CODE__POWER_BUTTON
|
|
from seedsigner.models.seed import Seed
|
|
from seedsigner.views.seed_views import SeedBackupView, SeedMnemonicEntryView, SeedOptionsView
|
|
from seedsigner.views.view import MainMenuView, PowerOptionsView, UnhandledExceptionView
|
|
from seedsigner.views.tools_views import ToolsMenuView, ToolsCalcFinalWordNumWordsView
|
|
|
|
|
|
|
|
class TestFlowTest(FlowTest):
|
|
|
|
def test_simple_flow(self):
|
|
"""
|
|
Basic test to ensure the FlowTest can flow through a sequence of Views and
|
|
terminate via the StopControllerCommand.
|
|
"""
|
|
self.run_sequence([
|
|
FlowStep(MainMenuView, button_data_selection=MainMenuView.TOOLS),
|
|
FlowStep(ToolsMenuView, button_data_selection=ToolsMenuView.KEYBOARD),
|
|
FlowStep(ToolsCalcFinalWordNumWordsView, button_data_selection=ToolsCalcFinalWordNumWordsView.TWELVE),
|
|
FlowStep(SeedMnemonicEntryView),
|
|
])
|
|
|
|
|
|
def test_wrong_View_cls(self):
|
|
"""
|
|
Ensure that the FlowTest will raise an AssertionError if the next View in the
|
|
sequence is not the expected View.
|
|
"""
|
|
with pytest.raises(FlowBasedTestUnexpectedViewError):
|
|
self.run_sequence([
|
|
FlowStep(MainMenuView, button_data_selection=RET_CODE__POWER_BUTTON),
|
|
FlowStep(ToolsMenuView), # <-- Wrong target View! Should raise an AssertionError.
|
|
])
|
|
|
|
|
|
def test_before_run_executes(self):
|
|
"""
|
|
Ensure that the FlowTest can execute a function before running a View.
|
|
"""
|
|
# TODO
|
|
pass
|
|
|
|
|
|
def test_back_button_flow(self):
|
|
"""
|
|
Ensure that the FlowTest works correctly with the Controller and its BackStack.
|
|
"""
|
|
self.run_sequence([
|
|
FlowStep(MainMenuView, screen_return_value=RET_CODE__POWER_BUTTON),
|
|
FlowStep(PowerOptionsView, screen_return_value=RET_CODE__BACK_BUTTON),
|
|
FlowStep(MainMenuView),
|
|
])
|
|
|
|
|
|
def test_initial_destination(self):
|
|
"""
|
|
Ensure that the FlowTest can start from a View other than MainMenuView.
|
|
"""
|
|
# Don't have to start at the MainMenuView; can jump straight in
|
|
self.run_sequence([
|
|
FlowStep(ToolsCalcFinalWordNumWordsView),
|
|
])
|
|
|
|
# And again, but this time with a View that requires input view_args
|
|
self.reset_controller()
|
|
|
|
# Load a seed into the Controller
|
|
controller = Controller.get_instance()
|
|
seed = Seed(mnemonic=["abandon "* 11 + "about"])
|
|
controller.storage.set_pending_seed(seed)
|
|
controller.storage.finalize_pending_seed()
|
|
|
|
self.run_sequence(
|
|
initial_destination_view_args=dict(seed_num=0),
|
|
sequence=[
|
|
FlowStep(SeedOptionsView, button_data_selection=SeedOptionsView.BACKUP),
|
|
FlowStep(SeedBackupView),
|
|
]
|
|
)
|
|
|
|
|
|
def test_raise_exception_via_screen_return_value(self):
|
|
"""
|
|
Ensure that the FlowTest can raise an exception via the screen_return_value.
|
|
"""
|
|
# A generic Exception should be caught by the Controller and routed to the
|
|
# UnhandledExceptionView.
|
|
self.run_sequence([
|
|
FlowStep(MainMenuView, screen_return_value=Exception("Test exception")),
|
|
FlowStep(UnhandledExceptionView),
|
|
])
|
|
|