import inspect
from unittest import TestCase
from unittest.mock import DEFAULT, Mock, patch


patchers = (
    patch("binascii.b64encode"),
    patch.dict("sys.modules", test_module=Mock()),
    patch.object(Mock, "my_func"),
    patch.multiple(Mock, my_attr=DEFAULT, my_other=DEFAULT),
)

async def coro():
    ...

patched_coros = (patcher(coro) for patcher in patchers)

assert inspect.iscoroutinefunction(coro)
assert all(map(inspect.iscoroutinefunction, patched_coros))


# Coroutines in test classes
for patcher in patchers:
    @patcher
    class TestClass(TestCase):
        async def test_coro(self):
            ...
    assert inspect.iscoroutinefunction(TestClass.test_coro)



# The behavior across patch functions is consistent with async generator functions
async def async_gen_func():
    yield

patched_async_gen_funcs = (patcher(async_gen_func) for patcher in patchers)
assert inspect.isasyncgenfunction(async_gen_func)
assert not all(map(inspect.isasyncgenfunction, patched_async_gen_funcs))