mirror of
https://github.com/joyieldInc/predixy.git
synced 2026-02-05 01:42:24 +08:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# Exercise server write mismatch handling to ensure proxy stays alive
|
|
#
|
|
|
|
import argparse
|
|
import socket
|
|
import sys
|
|
import time
|
|
import redis
|
|
|
|
|
|
def run_test(host, port):
|
|
# Send a malformed pipelined request and close quickly.
|
|
try:
|
|
sock = socket.create_connection((host, port), timeout=1.0)
|
|
sock.sendall(b"*2\r\n$4\r\nping\r\n$4\r\nping\r\n")
|
|
sock.close()
|
|
except Exception as exc:
|
|
print("WARN: socket setup failed:", exc)
|
|
|
|
# Ensure the proxy still accepts connections.
|
|
try:
|
|
c = redis.StrictRedis(host=host, port=port)
|
|
if c.ping() is not True:
|
|
print("FAIL: ping did not return True")
|
|
return False
|
|
except Exception as exc:
|
|
print("FAIL: ping after malformed request:", exc)
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(conflict_handler='resolve', description="Null response handling test")
|
|
parser.add_argument("-h", "--host", default="127.0.0.1")
|
|
parser.add_argument("-p", "--port", type=int, default=7617)
|
|
args = parser.parse_args()
|
|
|
|
if run_test(args.host, args.port):
|
|
print("PASS: null response handling")
|
|
sys.exit(0)
|
|
print("FAIL: null response handling")
|
|
sys.exit(1)
|