def test_process_syrx_file_path_no_bad_records(self, move, remove, open_mock, path_join, named_temporary_file, uow): syrx_file_handler = SyrxFileHandler() syrx_file_handler.logger = Mock() syrx_file_handler.syrx_upload_folder = Mock() syrx_file_handler.date_str = Mock() summary = ProcessSyrxRecordsReturn() summary.num_bad_records = 0 syrx_file_handler.process_syrx_file = Mock(return_value=summary) file_path = MagicMock() bad_record_file_mock = named_temporary_file.return_value read_file_mock = MagicMock() def open_mock_side_effect(path, mode): if path == file_path and mode == "r": return read_file_mock else: return None open_mock.side_effect = open_mock_side_effect syrx_file_handler.process_syrx_file_path(file_path) named_temporary_file.assert_called_with(delete=False) syrx_file_handler.process_syrx_file.assert_called_with(read_file_mock.__enter__.return_value, bad_record_file_mock) remove.assert_has_calls([call(file_path), call(bad_record_file_mock.name)]) bad_record_file_mock.close.assert_called_with()
def test_process_syrx_filename(self, path_join, uow): syrx_file_handler = SyrxFileHandler() syrx_file_handler.logger = Mock() syrx_file_handler.syrx_upload_folder = Mock() syrx_file_handler.process_syrx_file_path = Mock() file_name = Mock() syrx_file_handler.process_syrx_filename(0, file_name, 1) path_join.assert_called_with(syrx_file_handler.syrx_upload_folder, file_name) syrx_file_handler.process_syrx_file_path.assert_called_with(path_join.return_value) assert syrx_file_handler.process_syrx_file_path.call_count == 1
def test_process_johnson_filename_exception(self, path_join, uow): syrx_file_handler = SyrxFileHandler() syrx_file_handler.logger = Mock() syrx_file_handler.syrx_upload_folder = Mock() syrx_file_handler.date_str = Mock() syrx_file_handler.process_syrx_file_path = Mock(side_effect=Exception()) file_name = Mock() syrx_file_handler.process_syrx_filename(0, file_name, 1) path_join.assert_called_with(syrx_file_handler.syrx_upload_folder, file_name) syrx_file_handler.process_syrx_file_path.assert_called_with(path_join.return_value) assert syrx_file_handler.logger.exception.call_count == 1