Merge pull request plack#276 from wreis/head_streaming

Add support for streaming app to P::M::Head

lib/Plack/Middleware/Head.pm

@@ -11,7 +11,19 @@ sub call {
 
     $self->response_cb($self->app->($env), sub {
         my $res = shift;
-        $res->[2] = [];
+        if ( $res->[2] ) {
+          $res->[2] = [];
+        }
+        else {
+          my $done;
+          return sub {
+            unless ($done) {
+              return q{};
+            }
+            $done = 1;
+            return defined $_[0] ? q{} : undef;
+          };
+        }
     });
 }
 

t/Plack-Middleware/head_streaming.t

@@ -0,0 +1,32 @@
+use strict;
+use Test::More;
+use Plack::Test;
+use Plack::Builder;
+use HTTP::Request::Common;
+
+my $app = sub {
+    my $env = shift;
+    return sub {
+      my $writer = shift->( [ 200, [
+        'Content-Type' => 'text/plain',
+      ] ] );
+      $writer->write($_) for qw{Hello World};
+      $writer->close;
+    };
+};
+
+$app = builder { enable "Head"; $app };
+
+test_psgi $app, sub {
+    my $cb = shift;
+
+    my $res = $cb->(GET "/");
+    is $res->content, "HelloWorld";
+
+    $res = $cb->(HEAD "/");
+    ok !$res->content;
+    ok(!$res->content_length);
+};
+
+done_testing;
+