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()