def test_unload_infolist_dir_file_conflict(self):
m = DefaultMapper()
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo/')])
m._load_infolist('/tmp/demo2.zip', [zipinfo('demo')])
self.assertEqual(m.mapping, {'demo': {}})
m._unload_infolist('/tmp/demo2.zip')
self.assertEqual(m.mapping, {'demo': {}})
def test_unload_infolist_dir_file_conflict(self):
m = DefaultMapper()
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo/')])
m._load_infolist('/tmp/demo2.zip', [zipinfo('demo')])
self.assertEqual(m.mapping, {'demo': {}})
m._unload_infolist('/tmp/demo2.zip')
self.assertEqual(m.mapping, {'demo': {}})
def test_unload_infolist_dir_unloading(self):
m = DefaultMapper()
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo/demo/1.txt', 1)])
m._load_infolist('/tmp/demo2.zip', [zipinfo('demo/')])
m._load_infolist('/tmp/demo3.zip', [zipinfo('demo')])
m._load_infolist('/tmp/demo4.zip', [zipinfo('demo/demo4.txt', 1)])
self.assertEqual(
m.mapping, {
'demo': {
'demo': {
'1.txt': ('/tmp/demo1.zip', 'demo/demo/1.txt', 1)
},
'demo4.txt': ('/tmp/demo4.zip', 'demo/demo4.txt', 1),
}
})
m._unload_infolist('/tmp/demo1.zip')
self.assertEqual(
m.mapping,
{'demo': {
'demo4.txt': ('/tmp/demo4.zip', 'demo/demo4.txt', 1),
}})
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo/demo/1.txt', 1)])
m._unload_infolist('/tmp/demo4.zip')
self.assertEqual(
m.mapping, {
'demo': {
'demo': {
'1.txt': ('/tmp/demo1.zip', 'demo/demo/1.txt', 1)
},
}
})
def test_unload_infolist_file_dir_conflict(self):
m = DefaultMapper()
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo')])
m._load_infolist('/tmp/demo2.zip', [zipinfo('demo/')])
self.assertEqual(m.mapping, {'demo': ('/tmp/demo1.zip', 'demo', 0)})
m._unload_infolist('/tmp/demo1.zip')
# directory not created.
self.assertEqual(m.mapping, {})
def test_unload_infolist_file_dir_conflict(self):
m = DefaultMapper()
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo')])
m._load_infolist('/tmp/demo2.zip', [zipinfo('demo/')])
self.assertEqual(m.mapping, {'demo': ('/tmp/demo1.zip', 'demo', 0)})
m._unload_infolist('/tmp/demo1.zip')
# directory not created.
self.assertEqual(m.mapping, {})
def test_load_infolist_alternative_pathmaking(self):
demo1 = path('demo1.zip')
demo2 = path('demo2.zip')
m = DefaultMapper(pathmaker_name='flatten', include_arcname=True)
with ZipFile(demo1) as zf:
# loading as demo_demo to conflict with below
m._load_infolist('/tmp/demo_demo', zf.infolist())
with ZipFile(demo2) as zf:
m._load_infolist('/tmp/demo', zf.infolist())
self.assertEqual({k: list(v) for k, v in m.reverse_mapping.items()}, {
'': [('/tmp/demo', 'demo/', 0)],
'demo_demo_file1': [('/tmp/demo_demo', 'file1', 33),
('/tmp/demo', 'demo/file1', 33)],
'demo_demo_file2': [('/tmp/demo_demo', 'file2', 33),
('/tmp/demo', 'demo/file2', 33)],
'demo_demo_file3': [('/tmp/demo_demo', 'file3', 33),
('/tmp/demo', 'demo/file3', 33)],
'demo_demo_file4': [('/tmp/demo_demo', 'file4', 33),
('/tmp/demo', 'demo/file4', 33)],
'demo_demo_file5': [('/tmp/demo_demo', 'file5', 33),
('/tmp/demo', 'demo/file5', 33)],
'demo_demo_file6': [('/tmp/demo_demo', 'file6', 33),
('/tmp/demo', 'demo/file6', 33)],
})
self.assertEqual(m.mapping, {
'demo_demo_file1': ('/tmp/demo_demo', 'file1', 33),
'demo_demo_file2': ('/tmp/demo_demo', 'file2', 33),
'demo_demo_file3': ('/tmp/demo_demo', 'file3', 33),
'demo_demo_file4': ('/tmp/demo_demo', 'file4', 33),
'demo_demo_file5': ('/tmp/demo_demo', 'file5', 33),
'demo_demo_file6': ('/tmp/demo_demo', 'file6', 33),
})
m._unload_infolist('/tmp/demo_demo')
self.assertEqual({k: list(v) for k, v in m.reverse_mapping.items()}, {
'': [('/tmp/demo', 'demo/', 0)],
'demo_demo_file1': [('/tmp/demo', 'demo/file1', 33)],
'demo_demo_file2': [('/tmp/demo', 'demo/file2', 33)],
'demo_demo_file3': [('/tmp/demo', 'demo/file3', 33)],
'demo_demo_file4': [('/tmp/demo', 'demo/file4', 33)],
'demo_demo_file5': [('/tmp/demo', 'demo/file5', 33)],
'demo_demo_file6': [('/tmp/demo', 'demo/file6', 33)],
})
self.assertEqual(m.mapping, {
'demo_demo_file1': ('/tmp/demo', 'demo/file1', 33),
'demo_demo_file2': ('/tmp/demo', 'demo/file2', 33),
'demo_demo_file3': ('/tmp/demo', 'demo/file3', 33),
'demo_demo_file4': ('/tmp/demo', 'demo/file4', 33),
'demo_demo_file5': ('/tmp/demo', 'demo/file5', 33),
'demo_demo_file6': ('/tmp/demo', 'demo/file6', 33),
})
def test_unload_infolist_simple(self):
demo1 = path('demo1.zip')
m = DefaultMapper()
with ZipFile(demo1) as zf:
m._load_infolist('/tmp/demo1.zip', zf.infolist())
m._unload_infolist('/tmp/demo1.zip')
self.assertEqual(m.mapping, {})
self.assertEqual(m.reverse_mapping, {})
self.assertEqual(m.archives, {})
self.assertEqual(m.archive_ifilenames, {})
def test_unload_infolist_file_dir_dir_conflict(self):
m = DefaultMapper()
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo')])
m._load_infolist('/tmp/demo2.zip', [zipinfo('demo/')])
m._load_infolist('/tmp/demo3.zip', [zipinfo('demo/demo.txt', 1)])
self.assertEqual(m.mapping, {'demo': ('/tmp/demo1.zip', 'demo', 0)})
m._unload_infolist('/tmp/demo1.zip')
# directory not created. Alternative implementation could have
# added the same entry to a shadow location so it would show up
# here, hide it again when inactive.
self.assertEqual(m.mapping, {})
m._unload_infolist('/tmp/demo3.zip')
self.assertEqual(m.mapping, {})
def test_unload_infolist_simple(self):
demo1 = path('demo1.zip')
m = DefaultMapper()
with ZipFile(demo1) as zf:
m._load_infolist('/tmp/demo1.zip', zf.infolist())
m._unload_infolist('/tmp/demo1.zip')
self.assertEqual(m.mapping, {})
self.assertEqual(m.reverse_mapping, {})
self.assertEqual(m.archives, {})
self.assertEqual(m.archive_ifilenames, {})
def test_unload_infolist_file_dir_dir_conflict(self):
m = DefaultMapper()
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo')])
m._load_infolist('/tmp/demo2.zip', [zipinfo('demo/')])
m._load_infolist('/tmp/demo3.zip', [zipinfo('demo/demo.txt', 1)])
self.assertEqual(m.mapping, {'demo': ('/tmp/demo1.zip', 'demo', 0)})
m._unload_infolist('/tmp/demo1.zip')
# directory not created. Alternative implementation could have
# added the same entry to a shadow location so it would show up
# here, hide it again when inactive.
self.assertEqual(m.mapping, {})
m._unload_infolist('/tmp/demo3.zip')
self.assertEqual(m.mapping, {})
def test_unload_infolist_multiple_reload(self):
demo = path('demo2.zip')
m = DefaultMapper()
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo1.zip', zf.infolist())
m._load_infolist('/tmp/demo2.zip', zf.infolist())
m._load_infolist('/tmp/demo3.zip', zf.infolist())
m._unload_infolist('/tmp/demo1.zip')
d2list = {
'file1': ('/tmp/demo2.zip', 'demo/file1', 33),
'file2': ('/tmp/demo2.zip', 'demo/file2', 33),
'file3': ('/tmp/demo2.zip', 'demo/file3', 33),
'file4': ('/tmp/demo2.zip', 'demo/file4', 33),
'file5': ('/tmp/demo2.zip', 'demo/file5', 33),
'file6': ('/tmp/demo2.zip', 'demo/file6', 33),
}
self.assertEqual(m.mapping, {'demo': d2list})
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo4.zip', zf.infolist())
# Shouldn't overwrite now.
self.assertEqual(m.mapping['demo'], d2list)
m._unload_infolist('/tmp/demo2.zip')
# Shouldn't be affected because they don't belong to demo2.
self.assertEqual(
m.mapping['demo'],
{k: ('/tmp/demo3.zip', v[1], v[2])
for k, v in d2list.items()})
# latest one is unloaded.
self.assertEqual(list(v[0] for v in m.reverse_mapping['demo/']),
['/tmp/demo3.zip', '/tmp/demo4.zip'])
self.assertEqual(sorted(m.archives.keys()),
['/tmp/demo3.zip', '/tmp/demo4.zip'])
self.assertEqual(sorted(m.archive_ifilenames['/tmp/demo3.zip']), [
'demo/', 'demo/file1', 'demo/file2', 'demo/file3', 'demo/file4',
'demo/file5', 'demo/file6'
])
m._unload_infolist('/tmp/demo4.zip')
self.assertEqual(
m.mapping['demo'],
{k: ('/tmp/demo3.zip', v[1], v[2])
for k, v in d2list.items()})
# everything still removed, at last.
m._unload_infolist('/tmp/demo3.zip')
self.assertEqual(m.mapping, {})
def test_unload_infolist_multiple_reload(self):
demo = path('demo2.zip')
m = DefaultMapper()
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo1.zip', zf.infolist())
m._load_infolist('/tmp/demo2.zip', zf.infolist())
m._load_infolist('/tmp/demo3.zip', zf.infolist())
m._unload_infolist('/tmp/demo1.zip')
d2list = {
'file1': ('/tmp/demo2.zip', 'demo/file1', 33),
'file2': ('/tmp/demo2.zip', 'demo/file2', 33),
'file3': ('/tmp/demo2.zip', 'demo/file3', 33),
'file4': ('/tmp/demo2.zip', 'demo/file4', 33),
'file5': ('/tmp/demo2.zip', 'demo/file5', 33),
'file6': ('/tmp/demo2.zip', 'demo/file6', 33),
}
self.assertEqual(m.mapping, {'demo': d2list})
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo4.zip', zf.infolist())
# Shouldn't overwrite now.
self.assertEqual(m.mapping['demo'], d2list)
m._unload_infolist('/tmp/demo2.zip')
# Shouldn't be affected because they don't belong to demo2.
self.assertEqual(m.mapping['demo'], {
k: ('/tmp/demo3.zip', v[1], v[2]) for k, v in d2list.items()
})
# latest one is unloaded.
self.assertEqual(list(v[0] for v in m.reverse_mapping['demo/']),
['/tmp/demo3.zip', '/tmp/demo4.zip'])
self.assertEqual(sorted(m.archives.keys()),
['/tmp/demo3.zip', '/tmp/demo4.zip'])
self.assertEqual(
sorted(m.archive_ifilenames['/tmp/demo3.zip']), [
'demo/', 'demo/file1', 'demo/file2', 'demo/file3',
'demo/file4', 'demo/file5', 'demo/file6'
])
m._unload_infolist('/tmp/demo4.zip')
self.assertEqual(m.mapping['demo'], {
k: ('/tmp/demo3.zip', v[1], v[2]) for k, v in d2list.items()
})
# everything still removed, at last.
m._unload_infolist('/tmp/demo3.zip')
self.assertEqual(m.mapping, {})
def test_unload_infolist_dir_unloading(self):
m = DefaultMapper()
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo/demo/1.txt', 1)])
m._load_infolist('/tmp/demo2.zip', [zipinfo('demo/')])
m._load_infolist('/tmp/demo3.zip', [zipinfo('demo')])
m._load_infolist('/tmp/demo4.zip', [zipinfo('demo/demo4.txt', 1)])
self.assertEqual(m.mapping, {'demo': {
'demo': {'1.txt': ('/tmp/demo1.zip', 'demo/demo/1.txt', 1)},
'demo4.txt': ('/tmp/demo4.zip', 'demo/demo4.txt', 1),
}})
m._unload_infolist('/tmp/demo1.zip')
self.assertEqual(m.mapping, {'demo': {
'demo4.txt': ('/tmp/demo4.zip', 'demo/demo4.txt', 1),
}})
m._load_infolist('/tmp/demo1.zip', [zipinfo('demo/demo/1.txt', 1)])
m._unload_infolist('/tmp/demo4.zip')
self.assertEqual(m.mapping, {'demo': {
'demo': {'1.txt': ('/tmp/demo1.zip', 'demo/demo/1.txt', 1)},
}})
def test_load_infolist_alternative_pathmaking(self):
demo1 = path('demo1.zip')
demo2 = path('demo2.zip')
m = DefaultMapper(pathmaker_name='flatten', include_arcname=True)
with ZipFile(demo1) as zf:
# loading as demo_demo to conflict with below
m._load_infolist('/tmp/demo_demo', zf.infolist())
with ZipFile(demo2) as zf:
m._load_infolist('/tmp/demo', zf.infolist())
self.assertEqual(
{k: list(v)
for k, v in m.reverse_mapping.items()}, {
'': [('/tmp/demo', 'demo/', 0)],
'demo_demo_file1': [('/tmp/demo_demo', 'file1', 33),
('/tmp/demo', 'demo/file1', 33)],
'demo_demo_file2': [('/tmp/demo_demo', 'file2', 33),
('/tmp/demo', 'demo/file2', 33)],
'demo_demo_file3': [('/tmp/demo_demo', 'file3', 33),
('/tmp/demo', 'demo/file3', 33)],
'demo_demo_file4': [('/tmp/demo_demo', 'file4', 33),
('/tmp/demo', 'demo/file4', 33)],
'demo_demo_file5': [('/tmp/demo_demo', 'file5', 33),
('/tmp/demo', 'demo/file5', 33)],
'demo_demo_file6': [('/tmp/demo_demo', 'file6', 33),
('/tmp/demo', 'demo/file6', 33)],
})
self.assertEqual(
m.mapping, {
'demo_demo_file1': ('/tmp/demo_demo', 'file1', 33),
'demo_demo_file2': ('/tmp/demo_demo', 'file2', 33),
'demo_demo_file3': ('/tmp/demo_demo', 'file3', 33),
'demo_demo_file4': ('/tmp/demo_demo', 'file4', 33),
'demo_demo_file5': ('/tmp/demo_demo', 'file5', 33),
'demo_demo_file6': ('/tmp/demo_demo', 'file6', 33),
})
m._unload_infolist('/tmp/demo_demo')
self.assertEqual(
{k: list(v)
for k, v in m.reverse_mapping.items()}, {
'': [('/tmp/demo', 'demo/', 0)],
'demo_demo_file1': [('/tmp/demo', 'demo/file1', 33)],
'demo_demo_file2': [('/tmp/demo', 'demo/file2', 33)],
'demo_demo_file3': [('/tmp/demo', 'demo/file3', 33)],
'demo_demo_file4': [('/tmp/demo', 'demo/file4', 33)],
'demo_demo_file5': [('/tmp/demo', 'demo/file5', 33)],
'demo_demo_file6': [('/tmp/demo', 'demo/file6', 33)],
})
self.assertEqual(
m.mapping, {
'demo_demo_file1': ('/tmp/demo', 'demo/file1', 33),
'demo_demo_file2': ('/tmp/demo', 'demo/file2', 33),
'demo_demo_file3': ('/tmp/demo', 'demo/file3', 33),
'demo_demo_file4': ('/tmp/demo', 'demo/file4', 33),
'demo_demo_file5': ('/tmp/demo', 'demo/file5', 33),
'demo_demo_file6': ('/tmp/demo', 'demo/file6', 33),
})
def test_unload_infolist_multiple(self):
demo = path('demo2.zip')
m = DefaultMapper()
# XXX should just store the zf.infolist() somewhere.
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo1.zip', zf.infolist())
m._load_infolist('/tmp/demo2.zip', zf.infolist())
m._load_infolist('/tmp/demo3.zip', zf.infolist())
self.assertEqual(
list(f[0] for f in m.reverse_mapping['demo/']),
['/tmp/demo1.zip', '/tmp/demo2.zip', '/tmp/demo3.zip']
)
# lazy way to check
old_reverse_mapping = str(m.reverse_mapping)
m._unload_infolist('/tmp/demo2.zip')
# Not the oldest entry
self.assertEqual(m.mapping['demo'], {
'file1': ('/tmp/demo1.zip', 'demo/file1', 33),
'file2': ('/tmp/demo1.zip', 'demo/file2', 33),
'file3': ('/tmp/demo1.zip', 'demo/file3', 33),
'file4': ('/tmp/demo1.zip', 'demo/file4', 33),
'file5': ('/tmp/demo1.zip', 'demo/file5', 33),
'file6': ('/tmp/demo1.zip', 'demo/file6', 33),
})
self.assertEqual(str(m.reverse_mapping), old_reverse_mapping)
self.assertEqual(len(m.archives), 2)
self.assertNotIn('/tmp/demo2.zip', m.archives)
self.assertEqual(
list(m.archive_ifilenames.keys()), list(m.archives.keys()))
with self.assertRaises(KeyError):
m._unload_infolist('/tmp/demo2.zip')
m._unload_infolist('/tmp/demo1.zip')
self.assertEqual(m.mapping, {'demo': {
'file1': ('/tmp/demo3.zip', 'demo/file1', 33),
'file2': ('/tmp/demo3.zip', 'demo/file2', 33),
'file3': ('/tmp/demo3.zip', 'demo/file3', 33),
'file4': ('/tmp/demo3.zip', 'demo/file4', 33),
'file5': ('/tmp/demo3.zip', 'demo/file5', 33),
'file6': ('/tmp/demo3.zip', 'demo/file6', 33),
}})
self.assertEqual(
list(m.reverse_mapping['demo/']),
[('/tmp/demo3.zip', 'demo/', 0)],
)
self.assertEqual(sorted(m.archives.keys()), ['/tmp/demo3.zip'])
self.assertEqual(
sorted(m.archive_ifilenames['/tmp/demo3.zip']), [
'demo/', 'demo/file1', 'demo/file2', 'demo/file3',
'demo/file4', 'demo/file5', 'demo/file6'
])
# finally, unload the final one.
m._unload_infolist('/tmp/demo3.zip')
self.assertEqual(m.mapping, {})
self.assertEqual(m.reverse_mapping, {})
self.assertEqual(m.archives, {})
self.assertEqual(m.archive_ifilenames, {})
def test_unload_infolist_multiple_overwrite(self):
demo = path('demo2.zip')
m = DefaultMapper(overwrite=True)
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo1.zip', zf.infolist())
m._load_infolist('/tmp/demo2.zip', zf.infolist())
m._load_infolist('/tmp/demo3.zip', zf.infolist())
d3list = {
'file1': ('/tmp/demo3.zip', 'demo/file1', 33),
'file2': ('/tmp/demo3.zip', 'demo/file2', 33),
'file3': ('/tmp/demo3.zip', 'demo/file3', 33),
'file4': ('/tmp/demo3.zip', 'demo/file4', 33),
'file5': ('/tmp/demo3.zip', 'demo/file5', 33),
'file6': ('/tmp/demo3.zip', 'demo/file6', 33),
}
self.assertEqual(m.mapping, {'demo': d3list})
# Unloading the latest one.
m._unload_infolist('/tmp/demo3.zip')
self.assertEqual(
m.mapping['demo'],
{k: ('/tmp/demo2.zip', v[1], v[2])
for k, v in d3list.items()})
self.assertEqual(list(f[0] for f in m.reverse_mapping['demo/']),
['/tmp/demo1.zip', '/tmp/demo2.zip'])
self.assertEqual(sorted(m.archives.keys()),
['/tmp/demo1.zip', '/tmp/demo2.zip'])
self.assertEqual(sorted(m.archive_ifilenames['/tmp/demo1.zip']), [
'demo/', 'demo/file1', 'demo/file2', 'demo/file3', 'demo/file4',
'demo/file5', 'demo/file6'
])
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo4.zip', zf.infolist())
m._unload_infolist('/tmp/demo1.zip')
self.assertEqual(
m.mapping['demo'],
{k: ('/tmp/demo4.zip', v[1], v[2])
for k, v in d3list.items()})
self.assertEqual(
list(f[0] for f in m.reverse_mapping['demo/']),
['/tmp/demo1.zip', '/tmp/demo2.zip', '/tmp/demo4.zip'])
m._unload_infolist('/tmp/demo2.zip')
self.assertEqual(
m.mapping['demo'],
{k: ('/tmp/demo4.zip', v[1], v[2])
for k, v in d3list.items()})
self.assertEqual(
list(f[0] for f in m.reverse_mapping['demo/']),
['/tmp/demo1.zip', '/tmp/demo2.zip', '/tmp/demo4.zip'])
# Finally everything that can be cleaned, cleaned.
m._unload_infolist('/tmp/demo4.zip')
self.assertEqual(m.mapping, {})
self.assertEqual(m.reverse_mapping, {})
self.assertEqual(m.archives, {})
self.assertEqual(m.archive_ifilenames, {})
def test_unload_infolist_multiple(self):
demo = path('demo2.zip')
m = DefaultMapper()
# XXX should just store the zf.infolist() somewhere.
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo1.zip', zf.infolist())
m._load_infolist('/tmp/demo2.zip', zf.infolist())
m._load_infolist('/tmp/demo3.zip', zf.infolist())
self.assertEqual(
list(f[0] for f in m.reverse_mapping['demo/']),
['/tmp/demo1.zip', '/tmp/demo2.zip', '/tmp/demo3.zip'])
# lazy way to check
old_reverse_mapping = str(m.reverse_mapping)
m._unload_infolist('/tmp/demo2.zip')
# Not the oldest entry
self.assertEqual(
m.mapping['demo'], {
'file1': ('/tmp/demo1.zip', 'demo/file1', 33),
'file2': ('/tmp/demo1.zip', 'demo/file2', 33),
'file3': ('/tmp/demo1.zip', 'demo/file3', 33),
'file4': ('/tmp/demo1.zip', 'demo/file4', 33),
'file5': ('/tmp/demo1.zip', 'demo/file5', 33),
'file6': ('/tmp/demo1.zip', 'demo/file6', 33),
})
self.assertEqual(str(m.reverse_mapping), old_reverse_mapping)
self.assertEqual(len(m.archives), 2)
self.assertNotIn('/tmp/demo2.zip', m.archives)
self.assertEqual(list(m.archive_ifilenames.keys()),
list(m.archives.keys()))
with self.assertRaises(KeyError):
m._unload_infolist('/tmp/demo2.zip')
m._unload_infolist('/tmp/demo1.zip')
self.assertEqual(
m.mapping, {
'demo': {
'file1': ('/tmp/demo3.zip', 'demo/file1', 33),
'file2': ('/tmp/demo3.zip', 'demo/file2', 33),
'file3': ('/tmp/demo3.zip', 'demo/file3', 33),
'file4': ('/tmp/demo3.zip', 'demo/file4', 33),
'file5': ('/tmp/demo3.zip', 'demo/file5', 33),
'file6': ('/tmp/demo3.zip', 'demo/file6', 33),
}
})
self.assertEqual(
list(m.reverse_mapping['demo/']),
[('/tmp/demo3.zip', 'demo/', 0)],
)
self.assertEqual(sorted(m.archives.keys()), ['/tmp/demo3.zip'])
self.assertEqual(sorted(m.archive_ifilenames['/tmp/demo3.zip']), [
'demo/', 'demo/file1', 'demo/file2', 'demo/file3', 'demo/file4',
'demo/file5', 'demo/file6'
])
# finally, unload the final one.
m._unload_infolist('/tmp/demo3.zip')
self.assertEqual(m.mapping, {})
self.assertEqual(m.reverse_mapping, {})
self.assertEqual(m.archives, {})
self.assertEqual(m.archive_ifilenames, {})
def test_unload_infolist_multiple_overwrite(self):
demo = path('demo2.zip')
m = DefaultMapper(overwrite=True)
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo1.zip', zf.infolist())
m._load_infolist('/tmp/demo2.zip', zf.infolist())
m._load_infolist('/tmp/demo3.zip', zf.infolist())
d3list = {
'file1': ('/tmp/demo3.zip', 'demo/file1', 33),
'file2': ('/tmp/demo3.zip', 'demo/file2', 33),
'file3': ('/tmp/demo3.zip', 'demo/file3', 33),
'file4': ('/tmp/demo3.zip', 'demo/file4', 33),
'file5': ('/tmp/demo3.zip', 'demo/file5', 33),
'file6': ('/tmp/demo3.zip', 'demo/file6', 33),
}
self.assertEqual(m.mapping, {'demo': d3list})
# Unloading the latest one.
m._unload_infolist('/tmp/demo3.zip')
self.assertEqual(m.mapping['demo'], {
k: ('/tmp/demo2.zip', v[1], v[2]) for k, v in d3list.items()
})
self.assertEqual(list(f[0] for f in m.reverse_mapping['demo/']),
['/tmp/demo1.zip', '/tmp/demo2.zip'])
self.assertEqual(sorted(m.archives.keys()),
['/tmp/demo1.zip', '/tmp/demo2.zip'])
self.assertEqual(
sorted(m.archive_ifilenames['/tmp/demo1.zip']), [
'demo/', 'demo/file1', 'demo/file2', 'demo/file3',
'demo/file4', 'demo/file5', 'demo/file6'
])
with ZipFile(demo) as zf:
m._load_infolist('/tmp/demo4.zip', zf.infolist())
m._unload_infolist('/tmp/demo1.zip')
self.assertEqual(m.mapping['demo'], {
k: ('/tmp/demo4.zip', v[1], v[2]) for k, v in d3list.items()
})
self.assertEqual(
list(f[0] for f in m.reverse_mapping['demo/']),
['/tmp/demo1.zip', '/tmp/demo2.zip', '/tmp/demo4.zip']
)
m._unload_infolist('/tmp/demo2.zip')
self.assertEqual(m.mapping['demo'], {
k: ('/tmp/demo4.zip', v[1], v[2]) for k, v in d3list.items()
})
self.assertEqual(
list(f[0] for f in m.reverse_mapping['demo/']),
['/tmp/demo1.zip', '/tmp/demo2.zip', '/tmp/demo4.zip']
)
# Finally everything that can be cleaned, cleaned.
m._unload_infolist('/tmp/demo4.zip')
self.assertEqual(m.mapping, {})
self.assertEqual(m.reverse_mapping, {})
self.assertEqual(m.archives, {})
self.assertEqual(m.archive_ifilenames, {})