09-plugin-auth-acl-sub.py 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. #!/usr/bin/env python3
  2. # Test topic subscription. All topic are allowed but not using wildcard in subscribe.
  3. from mosq_test_helper import *
  4. def write_config(filename, port, plugin_ver):
  5. with open(filename, 'w') as f:
  6. f.write("port %d\n" % (port))
  7. f.write("auth_plugin c/auth_plugin_v%d.so\n" % (plugin_ver))
  8. f.write("allow_anonymous false\n")
  9. def do_test(plugin_ver):
  10. port = mosq_test.get_port()
  11. conf_file = os.path.basename(__file__).replace('.py', '.conf')
  12. write_config(conf_file, port, plugin_ver)
  13. rc = 1
  14. keepalive = 10
  15. connect_packet = mosq_test.gen_connect("connect-uname-pwd-test", keepalive=keepalive, username="readonly")
  16. connack_packet = mosq_test.gen_connack(rc=0)
  17. mid = 53
  18. subscribe_packet = mosq_test.gen_subscribe(mid, "qos0/test", 0)
  19. suback_packet = mosq_test.gen_suback(mid, 0)
  20. mid_fail = 54
  21. subscribe_packet_fail = mosq_test.gen_subscribe(mid_fail, "#", 0)
  22. suback_packet_fail = mosq_test.gen_suback(mid_fail, 0x80)
  23. broker = mosq_test.start_broker(filename=os.path.basename(__file__), use_conf=True, port=port)
  24. try:
  25. sock = mosq_test.do_client_connect(connect_packet, connack_packet, timeout=20, port=port)
  26. mosq_test.do_send_receive(sock, subscribe_packet, suback_packet, "suback")
  27. mosq_test.do_send_receive(sock, subscribe_packet_fail, suback_packet_fail, "suback")
  28. rc = 0
  29. sock.close()
  30. except mosq_test.TestError:
  31. pass
  32. finally:
  33. os.remove(conf_file)
  34. broker.terminate()
  35. broker.wait()
  36. (stdo, stde) = broker.communicate()
  37. if rc:
  38. print(stde.decode('utf-8'))
  39. exit(rc)
  40. do_test(4)
  41. do_test(5)