def test_register_type(registry):
class TestProvider:
pass
payload.register_payload(Payload, TestProvider)
p = payload.get_payload(TestProvider())
assert isinstance(p, Payload)
def test_register_type(registry) -> None:
class TestProvider:
pass
payload.register_payload(Payload, TestProvider)
p = payload.get_payload(TestProvider())
assert isinstance(p, Payload)
def test_register_unsupported_order(registry: Any) -> None:
class TestProvider:
pass
with pytest.raises(ValueError):
payload.register_payload(Payload, TestProvider,
order=object()) # type: ignore
def function1023(function332):
class Class57:
pass
payload.register_payload(Class409, Class57)
var674 = payload.get_payload(Class57())
assert isinstance(var674, Class409)
def test_register_unsupported_order(registry):
class TestProvider:
pass
with pytest.raises(ValueError):
payload.register_payload(Payload, TestProvider, order=object())
import asyncio
from functools import partial
from tempfile import SpooledTemporaryFile
import aiobotocore
import aiofiles
from aiohttp.payload import register_payload, BytesIOPayload
from aiofiles.threadpool import wrap as aio_wrap, AsyncFileIO
from . import settings
from .base import AbstractStorage
register_payload(BytesIOPayload, SpooledTemporaryFile)
@aio_wrap.register(SpooledTemporaryFile)
def _(file, *, loop=None, executor=None):
return AsyncFileIO(file, loop, executor)
class MinioStorage(AbstractStorage):
spool_size = 1024 * 1024
def __init__(self, bucket, **kwargs):
super().__init__(**kwargs)
self.bucket = bucket
self._session = aiobotocore.get_session()
@property
def _create_client(self):
return partial(
