def test_event_handler_failure():
# eeprom_request_bin = binascii.unhexlify('500800009f004037')
eeprom_response_bin = binascii.unhexlify(
'524700009f0041133e001e0e0400000000060a0000000000000000000000000000000000000000000000000000000000000000000000000000000000000000121510010705004e85'
)
eh = EventMessageHandler(lambda m: "beer")
eh.handle = mock.MagicMock()
loop = asyncio.get_event_loop()
mh = AsyncMessageManager(loop)
mh.register_handler(eh)
assert 1 == len(mh.handlers)
message = ReadEEPROMResponse.parse(eeprom_response_bin)
coro = asyncio.ensure_future(mh._handle_message(message))
loop.run_until_complete(coro)
assert coro.result(
) is None # failed to parse response message return None. Maybe needs to throw something.
assert 1 == len(mh.handlers)
eh.handle.assert_not_called()
def test_handler_two_messages():
def event_handler(message):
print("event")
return "event"
async def get_eeprom_result(mhm):
return await mhm.wait_for_message(
lambda m: m.fields.value.po.command == 0x5)
event_response_bin = b'\xe2\xff\xad\x06\x14\x13\x01\x04\x0e\x10\x00\x01\x05\x00\x00\x00\x00\x00\x02Living room \x00\xcc'
eeprom_response_bin = binascii.unhexlify(
'524700009f0041133e001e0e0400000000060a0000000000000000000000000000000000000000000000000000000000000000000000000000000000000000121510010705004e85'
)
loop = asyncio.get_event_loop()
mh = AsyncMessageManager(loop)
event_handler = EventMessageHandler(event_handler)
mh.register_handler(event_handler)
# running
task_handle_wait = loop.create_task(asyncio.sleep(0.1))
task_get_eeprom = loop.create_task(get_eeprom_result(mh))
task_handle_event1 = loop.create_task(
mh._handle_message(LiveEvent.parse(event_response_bin)))
loop.create_task(
mh._handle_message(ReadEEPROMResponse.parse(eeprom_response_bin)))
task_handle_event2 = loop.create_task(
mh._handle_message(LiveEvent.parse(event_response_bin)))
# assert 2 == len(mh.handlers)
loop.run_until_complete(
asyncio.gather(task_handle_wait, task_get_eeprom, loop=loop))
assert 1 == len(mh.handlers)
assert task_handle_event1.result(
) == "event" # failed to parse response message return None. Maybe needs to throw something.
assert isinstance(task_get_eeprom.result(), Container)
assert task_handle_event2.result() == "event"
assert 1 == len(mh.handlers)
def test_event_handler():
eh = EventMessageHandler(lambda m: "beer")
loop = asyncio.get_event_loop()
mh = AsyncMessageManager(loop)
mh.register_handler(eh)
assert 1 == len(mh.handlers)
payload = b'\xe2\xff\xad\x06\x14\x13\x01\x04\x0e\x10\x00\x01\x05\x00\x00\x00\x00\x00\x02Living room \x00\xcc'
message = LiveEvent.parse(payload)
coro = asyncio.ensure_future(mh._handle_message(message))
loop.run_until_complete(coro)
assert "beer" == coro.result()
assert 1 == len(mh.handlers)