tests: added three cases for upstream keepalive. (#51)

test 1/2: upstream send extra data.
test 3: upstream close the connection.

Author: zhuizhuhaomeng

.travis.yml

@@ -1,6 +1,10 @@
 sudo: required
 dist: trusty
 
+branches:
+  only:
+    - "plus"
+
 os: linux
 
 language: c

t/balancer-keepalive.t

@@ -1258,3 +1258,169 @@ SNI=one
 qr/^lua balancer: keepalive no free connection, host: 127.0.0.1:23456, name: host1
 lua balancer: keepalive saving connection \S+, host: 127.0.0.1:23456, name: host1
 $/
+
+
+
+=== TEST 29: keepalive idle connection recv data
+--- main_config
+stream {
+  server {
+    listen 60123 ssl;
+    ssl_certificate ../../cert/test.crt;
+    ssl_certificate_key ../../cert/test.key;
+      content_by_lua_block {
+          ngx.say("HTTP/1.1 200 OK")
+          ngx.say("Content-Length: 10")
+          ngx.say("Connection: keep-alive\n")
+          ngx.print("0123456789")
+          ngx.sleep(0.01)
+          ngx.print("trigger")
+      }
+  }
+}
+
+--- http_upstream
+    upstream test_upstream {
+        server 0.0.0.1;
+
+        balancer_by_lua_block {
+            local b = require "ngx.balancer"
+
+            local ok, err = b.set_current_peer("127.0.0.1",
+                                               60123,
+                                               nil,
+                                               ngx.var.arg_host)
+            if not ok then
+                ngx.log(ngx.ERR, "failed to set current peer: ", err)
+                return
+            end
+
+            local ok, err = b.enable_keepalive()
+            if not ok then
+                ngx.log(ngx.ERR, "failed to set keepalive: ", err)
+                return
+            end
+        }
+    }
+--- config
+    location = /t {
+        echo_subrequest GET '/proxy/echo_sni' -q 'sni=one&host=host1';
+    }
+--- wait: 0.2
+--- response_body chomp
+0123456789
+--- grep_error_log_out eval
+qr/^lua balancer: keepalive no free connection, host: 127.0.0.1:60123, name: host1
+lua balancer: keepalive saving connection (\S+?), host: 127.0.0.1:60123, name: host1
+lua balancer: keepalive closing connection \1
+$/
+
+
+
+=== TEST 30: keepalive idle connection recv data
+--- main_config
+stream {
+  server {
+    listen 60123 ssl;
+    ssl_certificate ../../cert/test.crt;
+    ssl_certificate_key ../../cert/test.key;
+      content_by_lua_block {
+          ngx.say("HTTP/1.1 200 OK")
+          ngx.say("Content-Length: 10")
+          ngx.say("Connection: keep-alive\n")
+          ngx.print("012345678912345")
+          ngx.sleep(0.01)
+      }
+  }
+}
+
+--- http_upstream
+    upstream test_upstream {
+        server 0.0.0.1;
+
+        balancer_by_lua_block {
+            local b = require "ngx.balancer"
+
+            local ok, err = b.set_current_peer("127.0.0.1",
+                                               60123,
+                                               nil,
+                                               ngx.var.arg_host)
+            if not ok then
+                ngx.log(ngx.ERR, "failed to set current peer: ", err)
+                return
+            end
+
+            local ok, err = b.enable_keepalive()
+            if not ok then
+                ngx.log(ngx.ERR, "failed to set keepalive: ", err)
+                return
+            end
+        }
+    }
+--- config
+    location = /t {
+        echo_subrequest GET '/proxy/echo_sni' -q 'sni=one&host=host1';
+    }
+--- wait: 0.2
+--- response_body chomp
+0123456789
+--- grep_error_log_out eval
+qr/^lua balancer: keepalive no free connection, host: 127.0.0.1:60123, name: host1
+lua balancer: keepalive not saving connection \S+
+$/
+
+
+
+=== TEST 31: keepalive idle connection recv data
+--- main_config
+stream {
+  server {
+    listen 60123 ssl;
+    ssl_certificate ../../cert/test.crt;
+    ssl_certificate_key ../../cert/test.key;
+      content_by_lua_block {
+          ngx.say("HTTP/1.1 200 OK")
+          ngx.say("Content-Length: 10")
+          ngx.say("Connection: keep-alive\n")
+          ngx.print("0123456789")
+          ngx.flush()
+          ngx.sleep(0.01)
+      }
+  }
+}
+
+--- http_upstream
+    upstream test_upstream {
+        server 0.0.0.1;
+
+        balancer_by_lua_block {
+            local b = require "ngx.balancer"
+
+            local ok, err = b.set_current_peer("127.0.0.1",
+                                               60123,
+                                               nil,
+                                               ngx.var.arg_host)
+            if not ok then
+                ngx.log(ngx.ERR, "failed to set current peer: ", err)
+                return
+            end
+
+            local ok, err = b.enable_keepalive()
+            if not ok then
+                ngx.log(ngx.ERR, "failed to set keepalive: ", err)
+                return
+            end
+        }
+    }
+--- config
+    location = /t {
+        echo_subrequest GET '/proxy/echo_sni' -q 'sni=one&host=host1';
+    }
+--- wait: 0.2
+--- response_body chomp
+0123456789
+--- grep_error_log_out eval
+qr/^lua balancer: keepalive no free connection, host: 127.0.0.1:60123, name: host1
+lua balancer: keepalive saving connection (\S+?), host: 127.0.0.1:60123, name: host1
+lua balancer: keepalive closing connection \1
+$/