def test_SourceWidget_setup():
"""
The setup method adds the controller as an attribute on the SourceWidget.
"""
mock_controller = mock.MagicMock()
mock_source = mock.MagicMock()
sw = SourceWidget(None, mock_source)
sw.setup(mock_controller)
assert sw.controller == mock_controller
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()
