def _do_merge_calendar(calendar_file, ntfs_file, output_file):
with ZipFile(calendar_file,
'r') as calendars_zip, ZipFile(ntfs_file, 'r') as ntfs_zip:
grid_calendar_data = GridCalendarData()
grid_calendar_data.load_zips(calendars_zip, ntfs_zip)
new_ntfs_zip = calendar_handler.merge_calendars_ntfs(
grid_calendar_data, ntfs_zip)
calendar_handler.save_zip_as_file(new_ntfs_zip, output_file)
def test_merge_ntfs_calendar_file(ntfs_zip, calendars_zip):
grid_calendar_data = GridCalendarData()
grid_calendar_data.load_zips(calendars_zip, ntfs_zip)
new_ntfs_zip = calendar_handler.merge_calendars_ntfs(grid_calendar_data, ntfs_zip)
assert calendar_handler.GRID_CALENDARS in new_ntfs_zip.namelist()
assert calendar_handler.GRID_PERIODS in new_ntfs_zip.namelist()
assert calendar_handler.GRID_CALENDAR_REL_LINE in new_ntfs_zip.namelist()
# we want to find all the non calendar file from the input ntfs in the newly generated one
assert set(ntfs_zip.namelist()) == {s for s in new_ntfs_zip.namelist() if not s.startswith('grid_')}
def test_merge_ntfs_no_calendar_file(ntfs_zip):
calendars_zip = ZipFile(BytesIO(), 'a', ZIP_DEFLATED, False)
calendars_zip.writestr('foo.txt', 'foo')
grid_calendar_data = GridCalendarData()
grid_calendar_data.load_zips(calendars_zip, ntfs_zip)
new_ntfs_zip = calendar_handler.merge_calendars_ntfs(grid_calendar_data, ntfs_zip)
assert calendar_handler.GRID_CALENDARS not in new_ntfs_zip.namelist()
assert calendar_handler.GRID_PERIODS not in new_ntfs_zip.namelist()
assert calendar_handler.GRID_CALENDAR_REL_LINE not in new_ntfs_zip.namelist()
# we want to find all the non calendar file from the input ntfs in the newly generated one
assert set(ntfs_zip.namelist()) == {s for s in new_ntfs_zip.namelist() if not s.startswith('grid_')}
def test_merge_ntfs_calendar_file(ntfs_zip, calendars_zip):
grid_calendar_data = GridCalendarData()
grid_calendar_data.load_zips(calendars_zip, ntfs_zip)
new_ntfs_zip = calendar_handler.merge_calendars_ntfs(
grid_calendar_data, ntfs_zip)
assert calendar_handler.GRID_CALENDARS in new_ntfs_zip.namelist()
assert calendar_handler.GRID_PERIODS in new_ntfs_zip.namelist()
assert calendar_handler.GRID_CALENDAR_REL_LINE in new_ntfs_zip.namelist()
# we want to find all the non calendar file from the input ntfs in the newly generated one
assert set(ntfs_zip.namelist()) == {
s
for s in new_ntfs_zip.namelist() if not s.startswith('grid_')
}
def test_merge_ntfs_no_calendar_file(ntfs_zip):
calendars_zip = ZipFile(BytesIO(), 'a', ZIP_DEFLATED, False)
calendars_zip.writestr('foo.txt', 'foo')
grid_calendar_data = GridCalendarData()
grid_calendar_data.load_zips(calendars_zip, ntfs_zip)
new_ntfs_zip = calendar_handler.merge_calendars_ntfs(
grid_calendar_data, ntfs_zip)
assert calendar_handler.GRID_CALENDARS not in new_ntfs_zip.namelist()
assert calendar_handler.GRID_PERIODS not in new_ntfs_zip.namelist()
assert calendar_handler.GRID_CALENDAR_REL_LINE not in new_ntfs_zip.namelist(
)
# we want to find all the non calendar file from the input ntfs in the newly generated one
assert set(ntfs_zip.namelist()) == {
s
for s in new_ntfs_zip.namelist() if not s.startswith('grid_')
}
def _do_merge_calendar(calendar_file, ntfs_file, output_file):
with ZipFile(calendar_file, 'r') as calendars_zip, ZipFile(ntfs_file, 'r') as ntfs_zip:
grid_calendar_data = GridCalendarData()
grid_calendar_data.load_zips(calendars_zip, ntfs_zip)
new_ntfs_zip = calendar_handler.merge_calendars_ntfs(grid_calendar_data, ntfs_zip)
calendar_handler.save_zip_as_file(new_ntfs_zip, output_file)