Update test_flows_seed.py

Added FlowStep(seed_views.SeedExportXpubQRAskVerifyAddView, screen_return_value=0) to seed flow tests.
SeedSigner · Dec 21, 2024 · 94fef88 · 94fef88
1 parent b8b2674
commit 94fef88
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_flows_seed.py b/tests/test_flows_seed.py
@@ -195,6 +195,7 @@ def flowtest_standard_xpub(sig_tuple, script_tuple, coord_tuple):
                     FlowStep(seed_views.SeedExportXpubWarningView, screen_return_value=0),
                     FlowStep(seed_views.SeedExportXpubDetailsView, screen_return_value=0),
                     FlowStep(seed_views.SeedExportXpubQRDisplayView, screen_return_value=0),
+                    FlowStep(seed_views.SeedExportXpubQRAskVerifyAddView, screen_return_value=0),
                     FlowStep(scan_views.ScanAddressView, screen_return_value=0),
                     FlowStep(MainMenuView),
                 ]
@@ -320,6 +321,7 @@ def test_export_xpub_custom_derivation_flow(self):
                 FlowStep(seed_views.SeedExportXpubWarningView, screen_return_value=0),
                 FlowStep(seed_views.SeedExportXpubDetailsView, screen_return_value=0),
                 FlowStep(seed_views.SeedExportXpubQRDisplayView, screen_return_value=0),
+                FlowStep(seed_views.SeedExportXpubQRAskVerifyAddView, screen_return_value=0),
                 FlowStep(scan_views.ScanAddressView, screen_return_value=0),
                 FlowStep(MainMenuView),
             ]
@@ -352,6 +354,7 @@ def test_export_xpub_skip_non_option_flow(self):
                 FlowStep(seed_views.SeedExportXpubWarningView, screen_return_value=0),
                 FlowStep(seed_views.SeedExportXpubDetailsView, screen_return_value=0),
                 FlowStep(seed_views.SeedExportXpubQRDisplayView, screen_return_value=0),
+                FlowStep(seed_views.SeedExportXpubQRAskVerifyAddView, screen_return_value=0),
                 FlowStep(scan_views.ScanAddressView, screen_return_value=0),
                 FlowStep(MainMenuView),
             ]
@@ -384,6 +387,7 @@ def test_export_xpub_electrum_seed_flow(self):
                 FlowStep(seed_views.SeedExportXpubWarningView, screen_return_value=0),
                 FlowStep(seed_views.SeedExportXpubDetailsView, screen_return_value=0),
                 FlowStep(seed_views.SeedExportXpubQRDisplayView, screen_return_value=0),
+                FlowStep(seed_views.SeedExportXpubQRAskVerifyAddView, screen_return_value=0),
                 FlowStep(scan_views.ScanAddressView, screen_return_value=0),
                 FlowStep(MainMenuView),
             ]