kopia lustrzana https://github.com/Yakifo/amqtt
296 wiersze
11 KiB
Python
296 wiersze
11 KiB
Python
# Copyright (c) 2015 Nicolas JOUANIN
|
|
#
|
|
# See the file license.txt for copying permission.
|
|
import unittest
|
|
import asyncio
|
|
import os
|
|
import logging
|
|
import urllib.request
|
|
import tempfile
|
|
import shutil
|
|
from hbmqtt.client import MQTTClient, ConnectException
|
|
from hbmqtt.broker import Broker
|
|
from hbmqtt.mqtt.constants import QOS_0, QOS_1, QOS_2
|
|
|
|
formatter = "[%(asctime)s] %(name)s {%(filename)s:%(lineno)d} %(levelname)s - %(message)s"
|
|
logging.basicConfig(level=logging.ERROR, format=formatter)
|
|
log = logging.getLogger(__name__)
|
|
|
|
broker_config = {
|
|
'listeners': {
|
|
'default': {
|
|
'type': 'tcp',
|
|
'bind': '127.0.0.1:1883',
|
|
'max_connections': 10
|
|
},
|
|
'ws': {
|
|
'type': 'ws',
|
|
'bind': '127.0.0.1:8080',
|
|
'max_connections': 10
|
|
},
|
|
'wss': {
|
|
'type': 'ws',
|
|
'bind': '127.0.0.1:8081',
|
|
'max_connections': 10
|
|
},
|
|
},
|
|
'sys_interval': 0,
|
|
'auth': {
|
|
'allow-anonymous': True,
|
|
}
|
|
}
|
|
|
|
|
|
class MQTTClientTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.loop = asyncio.new_event_loop()
|
|
asyncio.set_event_loop(self.loop)
|
|
|
|
self.temp_dir = tempfile.mkdtemp(prefix='hbmqtt-test-')
|
|
url = "http://test.mosquitto.org/ssl/mosquitto.org.crt"
|
|
self.ca_file = os.path.join(self.temp_dir, 'mosquitto.org.crt')
|
|
urllib.request.urlretrieve(url, self.ca_file)
|
|
log.info("stored mosquitto cert at %s" % self.ca_file)
|
|
|
|
def tearDown(self):
|
|
self.loop.close()
|
|
shutil.rmtree(self.temp_dir)
|
|
|
|
def test_connect_tcp(self):
|
|
async def test_coro():
|
|
try:
|
|
client = MQTTClient()
|
|
await client.connect('mqtt://test.mosquitto.org/')
|
|
self.assertIsNotNone(client.session)
|
|
await client.disconnect()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_connect_tcp_secure(self):
|
|
async def test_coro():
|
|
try:
|
|
client = MQTTClient(config={'check_hostname': False})
|
|
await client.connect('mqtts://test.mosquitto.org/', cafile=self.ca_file)
|
|
self.assertIsNotNone(client.session)
|
|
await client.disconnect()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_connect_tcp_failure(self):
|
|
async def test_coro():
|
|
try:
|
|
config = {'auto_reconnect': False}
|
|
client = MQTTClient(config=config)
|
|
await client.connect('mqtt://127.0.0.1/')
|
|
except ConnectException as e:
|
|
future.set_result(True)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_connect_ws(self):
|
|
async def test_coro():
|
|
try:
|
|
broker = Broker(broker_config, plugin_namespace="hbmqtt.test.plugins")
|
|
await broker.start()
|
|
client = MQTTClient()
|
|
await client.connect('ws://127.0.0.1:8080/')
|
|
self.assertIsNotNone(client.session)
|
|
await client.disconnect()
|
|
await broker.shutdown()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_reconnect_ws_retain_username_password(self):
|
|
async def test_coro():
|
|
try:
|
|
broker = Broker(broker_config, plugin_namespace="hbmqtt.test.plugins")
|
|
await broker.start()
|
|
client = MQTTClient()
|
|
await client.connect('ws://fred:password@127.0.0.1:8080/')
|
|
self.assertIsNotNone(client.session)
|
|
await client.disconnect()
|
|
await client.reconnect()
|
|
|
|
self.assertIsNotNone(client.session.username)
|
|
self.assertIsNotNone(client.session.password)
|
|
await broker.shutdown()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_connect_ws_secure(self):
|
|
async def test_coro():
|
|
try:
|
|
broker = Broker(broker_config, plugin_namespace="hbmqtt.test.plugins")
|
|
await broker.start()
|
|
client = MQTTClient()
|
|
await client.connect('ws://127.0.0.1:8081/', cafile=self.ca_file)
|
|
self.assertIsNotNone(client.session)
|
|
await client.disconnect()
|
|
await broker.shutdown()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_ping(self):
|
|
async def test_coro():
|
|
try:
|
|
broker = Broker(broker_config, plugin_namespace="hbmqtt.test.plugins")
|
|
await broker.start()
|
|
client = MQTTClient()
|
|
await client.connect('mqtt://127.0.0.1/')
|
|
self.assertIsNotNone(client.session)
|
|
await client.ping()
|
|
await client.disconnect()
|
|
await broker.shutdown()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_subscribe(self):
|
|
async def test_coro():
|
|
try:
|
|
broker = Broker(broker_config, plugin_namespace="hbmqtt.test.plugins")
|
|
await broker.start()
|
|
client = MQTTClient()
|
|
await client.connect('mqtt://127.0.0.1/')
|
|
self.assertIsNotNone(client.session)
|
|
ret = await client.subscribe([
|
|
('$SYS/broker/uptime', QOS_0),
|
|
('$SYS/broker/uptime', QOS_1),
|
|
('$SYS/broker/uptime', QOS_2),
|
|
])
|
|
self.assertEqual(ret[0], QOS_0)
|
|
self.assertEqual(ret[1], QOS_1)
|
|
self.assertEqual(ret[2], QOS_2)
|
|
await client.disconnect()
|
|
await broker.shutdown()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_unsubscribe(self):
|
|
async def test_coro():
|
|
try:
|
|
broker = Broker(broker_config, plugin_namespace="hbmqtt.test.plugins")
|
|
await broker.start()
|
|
client = MQTTClient()
|
|
await client.connect('mqtt://127.0.0.1/')
|
|
self.assertIsNotNone(client.session)
|
|
ret = await client.subscribe([
|
|
('$SYS/broker/uptime', QOS_0),
|
|
])
|
|
self.assertEqual(ret[0], QOS_0)
|
|
await client.unsubscribe(['$SYS/broker/uptime'])
|
|
await client.disconnect()
|
|
await broker.shutdown()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_deliver(self):
|
|
data = b'data'
|
|
|
|
async def test_coro():
|
|
try:
|
|
broker = Broker(broker_config, plugin_namespace="hbmqtt.test.plugins")
|
|
await broker.start()
|
|
client = MQTTClient()
|
|
await client.connect('mqtt://127.0.0.1/')
|
|
self.assertIsNotNone(client.session)
|
|
ret = await client.subscribe([
|
|
('test_topic', QOS_0),
|
|
])
|
|
self.assertEqual(ret[0], QOS_0)
|
|
client_pub = MQTTClient()
|
|
await client_pub.connect('mqtt://127.0.0.1/')
|
|
await client_pub.publish('test_topic', data, QOS_0)
|
|
await client_pub.disconnect()
|
|
message = await client.deliver_message()
|
|
self.assertIsNotNone(message)
|
|
self.assertIsNotNone(message.publish_packet)
|
|
self.assertEqual(message.data, data)
|
|
await client.unsubscribe(['$SYS/broker/uptime'])
|
|
await client.disconnect()
|
|
await broker.shutdown()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|
|
|
|
def test_deliver_timeout(self):
|
|
async def test_coro():
|
|
try:
|
|
broker = Broker(broker_config, plugin_namespace="hbmqtt.test.plugins")
|
|
await broker.start()
|
|
client = MQTTClient()
|
|
await client.connect('mqtt://127.0.0.1/')
|
|
self.assertIsNotNone(client.session)
|
|
ret = await client.subscribe([
|
|
('test_topic', QOS_0),
|
|
])
|
|
self.assertEqual(ret[0], QOS_0)
|
|
with self.assertRaises(asyncio.TimeoutError):
|
|
await client.deliver_message(timeout=2)
|
|
await client.unsubscribe(['$SYS/broker/uptime'])
|
|
await client.disconnect()
|
|
await broker.shutdown()
|
|
future.set_result(True)
|
|
except Exception as ae:
|
|
future.set_exception(ae)
|
|
|
|
future = asyncio.Future(loop=self.loop)
|
|
self.loop.run_until_complete(test_coro())
|
|
if future.exception():
|
|
raise future.exception()
|