From b30d9f68973bf0b79179a7d26d1f66949c4dbe90 Mon Sep 17 00:00:00 2001
From: matejcik <ja@matejcik.cz>
Date: Fri, 20 Sep 2019 17:39:25 +0200
Subject: [PATCH] tests: add backup test for SLIP39 Basic

---
 tests/device_tests/test_msg_backup_device.py | 68 +++++++++++++++++++-
 1 file changed, 66 insertions(+), 2 deletions(-)

diff --git a/tests/device_tests/test_msg_backup_device.py b/tests/device_tests/test_msg_backup_device.py
index ce963204a5..3a2685a38c 100644
--- a/tests/device_tests/test_msg_backup_device.py
+++ b/tests/device_tests/test_msg_backup_device.py
@@ -16,16 +16,22 @@
 
 
 import pytest
+import shamir_mnemonic as shamir
 
 from trezorlib import device, messages
 from trezorlib.messages import ButtonRequestType as B
 
-from ..common import MNEMONIC12, read_and_confirm_mnemonic
+from ..common import (
+    MNEMONIC12,
+    MNEMONIC_SLIP39_BASIC_20_3of6,
+    click_through,
+    read_and_confirm_mnemonic,
+)
 
 
 @pytest.mark.skip_t1  # TODO we want this for t1 too
 @pytest.mark.setup_client(mnemonic=MNEMONIC12)
-def test_backup(client):
+def test_backup_bip39(client):
     assert client.features.needs_backup is True
     mnemonic = None
 
@@ -54,3 +60,61 @@ def test_backup(client):
         device.backup(client)
 
     assert mnemonic == MNEMONIC12
+
+
+@pytest.mark.skip_t1
+@pytest.mark.setup_client(mnemonic=MNEMONIC_SLIP39_BASIC_20_3of6)
+def test_backup_slip39_basic(client):
+    assert client.features.needs_backup is True
+    mnemonics = []
+
+    def input_flow():
+        # 1. Checklist
+        # 2. Number of shares (5)
+        # 3. Checklist
+        # 4. Threshold (3)
+        # 5. Checklist
+        # 6. Confirm show seeds
+        yield from click_through(client.debug, screens=6, code=B.ResetDevice)
+
+        # Mnemonic phrases
+        for _ in range(5):
+            yield  # Phrase screen
+            mnemonic = read_and_confirm_mnemonic(client.debug, words=20)
+            mnemonics.append(mnemonic)
+            yield  # Confirm continue to next
+            client.debug.press_yes()
+
+        # Confirm backup
+        yield
+        client.debug.press_yes()
+
+    with client:
+        client.set_input_flow(input_flow)
+        client.set_expected_responses(
+            [
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.Success),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.Success),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.Success),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.Success),
+                messages.ButtonRequest(code=B.ResetDevice),
+                messages.ButtonRequest(code=B.Success),
+                messages.ButtonRequest(code=B.Success),
+                messages.Success(),
+            ]
+        )
+        device.backup(client)
+
+    expected_ms = shamir.combine_mnemonics(MNEMONIC_SLIP39_BASIC_20_3of6)
+    actual_ms = shamir.combine_mnemonics(mnemonics[:3])
+    assert expected_ms == actual_ms