def test_preservation_error_already_exists():
with tempfile.TemporaryDirectory() as d:
cache = MultiprocessFileCache(10, dir=d, do_pickle=True)
for i in range(10):
cache.put(i, str(i))
cache.preserve('preserved')
with pytest.raises(ValueError):
cache.preserve('preserved')
cache.close()
def test_preservation_interoperability():
with tempfile.TemporaryDirectory() as d:
cache = MultiprocessFileCache(10, dir=d, do_pickle=True)
for i in range(10):
cache.put(i, str(i))
cache.preserve('preserved')
cache.close()
cache2 = FileCache(10, dir=d, do_pickle=True)
cache2.preload('preserved')
for i in range(10):
assert str(i) == cache2.get(i)
cache2.close()
def test_preservation():
with tempfile.TemporaryDirectory() as d:
cache = MultiprocessFileCache(10, dir=d, do_pickle=True)
for i in range(10):
cache.put(i, str(i))
cache.preserve('preserved')
cache.close()
# Imitating a new process, fresh load
cache2 = MultiprocessFileCache(10, dir=d, do_pickle=True)
cache2.preload('preserved')
for i in range(10):
assert str(i) == cache2.get(i)
cache2.close()
# No temporary cache file should remain,
# and the preserved cache should be kept.
assert os.listdir(d) == ['preserved.cached', 'preserved.cachei']