mirror of
https://github.com/letscontrolit/ESPEasy.git
synced 2026-09-15 19:17:08 +00:00
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
import unittest
|
|
import settings
|
|
import time
|
|
import paho.mqtt.client as mqttclinet
|
|
|
|
|
|
class mqtt_basic(unittest.TestCase):
|
|
def on_message(self, client, userdata, message: mqttclinet.MQTTMessage):
|
|
self.message_queue.append(message)
|
|
message_queue = []
|
|
|
|
@classmethod
|
|
def setUpClass(self):
|
|
self.client = mqttclinet.Client("pubsubclient_ut", clean_session=True)
|
|
self.client.connect(settings.server_ip)
|
|
self.client.on_message = self.on_message
|
|
self.client.subscribe("outTopic", 0)
|
|
|
|
@classmethod
|
|
def tearDownClass(self):
|
|
self.client.disconnect()
|
|
|
|
def test_one(self):
|
|
i = 30
|
|
while len(self.message_queue) == 0 and i > 0:
|
|
self.client.loop()
|
|
time.sleep(0.5)
|
|
i -= 1
|
|
self.assertTrue(i > 0, "message receive timed-out")
|
|
self.assertEqual(len(self.message_queue), 1, "unexpected number of messages received")
|
|
msg = self.message_queue[0]
|
|
self.assertEqual(msg.mid, 0, "message id not 0")
|
|
self.assertEqual(msg.topic, "outTopic", "message topic incorrect")
|
|
self.assertEqual(msg.payload, "hello world")
|
|
self.assertEqual(msg.qos, 0, "message qos not 0")
|
|
self.assertEqual(msg.retain, False, "message retain flag incorrect")
|