fix test_widgets.py

freedomofpress · Feb 13, 2024 · 044f902 · 044f902
1 parent d574413
commit 044f902
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/client/tests/gui/test_widgets.py b/client/tests/gui/test_widgets.py
@@ -3589,8 +3589,6 @@ def test_FileWidget__on_export_clicked(mocker, session, source):
     controller = mocker.MagicMock(get_file=get_file)
     file_location = file.location(controller.data_dir)
 
-    export_device = mocker.patch("securedrop_client.export.Export")
-
     fw = FileWidget(
         file.uuid, controller, mocker.MagicMock(), mocker.MagicMock(), mocker.MagicMock(), 0, 123
     )
@@ -3599,6 +3597,7 @@ def test_FileWidget__on_export_clicked(mocker, session, source):
     controller.run_export_preflight_checks = mocker.MagicMock()
     controller.downloaded_file_exists = mocker.MagicMock(return_value=True)
 
+    export_device = mocker.patch("securedrop_client.gui.widgets.Export")
     wizard = mocker.patch("securedrop_client.gui.conversation.ExportWizard")
 
     fw._on_export_clicked()
@@ -3650,7 +3649,6 @@ def test_FileWidget__on_print_clicked(mocker, session, source):
 
     get_file = mocker.MagicMock(return_value=file)
     controller = mocker.MagicMock(get_file=get_file)
-    export_device = mocker.patch("securedrop_client.export.Export")
     file_location = file.location(controller.data_dir)
 
     fw = FileWidget(
@@ -3667,6 +3665,7 @@ def test_FileWidget__on_print_clicked(mocker, session, source):
     controller.print_file = mocker.MagicMock()
     controller.downloaded_file_exists = mocker.MagicMock(return_value=True)
 
+    export_device = mocker.patch("securedrop_client.gui.widgets.Export")
     dialog = mocker.patch("securedrop_client.gui.conversation.PrintDialog")
 
     fw._on_print_clicked()