From 2cb5a7d9dc96a654fae533f32bc363568c361a4c Mon Sep 17 00:00:00 2001 From: Nicolas Jouanin Date: Sat, 12 Sep 2015 14:47:36 +0200 Subject: [PATCH] Add init test --- tests/mqtt/protocol/test_handler.py | 25 ++++++++++++++++++++++++- 1 file changed, 24 insertions(+), 1 deletion(-) diff --git a/tests/mqtt/protocol/test_handler.py b/tests/mqtt/protocol/test_handler.py index e1bd617..8b81766 100644 --- a/tests/mqtt/protocol/test_handler.py +++ b/tests/mqtt/protocol/test_handler.py @@ -1 +1,24 @@ -__author__ = 'nico' +# Copyright (c) 2015 Nicolas JOUANIN +# +# See the file license.txt for copying permission. +import unittest +import asyncio +from hbmqtt.plugins.manager import PluginManager +from hbmqtt.session import Session +from hbmqtt.mqtt.protocol.handler import ProtocolHandler + + +class ProtocolHandlerTest(unittest.TestCase): + def setUp(self): + self.loop = asyncio.new_event_loop() + self.plugin_manager = PluginManager("hbmqtt.test.plugins", context=None, loop=self.loop) + + def test_init_handler(self): + s = Session() + handler = ProtocolHandler(s, self.plugin_manager, loop=self.loop) + self.assertIs(handler.session, s) + self.assertIs(handler._loop, self.loop) + self.assertFalse(handler._puback_waiters) + self.assertFalse(handler._pubrec_waiters) + self.assertFalse(handler._pubrel_waiters) + self.assertFalse(handler._pubcomp_waiters)