Add start/stop test

pull/8/head
Nico 2015-09-12 15:28:40 +02:00
rodzic a5c34ed315
commit 52c3b83704
1 zmienionych plików z 18 dodań i 0 usunięć

Wyświetl plik

@ -6,6 +6,7 @@ import asyncio
from hbmqtt.plugins.manager import PluginManager
from hbmqtt.session import Session
from hbmqtt.mqtt.protocol.handler import ProtocolHandler
from hbmqtt.adapters import BufferReader, BufferWriter
class ProtocolHandlerTest(unittest.TestCase):
@ -22,3 +23,20 @@ class ProtocolHandlerTest(unittest.TestCase):
self.assertFalse(handler._pubrec_waiters)
self.assertFalse(handler._pubrel_waiters)
self.assertFalse(handler._pubcomp_waiters)
def test_start_stop(self):
@asyncio.coroutine
def server_coro(reader, writer):
pass
@asyncio.coroutine
def test_coro():
s = Session()
buffer = b''
s.reader = BufferReader(buffer)
s.writer = BufferWriter(buffer)
handler = ProtocolHandler(s, self.plugin_manager, loop=self.loop)
yield from handler.start()
yield from handler.stop()
return handler
handler = self.loop.run_until_complete(test_coro())