(WIP) Add functional tests for Export Wizard

client/tests/conftest.py

@@ -5,6 +5,7 @@
 from configparser import ConfigParser
 from datetime import datetime
 from uuid import uuid4
+from unittest import mock
 
 import pytest
 from PyQt5.QtCore import Qt
@@ -22,6 +23,7 @@
     Source,
     make_session_maker,
 )
+from securedrop_client.export_status import ExportStatus
 from securedrop_client.gui import conversation
 from securedrop_client.gui.main import Window
 from securedrop_client.logic import Controller
@@ -47,7 +49,7 @@
 TIME_CLICK_ACTION = 1000
 TIME_RENDER_SOURCE_LIST = 20000
 TIME_RENDER_CONV_VIEW = 1000
-TIME_RENDER_EXPORT_DIALOG = 1000
+TIME_RENDER_EXPORT_WIZARD = 1000
 TIME_FILE_DOWNLOAD = 5000
 
 
@@ -168,13 +170,105 @@ def homedir(i18n):
 
 
 @pytest.fixture(scope="function")
-def mock_export():
+def mock_export_locked():
+    """
+    Represents the following scenario:
+        * Locked USB already inserted
+        * "Export" clicked, export wizard launched
+        * Passphrase successfully entered on first attempt (and export suceeeds)
+    """
+    device = conversation.ExportDevice()
+
+    device.run_export_preflight_checks = lambda: device.export_state_changed.emit(
+        ExportStatus.DEVICE_LOCKED
+    )
+    device.run_printer_preflight_checks = lambda: None
+    device.print = lambda filepaths: None
+    device.export = mock.MagicMock()
+    device.export.side_effect = [
+        lambda filepaths, passphrase: device.export_state_changed.emit(
+            ExportStatus.DEVICE_WRITABLE
+        ),
+        lambda filepaths, passphrase: device.export_state_changed.emit(ExportStatus.SUCCESS_EXPORT),
+    ]
+
+    return device
+
+
+@pytest.fixture(scope="function")
+def mock_export_unlocked():
+    """
+    Represents the following scenario:
+        * USB already inserted and unlocked by the user
+        * Export wizard launched
+        * Export succeeds
+    """
+    device = conversation.ExportDevice()
+
+    device.run_export_preflight_checks = lambda: device.export_state_changed.emit(
+        ExportStatus.DEVICE_WRITABLE
+    )
+    device.run_printer_preflight_checks = lambda: None
+    device.print = lambda filepaths: None
+    device.export = lambda filepaths, passphrase: device.export_state_changed.emit(
+        ExportStatus.SUCCESS_EXPORT
+    )
+
+    return device
+
+
+@pytest.fixture(scope="function")
+def mock_export_no_usb_then_bad_passphrase_then_fail():
+    """
+    Represents the following scenario:
+        * Export wizard launched
+        * Locked USB inserted
+        * Mistyped Passphrase
+        * Correct passphrase
+        * Export fails
+    """
+    device = conversation.ExportDevice()
+
+    device.run_export_preflight_checks = lambda: device.export_state_changed.emit(
+        ExportStatus.NO_DEVICE_DETECTED
+    )
+    device.run_printer_preflight_checks = lambda: None
+    device.print = lambda filepaths: None
+    device.export = mock.MagicMock()
+    device.export.side_effect = [
+        lambda filepaths, passphrase: device.export_state_changed.emit(ExportStatus.DEVICE_LOCKED),
+        lambda filepaths, passphrase: device.export_state_changed.emit(
+            ExportStatus.ERROR_UNLOCK_LUKS
+        ),
+        lambda filepaths, passphrase: device.export_state_changed.emit(
+            ExportStatus.DEVICE_WRITABLE
+        ),
+        lambda filepaths, passphrase: device.export_state_changed.emit(ExportStatus.ERROR_EXPORT),
+    ]
+
+    return device
+
+
+@pytest.fixture(scope="function")
+def mock_export_fail_early():
+    """
+    Represents the following scenario:
+        * Locked USB inserted
+        * Export wizard launched
+        * Unrecoverable error before export happens
+          (eg, mount error)
+    """
     device = conversation.ExportDevice()
 
-    device.run_export_preflight_checks = lambda dir: None
-    device.run_printer_preflight_checks = lambda dir: None
+    device.run_export_preflight_checks = lambda: device.export_state_changed.emit(
+        ExportStatus.DEVICE_LOCKED
+    )
+    device.run_printer_preflight_checks = lambda: None
     device.print = lambda filepaths: None
-    device.export = lambda filepaths, passphrase: None
+    device.export = mock.MagicMock()
+    device.export = lambda filepaths, passphrase: device.export_state_changed.emit(
+        ExportStatus.ERROR_MOUNT
+    )
 
     return device