Add basic tests for the ScreenshotDialog widget and fixes

napari/_qt/dialogs/_tests/test_screenshot_dialog.py

@@ -0,0 +1,89 @@
+import pytest
+from qtpy.QtCore import QTimer
+from qtpy.QtWidgets import QApplication, QFileDialog, QLineEdit, QMessageBox
+
+from napari._qt.dialogs.screenshot_dialog import ScreenshotDialog
+from napari.utils.history import get_save_history
+
+
+@pytest.mark.parametrize("filename", ["test", "test.png", "test.tif"])
+def test_screenshot_save(qtbot, tmp_path, filename):
+    """Check passing different extensions with the filename."""
+
+    def save_function(path):
+        # check incoming path has extension event when a filename without one
+        # was provided
+        assert filename in path
+        if "." not in filename:
+            assert ".png" in path
+
+        # create a file with the given path to check for
+        # non-native qt overwrite message
+        with open(path, "w") as mock_img:
+            mock_img.write("")
+
+    qt_overwrite_shown = False
+
+    def handle_qt_overwrite_message():
+        for widget in QApplication.topLevelWidgets():
+            if isinstance(widget, QMessageBox):
+                # test should not enter here!
+                widget.accept()
+                nonlocal qt_overwrite_shown
+                qt_overwrite_shown = True
+                break
+
+    # setup dialog
+    dialog = ScreenshotDialog(
+        save_function, directory=str(tmp_path), history=get_save_history()
+    )
+    qtbot.addWidget(dialog)
+    dialog.setOptions(QFileDialog.DontUseNativeDialog)
+    dialog.show()
+
+    # check dialog and set filename
+    assert dialog.windowTitle() == 'Save screenshot'
+    line_edit = dialog.findChild(QLineEdit)
+    line_edit.setText(filename)
+
+    # check that no warning message related with overwriting is shown
+    QTimer.singleShot(100, handle_qt_overwrite_message)
+    dialog.accept()
+    qtbot.wait(100)
+    assert not qt_overwrite_shown, "Qt non-native overwrite message was shown!"
+
+
+def test_screenshot_overwrite_save(qtbot, tmp_path, monkeypatch):
+    """Check overwriting file validation."""
+    filename = tmp_path / "test.png"
+    filename.write_text("")
+
+    def save_function(path):
+        assert "test.png" in path
+
+    def overwrite_message(*args):
+        box, parent, title, text, buttons, default = args
+        assert parent == dialog
+        assert title == "Confirm overwrite"
+        assert "test.png" in text
+        assert "already exists. Do you want to replace it?" in text
+        assert buttons == QMessageBox.Yes | QMessageBox.No
+        assert default == QMessageBox.No
+
+        return QMessageBox.Yes
+
+    # monkeypath custom overwrite QMessageBox usage
+    monkeypatch.setattr(QMessageBox, "warning", overwrite_message)
+
+    dialog = ScreenshotDialog(
+        save_function, directory=str(tmp_path), history=get_save_history()
+    )
+    qtbot.addWidget(dialog)
+    dialog.setOptions(QFileDialog.DontUseNativeDialog)
+    dialog.show()
+
+    # check dialog, set filename and trigger accept logic
+    assert dialog.windowTitle() == 'Save screenshot'
+    line_edit = dialog.findChild(QLineEdit)
+    line_edit.setText("test")
+    dialog.accept()

napari/_qt/dialogs/screenshot_dialog.py

@@ -62,8 +62,10 @@ def accept(self):
                     QMessageBox.No,
                 )
                 if res != QMessageBox.Yes:
-                    # standard accept return 1, reject 0. This inform that dialog should be reopened
-                    super().accept()
-                    self.exec_()
-        self.save_function(save_path)
-        return super().accept()
+                    # return in this case since still a valida name for the
+                    # file is needed and the dialog need to still be visible
+                    return None
+        base_result = super().accept()
+        if base_result:
+            self.save_function(save_path)
+        return base_result