# HG changeset patch
# User Sergey Kandaurov <pluknet@nginx.com>
# Date 1594039131 -10800
# Node ID 14eeaa39599db018d3fb00a84628654a42cfe255
# Parent  f55d25e08b3edaab6f3a30a000ac486334994978
Tests: HTTP/2 tests for lingering close.

diff --git a/h2_max_requests.t b/h2_max_requests.t
--- a/h2_max_requests.t
+++ b/h2_max_requests.t
@@ -12,6 +12,8 @@ use strict;
 
 use Test::More;
 
+use Socket qw(SOL_SOCKET SO_RCVBUF);
+
 BEGIN { use FindBin; chdir($FindBin::Bin); }
 
 use lib 'lib';
@@ -37,7 +39,7 @@ http {
     %%TEST_GLOBALS_HTTP%%
 
     server {
-        listen       127.0.0.1:8080 http2;
+        listen       127.0.0.1:8080 http2 sndbuf=1m;
         server_name  localhost;
 
         http2_max_requests 2;
@@ -48,12 +50,20 @@ http {
 
 EOF
 
-$t->write_file('index.html', '');
-$t->run()->plan(5);
+$t->write_file('index.html', 'SEE-THAT' x 50000);
+$t->run()->plan(10);
 
 ###############################################################################
 
 my $s = Test::Nginx::HTTP2->new();
+
+# to test lingering close, let full response settle down in send buffer space
+# so that client additional data past server-side close would trigger TCP RST
+
+$s->{socket}->setsockopt(SOL_SOCKET, SO_RCVBUF, 64*1024) or die $!;
+$s->h2_settings(0, 0x4 => 2**20);
+$s->h2_window(2**21);
+
 my $frames = $s->read(all => [{ sid => $s->new_stream(), fin => 1 }]);
 
 my ($frame) = grep { $_->{type} eq "HEADERS" } @$frames;
@@ -68,13 +78,67 @@ is($frame, undef, 'max requests - GOAWAY
 # max requests limited
 
 my $sid = $s->new_stream();
+
+# wait server to finish and close socket if lingering close were disabled
+
+select undef, undef, undef, 0.1;
+$s->h2_ping("SEE-THIS");
+
 $frames = $s->read(all => [{ sid => $sid, fin => 1 }, { type => 'GOAWAY' }]);
 
 ($frame) = grep { $_->{type} eq "HEADERS" } @$frames;
 is($frame->{headers}->{':status'}, 200, 'max requests limited');
 
+TODO: {
+local $TODO = 'not yet' if $^O eq 'linux' or $^O eq 'freebsd'
+	and !$t->has_version('1.19.1');
+
+my @data = grep { $_->{type} eq "DATA" } @$frames;
+my $sum = eval join '+', map { $_->{length} } @data;
+is($sum, 400000, 'max requests limited - all data received');
+
+}
+
 ($frame) = grep { $_->{type} eq "GOAWAY" } @$frames;
 ok($frame, 'max requests limited - GOAWAY');
 is($frame->{last_sid}, $sid, 'max requests limited - GOAWAY last stream');
 
+# graceful shutdown in idle state
+
+$s = Test::Nginx::HTTP2->new();
+$s->{socket}->setsockopt(SOL_SOCKET, SO_RCVBUF, 64*1024) or die $!;
+$s->h2_settings(0, 0x4 => 2**20);
+$s->h2_window(2**21);
+
+$sid = $s->new_stream();
+
+# wait server to finish and close socket if lingering close were disabled
+
+select undef, undef, undef, 0.1;
+
+$t->reload();
+
+select undef, undef, undef, 0.3;
+
+$s->h2_ping("SEE-THIS");
+
+$frames = $s->read(all => [{ sid => $sid, fin => 1 }, { type => 'GOAWAY' }]);
+
+($frame) = grep { $_->{type} eq "HEADERS" } @$frames;
+is($frame->{headers}->{':status'}, 200, 'graceful shutdown in idle');
+
+TODO: {
+local $TODO = 'not yet' if $^O eq 'linux' or $^O eq 'freebsd'
+	and !$t->has_version('1.19.1');
+
+my @data = grep { $_->{type} eq "DATA" } @$frames;
+my $sum = eval join '+', map { $_->{length} } @data;
+is($sum, 400000, 'graceful shutdown in idle - all data received');
+
+($frame) = grep { $_->{type} eq "GOAWAY" } @$frames;
+ok($frame, 'graceful shutdown in idle - GOAWAY');
+is($frame->{last_sid}, $sid, 'graceful shutdown in idle - GOAWAY last stream');
+
+}
+
 ###############################################################################