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
def test_process_all_syrx_files(self, listdir, uow): syrx_file_handler = SyrxFileHandler() syrx_file_handler.logger = Mock() syrx_file_handler.syrx_upload_folder = Mock() syrx_file_handler.process_syrx_filename = Mock() syrx_file_handler.process_syrx_filename.return_value = [] syrx_file_handler.compile_equipment_point_records = Mock() listdir.return_value = ["file1", "file2", "file3", "file4", "file5", "file6"] syrx_file_handler.process_all_syrx_files() syrx_file_handler.process_syrx_filename.assert_has_calls([call(0, "file1", 6), call(1, "file2", 6), call(2, "file3", 6), call(3, "file4", 6), call(4, "file5", 6), call(5, "file6", 6)])