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_SourceWidget_toggle_star():
"""
The toggle_star method should call self.controller.update_star
"""
mock_controller = mock.MagicMock()
mock_source = mock.MagicMock()
event = mock.MagicMock()
sw = SourceWidget(None, mock_source)
sw.controller = mock_controller
sw.controller.update_star = mock.MagicMock()
sw.toggle_star(event)
sw.controller.update_star.assert_called_once_with(mock_source)
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()
