import asyncio

import pytest

from zvk.event.consumer import event_consumer
from zvk.event.event import Event
from zvk.event.queue import EventQueue


@event_consumer(consumes=['event_loop'])
async def consumer_loop(event, counter):
  counter[0] += 1
  await asyncio.sleep(0.1)

  yield event


@pytest.mark.asyncio
async def test_event_cancellation():
  event_queue = EventQueue()

  counter = [0]

  event_queue.register_consumer(consumer_loop)

  starting_events = [Event('event_loop', counter=counter)]

  queue_task = asyncio.create_task(event_queue.run(starting_events))

  await asyncio.sleep(0.25)

  event_queue.omae_wa_mou_shindeiru()
  await queue_task

  assert counter[0] == 3