From 5f1f1a082056784b6f900d3554d6740f54ac70cf Mon Sep 17 00:00:00 2001 From: flandre Date: Sat, 18 Jan 2025 21:27:17 +0800 Subject: [PATCH] fix test --- test/testcppzmq.py | 32 +++++++++++++++++++++++++++----- 1 file changed, 27 insertions(+), 5 deletions(-) diff --git a/test/testcppzmq.py b/test/testcppzmq.py index 259d331..2eeced3 100644 --- a/test/testcppzmq.py +++ b/test/testcppzmq.py @@ -5,20 +5,42 @@ import zmq import time import sys -def f0(): + +def exit(): + ctx = zmq.Context() + sock = ctx.socket(zmq.REQ) + sock.connect('tcp://11.6.1.66:5556') + # sock.send(b'file' + Path('/home/lambda/source/scarlet/flandre/config/64-1.txt').read_bytes()) + sock.send(b'exit') + + +def cmd(c: str): + ctx = zmq.Context() + sock = ctx.socket(zmq.REQ) + sock.connect('tcp://11.6.1.66:5556') + sock.send(c.encode()) + + +def file(): ctx = zmq.Context() sock = ctx.socket(zmq.REQ) sock.connect('tcp://11.6.1.66:5556') sock.send(b'file' + Path('/home/lambda/source/scarlet/flandre/config/64-1.txt').read_bytes()) -def f1(): +def test(): ctx = zmq.Context() sock = ctx.socket(zmq.PULL) - sock.connect('tcp://11.6.1.66:5556') + sock.connect('tcp://11.6.1.66:5555') while True: s = sock.recv() - print(struct.unpack('=iqi', s)) + print(struct.unpack_from('=iqi', s)) + print(s[4 + 8 + 4:].__len__()) + break + if __name__ == '__main__': - f1() + if len(sys.argv) < 2: + test() + else: + globals()[sys.argv[1]](*sys.argv[2:])