Tidy up test collection creation

- No need to empty collections, they're generated with a UUID and should
  always be empty (the code to empty them does not actually run in CI
  right now).
- Move deletion code to _after_ the yield, since that's the order in
  which things happen.
- Delete all collections asynchronously.
This commit is contained in:
Hugo Osvaldo Barrera 2021-06-27 13:14:17 +02:00
parent 17f422c1b7
commit 71879045e4

View file

@ -1,7 +1,9 @@
import asyncio
import contextlib import contextlib
import subprocess import subprocess
import time import time
import uuid import uuid
from typing import Type
import aiostream import aiostream
import pytest import pytest
@ -87,26 +89,26 @@ async def slow_create_collection(request, aio_connector):
# storage limits. # storage limits.
to_delete = [] to_delete = []
async def delete_collections(): async def inner(cls: Type, args: dict, collection_name: str) -> dict:
for s in to_delete: """Create a collection
await s.session.request("DELETE", "")
async def inner(cls, args, collection): Returns args necessary to create a Storage instance pointing to it.
assert collection.startswith("test") """
collection += "-vdirsyncer-ci-" + str(uuid.uuid4()) assert collection_name.startswith("test")
args = await cls.create_collection(collection, **args) # Make each name unique
s = cls(**args) collection_name = f"{collection_name}-vdirsyncer-ci-{uuid.uuid4()}"
await _clear_collection(s)
assert not await aiostream.stream.list(s.list()) # Create the collection:
to_delete.append(s) args = await cls.create_collection(collection_name, **args)
collection = cls(**args)
# Keep collection in a list to be deleted once tests end:
to_delete.append(collection)
assert not await aiostream.stream.list(collection.list())
return args return args
yield inner yield inner
await delete_collections() await asyncio.gather(*(c.session.request("DELETE", "") for c in to_delete))
async def _clear_collection(s):
async for href, etag in s.list():
s.delete(href, etag)