07-will-no-flag.py 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. #!/usr/bin/env python3
  2. # Test whether a connection is disconnected if it sets the will flag but does
  3. # not provide a will payload.
  4. from mosq_test_helper import *
  5. def do_test(proto_ver):
  6. rc = 1
  7. keepalive = 10
  8. connect_packet = mosq_test.gen_connect("will-no-payload", keepalive=keepalive, will_topic="will/topic", will_qos=1, will_retain=True, proto_ver=proto_ver)
  9. b = list(struct.unpack("B"*len(connect_packet), connect_packet))
  10. bmod = b[0:len(b)-2]
  11. bmod[1] = bmod[1] - 2 # Reduce remaining length by two to remove final two payload length values
  12. connect_packet = struct.pack("B"*len(bmod), *bmod)
  13. port = mosq_test.get_port()
  14. broker = mosq_test.start_broker(filename=os.path.basename(__file__), port=port)
  15. try:
  16. sock = mosq_test.do_client_connect(connect_packet, b"", port=port)
  17. sock.close()
  18. rc = 0
  19. except mosq_test.TestError:
  20. pass
  21. finally:
  22. broker.terminate()
  23. broker.wait()
  24. (stdo, stde) = broker.communicate()
  25. if rc:
  26. print(stde.decode('utf-8'))
  27. print("proto_ver=%d" % (proto_ver))
  28. exit(rc)
  29. do_test(proto_ver=4)
  30. do_test(proto_ver=5)
  31. exit(0)