tests: ported ngx_http_lua test cases for 'tcpsock:receiveany()' API of upstream TCP cosockets.

Author: thibaultcha

t/058-tcp-socket.t

@@ -4,7 +4,7 @@ use Test::Nginx::Socket::Lua::Stream;
 
 repeat_each(2);
 
-plan tests => repeat_each() * 199;
+plan tests => repeat_each() * 219;
 
 our $HtmlDir = html_dir;
 
@@ -3238,3 +3238,291 @@ connected: 1
 --- error_log
 cleanup lua tcp socket request
 GC cycle done
+
+
+
+=== TEST 62: receiveany method in cosocket
+--- config
+    location = /foo {
+        server_tokens off;
+
+        content_by_lua_block {
+            local resp = {
+                '1',
+                '22',
+                'hello world',
+            }
+
+            local length = 0
+            for _, v in ipairs(resp) do
+                length = length + #v
+            end
+
+            -- flush http header
+            ngx.header['Content-Length'] = length
+            ngx.flush(true)
+            ngx.sleep(0.01)
+
+            -- send http body
+            for _, v in ipairs(resp) do
+                ngx.print(v)
+                ngx.flush(true)
+                ngx.sleep(0.01)
+            end
+        }
+    }
+--- stream_server_config
+    content_by_lua_block {
+        local sock = ngx.socket.tcp()
+        sock:settimeout(500)
+
+        assert(sock:connect("127.0.0.1", $TEST_NGINX_SERVER_PORT))
+        local req = {
+            'GET /foo HTTP/1.0\r\n',
+            'Host: localhost\r\n',
+            'Connection: close\r\n\r\n',
+        }
+        local ok, err = sock:send(req)
+        if not ok then
+            ngx.say("send request failed: ", err)
+            return
+        end
+
+        -- skip http header
+        while true do
+            local data, err, _ = sock:receive('*l')
+            if err then
+                ngx.say('unexpected error occurs when receiving http head: ', err)
+                return
+            end
+
+            if #data == 0 then -- read last line of head
+                break
+            end
+        end
+
+        -- receive http body
+        while true do
+            local data, err = sock:receiveany(1024)
+            if err then
+                if err ~= 'closed' then
+                    ngx.say('unexpected err: ', err)
+                end
+                break
+            end
+            ngx.say(data)
+        end
+
+        sock:close()
+    }
+--- stream_response
+1
+22
+hello world
+--- no_error_log
+[error]
+--- error_log
+lua tcp socket read any
+
+
+
+=== TEST 63: receiveany send data after read side closed
+--- stream_server_config
+    content_by_lua_block {
+        local sock = ngx.socket.tcp()
+        sock:settimeout(500)
+        assert(sock:connect("127.0.0.1", 7658))
+
+        while true do
+            local data, err = sock:receiveany(1024)
+            if err then
+                if err ~= 'closed' then
+                    ngx.say('unexpected err: ', err)
+                    break
+                end
+
+                local data = "send data after read side closed"
+                local bytes, err = sock:send(data)
+                if not bytes then
+                    ngx.say(err)
+                end
+
+                break
+            end
+            ngx.say(data)
+        end
+
+        sock:close()
+    }
+--- tcp_listen: 7658
+--- tcp_shutdown: 1
+--- tcp_query eval: "send data after read side closed"
+--- tcp_query_len: 32
+--- stream_response
+--- no_error_log
+[error]
+
+
+
+=== TEST 64: receiveany with limited, max <= 0
+--- stream_server_config
+    content_by_lua_block {
+        local sock = ngx.socket.tcp()
+        sock:settimeout(500)
+        assert(sock:connect("127.0.0.1", $TEST_NGINX_SERVER_PORT))
+
+        local function receiveany_say_err(...)
+            local ok, err = pcall(sock.receiveany, sock, ...)
+            if not ok then
+                ngx.say(err)
+            end
+        end
+
+
+        receiveany_say_err(0)
+        receiveany_say_err(-1)
+        receiveany_say_err()
+        receiveany_say_err(nil)
+    }
+--- stream_response
+bad argument #2 to '?' (bad max argument)
+bad argument #2 to '?' (bad max argument)
+expecting 2 arguments (including the object), but got 1
+bad argument #2 to '?' (bad max argument)
+--- no_error_log
+[error]
+
+
+
+=== TEST 65: receiveany with limited, max is larger than data
+--- config
+    location = /foo {
+        server_tokens off;
+
+        content_by_lua_block {
+            local resp = 'hello world'
+            local length = #resp
+
+            ngx.header['Content-Length'] = length
+            ngx.flush(true)
+            ngx.sleep(0.01)
+
+            ngx.print(resp)
+        }
+    }
+--- stream_server_config
+    content_by_lua_block {
+        local sock = ngx.socket.tcp()
+        sock:settimeout(500)
+
+        assert(sock:connect("127.0.0.1", $TEST_NGINX_SERVER_PORT))
+        local req = {
+            'GET /foo HTTP/1.0\r\n',
+            'Host: localhost\r\n',
+            'Connection: close\r\n\r\n',
+        }
+        local ok, err = sock:send(req)
+        if not ok then
+            ngx.say("send request failed: ", err)
+            return
+        end
+
+        while true do
+            local data, err, _ = sock:receive('*l')
+            if err then
+                ngx.say('unexpected error occurs when receiving http head: ', err)
+                return
+            end
+
+            if #data == 0 then -- read last line of head
+                break
+            end
+        end
+
+        local data, err = sock:receiveany(128)
+        if err then
+            if err ~= 'closed' then
+                ngx.say('unexpected err: ', err)
+            end
+        else
+            ngx.say(data)
+        end
+
+        sock:close()
+    }
+--- stream_response
+hello world
+--- no_error_log
+[error]
+--- error_log
+lua tcp socket calling receiveany() method to read at most 128 bytes
+
+
+
+=== TEST 66: receiveany with limited, max is smaller than data
+--- config
+    location = /foo {
+        server_tokens off;
+
+        content_by_lua_block {
+            local resp = 'hello world'
+            local length = #resp
+
+            ngx.header['Content-Length'] = length
+            ngx.flush(true)
+            ngx.sleep(0.01)
+
+            ngx.print(resp)
+        }
+    }
+--- stream_server_config
+    content_by_lua_block {
+        local sock = ngx.socket.tcp()
+        sock:settimeout(500)
+
+        assert(sock:connect("127.0.0.1", $TEST_NGINX_SERVER_PORT))
+        local req = {
+            'GET /foo HTTP/1.0\r\n',
+            'Host: localhost\r\n',
+            'Connection: close\r\n\r\n',
+        }
+        local ok, err = sock:send(req)
+        if not ok then
+            ngx.say("send request failed: ", err)
+            return
+        end
+
+        while true do
+            local data, err, _ = sock:receive('*l')
+            if err then
+                ngx.say('unexpected error occurs when receiving http head: ', err)
+                return
+            end
+
+            if #data == 0 then -- read last line of head
+                break
+            end
+        end
+
+        while true do
+            local data, err = sock:receiveany(7)
+            if err then
+                if err ~= 'closed' then
+                    ngx.say('unexpected err: ', err)
+                end
+                break
+
+            else
+                ngx.say(data)
+            end
+        end
+
+        sock:close()
+    }
+--- stream_response
+hello w
+orld
+--- no_error_log
+[error]
+--- error_log
+lua tcp socket calling receiveany() method to read at most 7 bytes

t/062-count.t

@@ -238,7 +238,24 @@ worker: 4
 
 
 
-=== TEST 13: entries under the metatable of udp sockets
+=== TEST 13: entries under the metatable of tcp sockets
+--- stream_server_config
+    content_by_lua_block {
+        local n = 0
+        local sock = ngx.socket.tcp()
+        for k, v in pairs(getmetatable(sock)) do
+            n = n + 1
+        end
+        ngx.say("n = ", n)
+    }
+--- stream_response
+n = 14
+--- no_error_log
+[error]
+
+
+
+=== TEST 14: entries under the metatable of udp sockets
 --- stream_server_config
     content_by_lua_block {
         local n = 0
@@ -255,7 +272,7 @@ n = 6
 
 
 
-=== TEST 14: entries under the metatable of req raw sockets
+=== TEST 15: entries under the metatable of req raw sockets
 --- stream_server_config
     content_by_lua_block {
         local n = 0