def test_process_syrx_file_mass_records(self, uow):
syrx_file_handler = SyrxFileHandler()
syrx_file_handler.logger = Mock()
syrx_file_handler.date_str = Mock()
bad_record_file_mock = MagicMock()
summary = ProcessSyrxRecordsReturn()
summary.num_good_records = 200
summary.num_bad_records = 0
syrx_file_handler.process_syrx_records = Mock(return_value=summary)
read_file_stream = io.StringIO()
for i in range(400):
read_file_stream.write(u'{"name": "record"}\n')
# add extra record to force small batch for good measure
read_file_stream.write(u'{"name": "record"}')
read_file_stream.seek(0)
parsed_list = [{"name": "record"} for i in range(200)]
rv = syrx_file_handler.process_syrx_file(read_file_stream, bad_record_file_mock)
syrx_file_handler.process_syrx_records.assert_has_calls([call(parsed_list, bad_record_file_mock),
call(parsed_list, bad_record_file_mock),
call([{"name": "record"}], bad_record_file_mock)])
assert rv.num_good_records == 600
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_file(self, uow):
syrx_file_handler = SyrxFileHandler()
syrx_file_handler.logger = Mock()
syrx_file_handler.syrx_upload_folder = Mock()
syrx_file_handler.date_str = Mock()
bad_record_file_mock = MagicMock()
summary = ProcessSyrxRecordsReturn()
summary.num_good_records = 3
summary.num_bad_records = 1
syrx_file_handler.process_syrx_records = Mock(return_value=summary)
read_file_stream = io.StringIO()
read_file_stream.write(u'{"name": "record0"}\n'
u'{"name": "record1"}\n'
u'{"name": "record2"}\n'
u'{"name": "record3"}')
read_file_stream.seek(0)
parsed_list = [{"name": "record0"}, {"name": "record1"}, {"name": "record2"}, {"name": "record3"}]
rv = syrx_file_handler.process_syrx_file(read_file_stream, bad_record_file_mock)
syrx_file_handler.process_syrx_records.assert_called_with(parsed_list, bad_record_file_mock)
assert rv.num_good_records == 3 and rv.num_bad_records == 1
