def test_delete_by_prefix(self, tmpdir):
storage_dir = str(tmpdir.join("storage"))
os.makedirs(os.path.join(storage_dir, 'foo'), exist_ok=True)
os.makedirs(os.path.join(storage_dir, 'bar'), exist_ok=True)
filename0 = str(tmpdir.join("storage/foo/testfile0.txt"))
with open(filename0, "wb") as fp:
fp.write(b"Zeroth Test File!")
filename1 = str(tmpdir.join("storage/foo/testfile1.txt"))
with open(filename1, "wb") as fp:
fp.write(b"First Test File!")
filename2 = str(tmpdir.join("storage/bar/testfile2.txt"))
with open(filename2, "wb") as fp:
fp.write(b"Second Test File!")
storage = LocalDocsStorage(storage_dir)
storage.remove_by_prefix('foo')
assert not os.path.exists(os.path.join(storage_dir, 'foo'))
with open(os.path.join(storage_dir, "bar/testfile2.txt"), "rb") as fp:
assert fp.read() == b"Second Test File!"
def test_delete_already_gone(self, tmpdir):
storage_dir = str(tmpdir.join("storage"))
storage = LocalDocsStorage(storage_dir)
response = storage.remove_by_prefix('foo')
assert response is None