需要使用两个知识点，一个是多线程，一个是队列queue。

需要先开一个线程，该线程是订阅topic，然后等待消息。

另外一个线程，就是发布消息。

比较坑的是mqtt的python包，只说了回调函数on_message，但是该回调是通过subscribe函数调用的，无法返回message到主程序。

经过多次尝试，最后想到用queue，在on_message中put message，然后在主函数中get message，再在返回message。

on_message函数将消息写入queue

def on_message(client, userdata, msg):

mss = str((msg.payload).decode('utf-8'))

print(mss)

mp.put(mss)

主函数get后返回给主程序，并且开了两个线程

def mqtt_send_check(client, topic, msg):

client.loop_start()

thread_sub = threading.Thread(target=mqtt_subscribe, args=(client,topic))

thread_sub.start()

thread_pub = threading.Thread(target=mqtt_publish, args=(client,topic,msg))

thread_pub.start()

time.sleep(4)

client.loop_stop()

return(mp.get())

pytest测试程序

class Test_comm:

TOPIC_REPORT = PRODUCT_KEY + "/" + DEVICE_NAME + "/abc"

def test_comm(self):

MSG = "aaa"

client = mqtt_connect(MQTT_SERVER_ADDR, MQTT_SERVER_PORT,PRODUCT_KEY, DEVICE_NAME,DEVICE_SECRET, TIMESTAMP)

pps = mqtt_send_check(client, self.TOPIC_REPORT, MSG)

print('pps:',pps)

pytest程序通过这种方式，可以得到message消息，实现断言了。