def create_chunks(self, uuid, nb=3, last=None):
for i in range(nb):
storages.chunks.write(chunk_filename(uuid, i), faker.word())
storages.chunks.write(chunk_filename(uuid, META), json.dumps({
'uuid': str(uuid),
'filename': faker.file_name(),
'totalparts': nb + 1,
'lastchunk': last or datetime.now(),
}))
def test_chunks_kept_before_max_retention(self, client):
not_expired = datetime.now()
expired = datetime.now() - timedelta(hours=2)
expired_uuid = str(uuid4())
active_uuid = str(uuid4())
parts = 3
self.create_chunks(expired_uuid, nb=parts, last=expired)
self.create_chunks(active_uuid, nb=parts, last=not_expired)
purge_chunks.apply()
expected = set([chunk_filename(active_uuid, i) for i in range(parts)])
expected.add(chunk_filename(active_uuid, META))
assert set(storages.chunks.list_files()) == expected
def test_chunks_kept_before_max_retention(self, client):
not_expired = datetime.now()
expired = datetime.now() - timedelta(hours=2)
expired_uuid = str(uuid4())
active_uuid = str(uuid4())
parts = 3
self.create_chunks(expired_uuid, nb=parts, last=expired)
self.create_chunks(active_uuid, nb=parts, last=not_expired)
purge_chunks.apply()
expected = set([
chunk_filename(active_uuid, i) for i in range(parts)
])
expected.add(chunk_filename(active_uuid, META))
assert set(storages.chunks.list_files()) == expected
assert not storages.chunks.exists(expired_uuid) # Directory should be removed too
