mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-18 05:28:40 +00:00
91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
# This file is part of the Trezor project.
|
|
#
|
|
# Copyright (C) 2012-2023 SatoshiLabs and contributors
|
|
#
|
|
# This library is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License version 3
|
|
# as published by the Free Software Foundation.
|
|
#
|
|
# This library is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the License along with this library.
|
|
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
|
|
|
|
from contextlib import contextmanager
|
|
from typing import TYPE_CHECKING, Generator
|
|
|
|
import pytest
|
|
|
|
from trezorlib import device
|
|
from trezorlib.exceptions import Cancelled
|
|
|
|
if TYPE_CHECKING:
|
|
from trezorlib.debuglink import DebugLink
|
|
|
|
from ..device_handler import BackgroundDeviceHandler
|
|
|
|
|
|
# TR-only
|
|
pytestmark = [pytest.mark.skip_t1, pytest.mark.skip_t2]
|
|
|
|
|
|
@contextmanager
|
|
def prepare_tutorial_and_cancel_after_it(
|
|
device_handler: "BackgroundDeviceHandler", cancelled: bool = False
|
|
) -> Generator["DebugLink", None, None]:
|
|
debug = device_handler.debuglink()
|
|
device_handler.run(device.show_device_tutorial)
|
|
|
|
yield debug
|
|
|
|
try:
|
|
device_handler.result()
|
|
except Cancelled:
|
|
if not cancelled:
|
|
raise
|
|
|
|
|
|
def go_through_tutorial(debug: "DebugLink") -> None:
|
|
debug.press_right(wait=True)
|
|
debug.press_right(wait=True)
|
|
debug.press_right_htc(hold_ms=1000)
|
|
debug.press_right(wait=True)
|
|
debug.press_right(wait=True)
|
|
layout = debug.press_middle(wait=True)
|
|
assert layout.title() == "TUTORIAL COMPLETE"
|
|
|
|
|
|
@pytest.mark.setup_client(uninitialized=True)
|
|
def test_tutorial_finish(device_handler: "BackgroundDeviceHandler"):
|
|
with prepare_tutorial_and_cancel_after_it(device_handler) as debug:
|
|
# CLICK THROUGH
|
|
go_through_tutorial(debug)
|
|
|
|
# FINISH
|
|
debug.press_right(wait=True)
|
|
|
|
|
|
@pytest.mark.setup_client(uninitialized=True)
|
|
def test_tutorial_skip(device_handler: "BackgroundDeviceHandler"):
|
|
with prepare_tutorial_and_cancel_after_it(device_handler, cancelled=True) as debug:
|
|
# SKIP
|
|
debug.press_left(wait=True)
|
|
debug.press_right(wait=True)
|
|
|
|
|
|
@pytest.mark.setup_client(uninitialized=True)
|
|
def test_tutorial_again_and_skip(device_handler: "BackgroundDeviceHandler"):
|
|
with prepare_tutorial_and_cancel_after_it(device_handler, cancelled=True) as debug:
|
|
# CLICK THROUGH
|
|
go_through_tutorial(debug)
|
|
|
|
# AGAIN
|
|
debug.press_left(wait=True)
|
|
|
|
# SKIP
|
|
debug.press_left(wait=True)
|
|
debug.press_right(wait=True)
|