def test_compile_equipment_point_records(self, energy_record_compiler, uow):
syrx_file_handler = SyrxFileHandler()
range1 = RecordRange()
range1.syrx_num = "syrx_num_1"
range1.start_date = datetime.datetime(2012, 1, 1)
range1.end_date = datetime.datetime(2012, 3, 1)
range2 = RecordRange()
range2.syrx_num = "syrx_num_2"
range2.start_date = datetime.datetime(2012, 2, 1)
range2.end_date = datetime.datetime(2012, 3, 1)
ranges = [range1, range2]
syrx_file_handler.compile_equipment_point_records(ranges)
compiler = energy_record_compiler.return_value
uow.return_value.compiled_energy_records.delete_compiled_equipment_point_records\
.assert_has_calls([call(range1.syrx_num, range1.start_date, range1.end_date),
call(range2.syrx_num, range2.start_date, range2.end_date)])
compiler.compile_component_point_records_by_year_span\
.assert_has_calls([call(range1.syrx_num, range1.start_date, range1.end_date),
call(range2.syrx_num, range2.start_date, range2.end_date)])
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)])
