tests: ported ngx_http_lua test cases for TCP and UDP cosockets sending boolean and nil values.

Author: thibaultcha

t/058-tcp-socket.t

@@ -4,7 +4,7 @@ use Test::Nginx::Socket::Lua::Stream;
 
 repeat_each(2);
 
-plan tests => repeat_each() * 196;
+plan tests => repeat_each() * 199;
 
 our $HtmlDir = html_dir;
 
@@ -3148,7 +3148,63 @@ failed to connect: bad port number: 65536
 
 
 
-=== TEST 60: TCP socket GC'ed in preread phase without Lua content phase
+=== TEST 60: send boolean and nil
+--- stream_server_config
+    content_by_lua_block {
+        local sock = ngx.socket.tcp()
+        local port = $TEST_NGINX_SERVER_PORT
+
+        local ok, err = sock:connect("127.0.0.1", port)
+        if not ok then
+            ngx.say("failed to connect: ", err)
+            return
+        end
+
+        local function send(data)
+            local bytes, err = sock:send(data)
+            if not bytes then
+                ngx.say("failed to send request: ", err)
+                return
+            end
+        end
+
+        local req = "GET /foo HTTP/1.0\r\nHost: localhost\r\nConnection: close\r\nTest: "
+        send(req)
+        send(true)
+        send(false)
+        send(nil)
+        send("\r\n\r\n")
+
+        while true do
+            local line, err, part = sock:receive()
+            if line then
+                ngx.say("received: ", line)
+            else
+                break
+            end
+        end
+
+        ok, err = sock:close()
+    }
+--- config
+    location /foo {
+        server_tokens off;
+        more_clear_headers Date;
+        echo $http_test;
+    }
+--- stream_response
+received: HTTP/1.1 200 OK
+received: Server: nginx
+received: Content-Type: text/plain
+received: Connection: close
+received: 
+received: truefalsenil
+--- no_error_log
+[error]
+
+
+
+=== TEST 61: TCP socket GC'ed in preread phase without Lua content phase
 --- stream_server_config
     lua_socket_connect_timeout 1s;
     resolver $TEST_NGINX_RESOLVER ipv6=off;

t/087-udp-socket.t

@@ -792,10 +792,9 @@ qr/runtime error: content_by_lua\(nginx\.conf:\d+\):13: bad request/
         else
             ngx.say("peer set")
         end
-
         local function f()
             local sock = test.get_sock()
-            sock:send("a")
+            sock:close()
         end
         ngx.timer.at(0, f)
         ngx.sleep(0.001)
@@ -818,73 +817,71 @@ end
 peer set
 
 --- error_log eval
-qr/runtime error: content_by_lua\(nginx\.conf:\d+\):13: bad request/
+qr/runtime error: content_by_lua\(nginx\.conf:\d+\):12: bad request/
 
 --- no_error_log
 [alert]
 
 
 
-=== TEST 17: bad request tries to receive
+=== TEST 17: the upper bound of port range should be 2^16 - 1
 --- stream_config eval
     "lua_package_path '$::HtmlDir/?.lua;./?.lua;;';"
 --- stream_server_config
     content_by_lua_block {
-        local test = require "test"
-        local sock = test.new_sock()
-        local ok, err = sock:setpeername("127.0.0.1", $TEST_NGINX_MEMCACHED_PORT)
+        local sock = ngx.socket.udp()
+        local ok, err = sock:setpeername("127.0.0.1", 65536)
         if not ok then
-            ngx.say("failed to set peer: ", err)
-        else
-            ngx.say("peer set")
-        end
-        function f()
-            local sock = test.get_sock()
-            sock:close()
+            ngx.say("failed to connect: ", err)
         end
-        ngx.timer.at(0, f)
-        ngx.sleep(0.001)
     }
---- user_files
->>> test.lua
-module("test", package.seeall)
-
-local sock
-
-function new_sock()
-    sock = ngx.socket.udp()
-    return sock
-end
-
-function get_sock()
-    return sock
-end
 --- stream_response
-peer set
-
---- error_log eval
-qr/runtime error: content_by_lua\(nginx\.conf:\d+\):12: bad request/
-
+failed to connect: bad port number: 65536
 --- no_error_log
-[alert]
+[error]
 
 
 
-=== TEST 18: the upper bound of port range should be 2^16 - 1
---- stream_config eval
-    "lua_package_path '$::HtmlDir/?.lua;./?.lua;;';"
+=== TEST 18: send boolean and nil
 --- stream_server_config
     content_by_lua_block {
-        local sock = ngx.socket.udp()
-        local ok, err = sock:setpeername("127.0.0.1", 65536)
+        local socket = ngx.socket
+        local udp = socket.udp()
+        local port = ngx.var.port
+        udp:settimeout(1000) -- 1 sec
+
+        local ok, err = udp:setpeername("127.0.0.1", $TEST_NGINX_MEMCACHED_PORT)
         if not ok then
             ngx.say("failed to connect: ", err)
+            return
         end
+
+        local function send(data)
+            local bytes, err = udp:send(data)
+            if not bytes then
+                ngx.say("failed to send: ", err)
+                return
+            end
+            ngx.say("sent ok")
+        end
+
+        send(true)
+        send(false)
+        send(nil)
     }
 --- stream_response
-failed to connect: bad port number: 65536
+sent ok
+sent ok
+sent ok
 --- no_error_log
 [error]
+--- grep_error_log eval
+qr/sendto: fd:\d+ \d+ of \d+/
+--- grep_error_log_out eval
+qr/sendto: fd:\d+ 4 of 4
+sendto: fd:\d+ 5 of 5
+sendto: fd:\d+ 3 of 3/
+--- log_level: debug