bugfix: Test::Nginx::Socket::Lua::Dgram: we send duplicate UDP reques…

…ts when --- dgram_request is specified.
riveridea · Aug 29, 2016 · 83bbfa4 · 83bbfa4
1 parent c24a52b
commit 83bbfa4
Showing 1 changed file with 40 additions and 19 deletions.
diff --git a/lib/Test/Nginx/Socket/Lua/Dgram.pm b/lib/Test/Nginx/Socket/Lua/Dgram.pm
@@ -101,13 +101,11 @@ _EOC_
 _EOC_
         } else {
             $new_http_server_config .= <<_EOC_;
-                        sock:send('trigger dgram req')
-                        local data, err = sock:receive()
-                        if not data then
-                            sock:close()
-                            ngx.say("receive stream response error: ", err)
-                            return
-                        end
+                    local bytes, err = sock:send('trigger dgram req')
+                    if not bytes then
+                        ngx.say("send stream request error: ", err)
+                        return
+                    end
 _EOC_
         }
 
@@ -123,6 +121,15 @@ _EOC_
             $block->set_value("abort", undef);
         }
 
+        $new_http_server_config .= <<_EOC_;
+                    local data, err = sock:receive()
+                    if not data then
+                        sock:close()
+                        ngx.say("receive stream response error: ", err)
+                        return
+                    end
+_EOC_
+
         if (defined $block->response_body
             || defined $block->response_body_like
             || defined $block->dgram_response
@@ -161,16 +168,23 @@ _EOC_
                         ngx.say("send stream request error: ", err)
                         return
                     end
+_EOC_
+            } else {
+                $new_http_server_config .= <<_EOC_;
+                    local bytes, err = sock:send('trigger dgram req 2')
+                    if not bytes then
+                        ngx.say("send stream request error: ", err)
+                        return
+                    end
 _EOC_
             }
 
             $new_http_server_config .= <<_EOC_;
-                        sock:send('trigger_dgram_req2')
-                        local data, err = sock:receive()
-                        if not data then
-                            ngx.say("receive stream response error: ", err)
-                            return
-                        end
+                    local data, err = sock:receive()
+                    if not data then
+                        ngx.say("receive stream response error: ", err)
+                        return
+                    end
 _EOC_
 
             if (defined $block->response_body
@@ -205,16 +219,23 @@ _EOC_
                             ngx.say("send stream request error: ", err)
                             return
                         end
+_EOC_
+            } else {
+                $new_http_server_config .= <<_EOC_;
+                    local bytes, err = sock:send('trigger dgram req 3')
+                    if not bytes then
+                        ngx.say("send stream request error: ", err)
+                        return
+                    end
 _EOC_
             }
 
             $new_http_server_config .= <<_EOC_;
-                        sock:send('trigger_dgram_req3')
-                        local data, err = sock:receive()
-                        if not data then
-                            ngx.say("receive stream response error: ", err)
-                            return
-                        end
+                    local data, err = sock:receive()
+                    if not data then
+                        ngx.say("receive stream response error: ", err)
+                        return
+                    end
 _EOC_
 
             if (defined $block->response_body