mosquitto/test/broker/03-publish-qos2.py
2026-05-03 01:17:26 +01:00

57 lines
1.8 KiB
Python
Executable file

#!/usr/bin/env python3
# Test whether a PUBLISH to a topic with QoS 2 results in the correct packet flow.
from mosq_test_helper import *
def do_test(start_broker, proto_ver):
rc = 1
connect_packet = mosq_test.gen_connect("03-pub-qos2-test", proto_ver=proto_ver)
connack_packet = mosq_test.gen_connack(rc=0, proto_ver=proto_ver)
mid = 312
publish_packet = mosq_test.gen_publish("03/pub/qos2/test", qos=2, mid=mid, payload="message", proto_ver=proto_ver)
pubrec_packet = mosq_test.gen_pubrec(mid, proto_ver=proto_ver)
pubrel_packet = mosq_test.gen_pubrel(mid, proto_ver=proto_ver)
pubcomp_packet = mosq_test.gen_pubcomp(mid, proto_ver=proto_ver)
port = mosq_test.get_port()
if start_broker:
broker = mosq_test.start_broker(filename=os.path.basename(__file__), port=port)
try:
sock = mosq_test.do_client_connect(connect_packet, connack_packet, port=port)
mosq_test.do_send_receive(sock, publish_packet, pubrec_packet, "pubrec")
mosq_test.do_send_receive(sock, pubrel_packet, pubcomp_packet, "pubcomp")
rc = 0
sock.close()
except mosq_test.TestError:
pass
finally:
if start_broker:
mosq_test.terminate_broker(broker)
if mosq_test.wait_for_subprocess(broker):
print("broker not terminated")
if rc == 0: rc=1
if rc:
print(mosq_test.broker_log(broker))
print("proto_ver=%d" % (proto_ver))
exit(rc)
else:
return rc
def all_tests(start_broker=False):
rc = do_test(start_broker, proto_ver=4)
if rc:
return rc;
rc = do_test(start_broker, proto_ver=5)
if rc:
return rc;
return 0
if __name__ == '__main__':
all_tests(True)