import pytest

from zvk.util.zlogging import logger


@pytest.mark.asyncio
async def test_async_gen():
  async def f1():
    logger.info('hi 1')

  async def f2():
    logger.info('hi 2')
    yield 1

  async def f3():
    logger.info('hi 3')
    if False:
      yield 1

  with pytest.raises(Exception):
    async for i in f1():
      assert i == 1

  async for i in f2():
    assert i == 1

  async for i in f3():
    assert i == 1