Skip to content

Commit

Permalink
Further test fixing
Browse files Browse the repository at this point in the history
  • Loading branch information
adamcharnock committed Apr 17, 2018
1 parent 1e50225 commit c5de866
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 3 deletions.
3 changes: 3 additions & 0 deletions lightbus/bus.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
RedisResultTransport, RedisEventTransport
from lightbus.transports.base import SchemaTransport, TransportRegistry
from lightbus.transports.redis import RedisSchemaTransport
from lightbus.utilities.config import random_name
from lightbus.utilities.frozendict import frozendict
from lightbus.utilities.human import human_time
from lightbus.utilities.async import handle_aio_exceptions, block, get_event_loop, cancel
Expand Down Expand Up @@ -380,6 +381,8 @@ async def listen_for_events(self,
self._validate_name(api_name, 'event', name)

options = options or {}
# Set a random default name for this new consumer we are creating
options.setdefault('name', random_name(length=4))
listener_context = {}

async def listen_for_event_task(event_transport, events):
Expand Down
3 changes: 2 additions & 1 deletion lightbus/transports/debug.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,14 +60,15 @@ async def send_event(self, event_message: EventMessage, options: dict):
event_message.kwargs
))

async def fetch(self, listen_for: List[Tuple[str, str]], context: dict, **kwargs) -> Generator[EventMessage, None, None]:
async def fetch(self, listen_for: List[Tuple[str, str]], context: dict, name: str, **kwargs) -> Generator[EventMessage, None, None]:
"""Consume RPC events for the given API"""

logger.info("⌛ Faking listening for events {}.".format(self._events))

while True:
await asyncio.sleep(0.1)
yield self._get_fake_message()
yield True

def _get_fake_message(self):
return EventMessage(api_name='my_company.auth', event_name='user_registered', kwargs={'example': 'value'})
Expand Down
2 changes: 1 addition & 1 deletion tests/redis_transports/test_unit_redis_events.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@ async def test_consume_events_since_datetime(redis_event_transport: RedisEventTr
consumer = redis_event_transport.consume([('my.dummy', 'my_event')], {}, since=since_datetime, forever=False)

yields = []

async def co():
async for m in consumer:
yields.append(m)
Expand Down
2 changes: 1 addition & 1 deletion tests/test_transports_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ def test_transport_registry_get_rpc_transports(redis_default_config):
def test_transport_registry_get_event_transports(redis_default_config):
registry = TransportRegistry().load_config(redis_default_config)
debug_transport = DebugEventTransport()
redis_transport = RedisEventTransport(consumer_group_name='foo')
redis_transport = RedisEventTransport(consumer_group_prefix='foo', consumer_name='bar')

registry.set_event_transport('redis1', redis_transport)
registry.set_event_transport('redis2', redis_transport)
Expand Down

0 comments on commit c5de866

Please sign in to comment.