def test_print(mocker): """ Ensure TemporaryDirectory is used when creating and sending the archive containing the file to print and that the success signal is emitted. """ mock_temp_dir = mocker.MagicMock() mock_temp_dir.__enter__ = mocker.MagicMock(return_value="mock_temp_dir") mocker.patch("securedrop_client.export.TemporaryDirectory", return_value=mock_temp_dir) export = Export() export.print_call_success = mocker.MagicMock() export.print_call_success.emit = mocker.MagicMock() export.export_completed = mocker.MagicMock() export.export_completed.emit = mocker.MagicMock() _run_print = mocker.patch.object(export, "_run_print") mocker.patch("os.path.exists", return_value=True) export.print(["path1", "path2"]) _run_print.assert_called_once_with("mock_temp_dir", ["path1", "path2"]) export.print_call_success.emit.assert_called_once_with() export.export_completed.emit.assert_called_once_with(["path1", "path2"])
def test_print(mocker): ''' Ensure TemporaryDirectory is used when creating and sending the archive containing the file to print and that the success signal is emitted. ''' mock_temp_dir = mocker.MagicMock() mock_temp_dir.__enter__ = mocker.MagicMock(return_value='mock_temp_dir') mocker.patch('securedrop_client.export.TemporaryDirectory', return_value=mock_temp_dir) export = Export() export.print_call_success = mocker.MagicMock() export.print_call_success.emit = mocker.MagicMock() export.export_completed = mocker.MagicMock() export.export_completed.emit = mocker.MagicMock() _run_print = mocker.patch.object(export, '_run_print') mocker.patch('os.path.exists', return_value=True) export.print(['path1', 'path2']) _run_print.assert_called_once_with('mock_temp_dir', ['path1', 'path2']) export.print_call_success.emit.assert_called_once_with() export.export_completed.emit.assert_called_once_with(['path1', 'path2'])