01-connect-575314.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. #!/usr/bin/env python3
  2. # Check for performance of processing user-property on CONNECT
  3. from mosq_test_helper import *
  4. def do_test():
  5. rc = 1
  6. props = mqtt5_props.gen_string_pair_prop(mqtt5_props.PROP_USER_PROPERTY, "key", "value")
  7. for i in range(0, 5000):
  8. props += mqtt5_props.gen_string_pair_prop(mqtt5_props.PROP_USER_PROPERTY, "key", "value")
  9. connect_packet_slow = mosq_test.gen_connect("connect-user-property", proto_ver=5, properties=props)
  10. connect_packet_fast = mosq_test.gen_connect("a"*65000, proto_ver=5)
  11. connack_packet = mosq_test.gen_connack(rc=0, proto_ver=5)
  12. port = mosq_test.get_port()
  13. broker = mosq_test.start_broker(filename=os.path.basename(__file__), port=port)
  14. try:
  15. t_start = time.monotonic()
  16. sock = mosq_test.do_client_connect(connect_packet_slow, connack_packet, port=port)
  17. t_stop = time.monotonic()
  18. sock.close()
  19. t_diff_slow = t_stop - t_start
  20. t_start = time.monotonic()
  21. sock = mosq_test.do_client_connect(connect_packet_fast, connack_packet, port=port)
  22. t_stop = time.monotonic()
  23. sock.close()
  24. t_diff_fast = t_stop - t_start
  25. # 20 is chosen as a factor that works in plain mode and running under
  26. # valgrind. The slow performance manifests as a factor of >100. Fast is <10.
  27. if t_diff_slow / t_diff_fast < 20:
  28. rc = 0
  29. except mosq_test.TestError:
  30. pass
  31. finally:
  32. broker.terminate()
  33. broker.wait()
  34. (stdo, stde) = broker.communicate()
  35. if rc:
  36. print(stde.decode('utf-8'))
  37. exit(rc)
  38. do_test()
  39. exit(0)