Add unit test to document change button

Fixes #428

dangerzone/gui/main_window.py

@@ -377,7 +377,9 @@ def dangerous_doc_button_clicked(self) -> None:
             self.file_dialog.setDirectory(first_doc_dir)
 
         if self.file_dialog.exec():
-            documents = [Document(filename) for filename in self.file_dialog.selectedFiles()]
+            documents = [
+                Document(filename) for filename in self.file_dialog.selectedFiles()
+            ]
             self.documents_selected.emit(documents)
         else:
             # No files selected

pyproject.toml

@@ -39,6 +39,7 @@ pytest-mock = "^3.10.0"
 pytest-xdist = "^2.5.0"
 pytest-cov = "^3.0.0"
 strip-ansi = "*"
+pytest-qt = "^4.2.0"
 
 [build-system]
 requires = ["poetry>=1.1.4"]

tests/__init__.py

@@ -9,6 +9,7 @@
 
 SAMPLE_DIRECTORY = "test_docs"
 BASIC_SAMPLE = "sample-pdf.pdf"
+BASIC_SAMPLE2 = "sample-doc.doc"
 test_docs_dir = Path(__file__).parent.joinpath(SAMPLE_DIRECTORY)
 test_docs = [
     p
@@ -30,6 +31,11 @@ def sample_doc() -> str:
     return str(test_docs_dir.joinpath(BASIC_SAMPLE))
 
 
+@pytest.fixture
+def sample_doc2() -> str:
+    return str(test_docs_dir.joinpath(BASIC_SAMPLE2))
+
+
 @pytest.fixture
 def unreadable_pdf(tmp_path: Path) -> str:
     file_path = tmp_path / "document.pdf"

tests/gui/test_main_window.py

@@ -0,0 +1,80 @@
+import pytest
+from pytest_mock import MockerFixture
+from pytestqt.qtbot import QtBot
+
+from dangerzone.gui.main_window import *
+
+from .. import sample_doc, sample_doc2
+
+# FIXME: See https://github.com/freedomofpress/dangerzone/issues/320 for more details.
+if typing.TYPE_CHECKING:
+    from PySide2 import QtCore
+else:
+    try:
+        from PySide6 import QtCore, QtGui, QtWidgets
+    except ImportError:
+        from PySide2 import QtCore, QtGui, QtWidgets
+
+##
+## Widget Fixtures
+##
+
+
+@pytest.fixture
+def content_widget(qtbot: QtBot, mocker: MockerFixture) -> QtWidgets.QWidget:
+    # Setup
+    mock_app = mocker.MagicMock()
+    dummy = mocker.MagicMock()
+    dz = DangerzoneGui(mock_app, dummy)
+    w = ContentWidget(dz)
+    qtbot.addWidget(w)
+    return w
+
+
+##
+## Document Selection tests
+##
+
+
+def test_change_document_button(
+    content_widget: ContentWidget,
+    qtbot: QtBot,
+    mocker: MockerFixture,
+    sample_doc: str,
+    sample_doc2: str,
+) -> None:
+    # Setup first doc selection
+    file_dialog_mock = mocker.MagicMock()
+    file_dialog_mock.selectedFiles.return_value = (sample_doc,)
+    content_widget.doc_selection_widget.file_dialog = file_dialog_mock
+
+    # Select first file
+    with qtbot.waitSignal(content_widget.documents_added):
+        qtbot.mouseClick(
+            content_widget.doc_selection_widget.dangerous_doc_button,
+            QtCore.Qt.MouseButton.LeftButton,
+        )
+        file_dialog_mock.accept()
+
+    # Setup doc change
+    file_dialog_mock.selectedFiles.return_value = (sample_doc2,)
+
+    # When clicking on "select docs" button
+    with qtbot.waitSignal(content_widget.documents_added):
+        qtbot.mouseClick(
+            content_widget.settings_widget.change_selection_button,
+            QtCore.Qt.MouseButton.LeftButton,
+        )
+        file_dialog_mock.accept()
+
+    # Then two dialogs should have been open
+    assert file_dialog_mock.exec.call_count is 2
+    assert file_dialog_mock.selectedFiles.call_count is 2
+
+    # Then the final document should be only the second one
+    docs = [
+        doc.input_filename
+        for doc in content_widget.dangerzone.get_unconverted_documents()
+    ]
+    assert len(docs) is 1
+    assert docs[0] == sample_doc2