mirror of
https://github.com/joyieldInc/predixy.git
synced 2026-02-05 01:42:24 +08:00
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# Ensure large pubsub payloads are parsed correctly.
|
|
#
|
|
|
|
from test_util import parse_args, make_clients, exit_with_result
|
|
|
|
|
|
def normalize_bytes(value):
|
|
if isinstance(value, bytes):
|
|
return value.decode("utf-8")
|
|
return value
|
|
|
|
|
|
def wait_for_type(ps, msg_type, attempts=20, timeout=1.0):
|
|
for _ in range(attempts):
|
|
msg = ps.get_message(timeout=timeout)
|
|
if msg and msg.get("type") == msg_type:
|
|
return msg
|
|
return None
|
|
|
|
|
|
def run_test(host, port):
|
|
c1, c2 = make_clients(host, port, count=2)
|
|
|
|
ps = c1.pubsub()
|
|
ps.subscribe("big_payload")
|
|
msg = wait_for_type(ps, "subscribe")
|
|
if not msg:
|
|
print("FAIL: missing subscribe confirmation")
|
|
return False
|
|
|
|
payload = "x" * 10000
|
|
publish_result = c2.publish("big_payload", payload)
|
|
if publish_result < 1:
|
|
print("FAIL: publish did not reach subscribers:", publish_result)
|
|
return False
|
|
|
|
msg = wait_for_type(ps, "message", attempts=30, timeout=1.0)
|
|
if not msg:
|
|
print("FAIL: missing message response")
|
|
return False
|
|
|
|
data = normalize_bytes(msg.get("data"))
|
|
if data != payload:
|
|
print("FAIL: payload mismatch (len)", len(data) if data else 0)
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
args = parse_args("Pubsub large message test")
|
|
success = run_test(args.host, args.port)
|
|
exit_with_result(success, "pubsub large message",
|
|
"pubsub large message")
|