def test_SourceWidget_delete_source_when_user_chooses_cancel(mocker):
mock_message_box_question = mocker.MagicMock(QMessageBox.question)
mock_message_box_question.return_value = QMessageBox.Cancel
mock_source = mocker.MagicMock()
mock_source.journalist_designation = 'foo bar baz'
mock_source.submissions = []
submission_files = (
"submission_1-msg.gpg",
"submission_2-msg.gpg",
"submission_3-doc.gpg",
)
for filename in submission_files:
submission = mocker.MagicMock()
submission.filename = filename
mock_source.submissions.append(submission)
mock_controller = mocker.MagicMock()
sw = SourceWidget(None, mock_source)
sw.controller = mock_controller
mocker.patch(
"securedrop_client.gui.widgets.QMessageBox.question",
mock_message_box_question,
)
sw.delete_source(None)
sw.controller.delete_source.assert_not_called()
def test_DeleteSource_from_source_widget_when_user_is_loggedout(mocker):
mock_source = mocker.MagicMock()
mock_controller = mocker.MagicMock(logic.Client)
mock_controller.api = None
mock_event = mocker.MagicMock()
mock_delete_source_message_box_obj = mocker.MagicMock()
mock_delete_source_message_box = mocker.MagicMock()
mock_delete_source_message_box.return_value = (
mock_delete_source_message_box_obj)
with mocker.patch('securedrop_client.gui.widgets.DeleteSourceMessageBox',
mock_delete_source_message_box):
source_widget = SourceWidget(None, mock_source)
source_widget.setup(mock_controller)
source_widget.delete_source(mock_event)
mock_delete_source_message_box_obj.launch.assert_not_called()
def test_SourceWidget_delete_source(mocker):
mock_source = mocker.MagicMock()
mock_delete_source_message_box_object = mocker.MagicMock(
DeleteSourceMessageBox)
mock_controller = mocker.MagicMock()
mock_delete_source_message = mocker.MagicMock(
return_value=mock_delete_source_message_box_object)
sw = SourceWidget(None, mock_source)
sw.controller = mock_controller
mocker.patch(
"securedrop_client.gui.widgets.DeleteSourceMessageBox",
mock_delete_source_message,
)
sw.delete_source(None)
mock_delete_source_message_box_object.launch.assert_called_with()
