02-subpub-qos1.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. #!/usr/bin/env python3
  2. # Test whether a client subscribed to a topic receives its own message sent to that topic.
  3. from mosq_test_helper import *
  4. def do_test(proto_ver):
  5. rc = 1
  6. mid = 530
  7. keepalive = 60
  8. connect_packet = mosq_test.gen_connect("subpub-qos1-test", keepalive=keepalive, proto_ver=proto_ver)
  9. connack_packet = mosq_test.gen_connack(rc=0, proto_ver=proto_ver)
  10. subscribe_packet = mosq_test.gen_subscribe(mid, "subpub/qos1", 1, proto_ver=proto_ver)
  11. suback_packet = mosq_test.gen_suback(mid, 1, proto_ver=proto_ver)
  12. mid = 300
  13. publish_packet = mosq_test.gen_publish("subpub/qos1", qos=1, mid=mid, payload="message", proto_ver=proto_ver)
  14. puback_packet = mosq_test.gen_puback(mid, proto_ver=proto_ver)
  15. mid = 1
  16. publish_packet2 = mosq_test.gen_publish("subpub/qos1", qos=1, mid=mid, payload="message", proto_ver=proto_ver)
  17. port = mosq_test.get_port()
  18. broker = mosq_test.start_broker(filename=os.path.basename(__file__), port=port)
  19. try:
  20. sock = mosq_test.do_client_connect(connect_packet, connack_packet, timeout=20, port=port)
  21. mosq_test.do_send_receive(sock, subscribe_packet, suback_packet, "suback")
  22. sock.send(publish_packet)
  23. mosq_test.receive_unordered(sock, puback_packet, publish_packet2, "puback/publish2")
  24. rc = 0
  25. sock.close()
  26. except mosq_test.TestError:
  27. pass
  28. finally:
  29. broker.terminate()
  30. broker.wait()
  31. (stdo, stde) = broker.communicate()
  32. if rc:
  33. print(stde.decode('utf-8'))
  34. print("proto_ver=%d" % (proto_ver))
  35. exit(rc)
  36. do_test(proto_ver=4)
  37. do_test(proto_ver=5)
  38. exit(0)