1 files changed, 88 insertions, 0 deletions
diff --git a/web/server/h2o/libh2o/t/50compress-hint.t b/web/server/h2o/libh2o/t/50compress-hint.t
new file mode 100644
index 00000000..23e52c50
--- /dev/null
+++ b/web/server/h2o/libh2o/t/50compress-hint.t
@@ -0,0 +1,88 @@
+use strict;
+use warnings;
+use Net::EmptyPort qw(check_port empty_port);
+use Test::More;
+use t::Util;
+
+plan skip_all => 'nc not found'
+    unless prog_exists('nc');
+
+plan skip_all => 'curl not found'
+    unless prog_exists('curl');
+
+my $upstream_port = empty_port();
+$| = 1;
+my $socket = new IO::Socket::INET (
+    LocalHost => '127.0.0.1',
+    LocalPort => $upstream_port,
+    Proto => 'tcp',
+    Listen => 1,
+    Reuse => 1
+);
+die "cannot create socket $!\n" unless $socket;
+
+check_port($upstream_port) or die "can't connect to server socket";
+# accent and close check_port's connection
+my $client_socket = $socket->accept();
+close($client_socket);
+
+my $server = spawn_h2o(<< "EOT");
+compress: ON
+compress-minimum-size: 10
+hosts:
+  default:
+    paths:
+      "/":
+        proxy.reverse.url: http://127.0.0.1:$upstream_port
+EOT
+
+sub doit {
+    my $msg = shift;
+    my $x_compress_header = shift;
+    my $expect_content_encoding = shift;
+    my $accept_gzip = shift;
+
+    my $ae_header = "";
+    if ($accept_gzip) {
+        $ae_header = "-Haccept-encoding:gzip";
+    }
+    open(CURL, "curl $ae_header -Hhost:host.example.com -svo /dev/null http://127.0.0.1:$server->{'port'}/ 2>&1 |");
+
+    my $req;
+    $client_socket = $socket->accept();
+    $client_socket->recv($req, 1024);
+    my $cl = length($msg);
+    $client_socket->send("HTTP/1.1 200 Ok\r\ncontent-length:${cl}\r\ncontent-type:text/html\r\n${x_compress_header}Connection:close\r\n\r\n$msg");
+    close($client_socket);
+
+    my $seen_content_encoding = 0;
+    while(<CURL>) {
+        if (/< content-encoding/) {
+            $seen_content_encoding = 1;
+        }
+    }
+
+    my $neg = "";
+    if ($expect_content_encoding == 0) {
+        $neg = "not ";
+    }
+    ok($seen_content_encoding == $expect_content_encoding, "The body was ${neg}encoded as expected");
+}
+
+doit("This is large enough to be compressed", "", 1, 1);
+doit("This is large enough to be compressed", "x-compress-hint: auto\r\n", 1, 1);
+doit("This is large enough to be compressed", "x-compress-hint: on\r\n", 1, 1);
+doit("This is large enough to be compressed", "x-compress-hint: off\r\n", 0, 1);
+
+doit("too small", "", 0, 1);
+doit("too small", "x-compress-hint: auto\r\n", 0, 1);
+doit("too small", "x-compress-hint: on\r\n", 1, 1);
+doit("too small", "x-compress-hint: off\r\n", 0, 1);
+
+doit("This is large enough to be compressed", "", 0, 0);
+doit("This is large enough to be compressed", "x-compress-hint: auto\r\n", 0, 0);
+doit("This is large enough to be compressed", "x-compress-hint: on\r\n", 0, 0);
+doit("This is large enough to be compressed", "x-compress-hint: off\r\n", 0, 0);
+
+$socket->close();
+done_testing();