diff options
| -rw-r--r-- | tests/modules/socket/tcp/echo_client.lua | 13 | ||||
| -rw-r--r-- | tests/modules/socket/tcp/echo_server.lua | 15 |
2 files changed, 24 insertions, 4 deletions
diff --git a/tests/modules/socket/tcp/echo_client.lua b/tests/modules/socket/tcp/echo_client.lua index 7c88a5992..f3d6d9bae 100644 --- a/tests/modules/socket/tcp/echo_client.lua +++ b/tests/modules/socket/tcp/echo_client.lua @@ -6,7 +6,16 @@ function main() print("connect %s:%d ..", addr, port) local sock = socket.connect(addr, port) print("%s: connected!", sock) - local send = sock:send("hello world..", {block = true}) - print("%s: send %d", sock, send) + local count = 0 + while count < 10000 do + local send = sock:send("hello world..", {block = true}) + if send > 0 then + sock:recv(13, {block = true}) + else + break + end + count = count + 1 + end + print("%s: send ok, count: %d!", sock, count) sock:close() end diff --git a/tests/modules/socket/tcp/echo_server.lua b/tests/modules/socket/tcp/echo_server.lua index 48f22a977..d08868b54 100644 --- a/tests/modules/socket/tcp/echo_server.lua +++ b/tests/modules/socket/tcp/echo_server.lua @@ -11,8 +11,19 @@ function main() local sock_client = sock:accept() if sock_client then print("%s: accepted", sock_client) - local recv, data = sock_client:recv(13, {block = true}) - print("%s: recv %d, data: %s", sock_client, recv, data or "") + local count = 0 + local result = nil + while true do + local recv, data = sock_client:recv(13, {block = true}) + if recv > 0 then + result = data + sock_client:send(data, {block = true}) + count = count + 1 + else + break + end + end + print("%s: recv: %s, count: %d", sock_client, result or "", count) sock_client:close() end end |