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