t/aggregate/live_component_controller_action_streaming.t

   1 #!perl
   2
   3 use strict;
   4 use warnings;
   5
   6 use FindBin;
   7 use lib "$FindBin::Bin/../lib";
   8
   9 our $iters;
  10
  11 BEGIN { $iters = $ENV{CAT_BENCH_ITERS} || 1; }
  12
  13 use Test::More;
  14 use Catalyst::Test 'TestApp';
  15
  16 if ( $ENV{CAT_BENCHMARK} ) {
  17     require Benchmark;
  18     Benchmark::timethis( $iters, \&run_tests );
  19 }
  20 else {
  21     for ( 1 .. $iters ) {
  22         run_tests();
  23     }
  24 }
  25
  26 sub run_tests {
  27     # test direct streaming
  28     {
  29         ok( my $response = request('http://localhost/streaming'), 'Request' );
  30         ok( $response->is_success, 'Response Successful 2xx' );
  31         is( $response->content_type, 'text/plain', 'Response Content-Type' );
  32         is( $response->header('X-Test-Header'), 'valid', 'Headers sent properly' );
  33         is( $response->header('X-Test-Header-Call-Count'), 1);
  34
  35         SKIP:
  36         {
  37             if ( $ENV{CATALYST_SERVER} ) {
  38                 skip "Using remote server", 1;
  39             }
  40
  41             ok(!defined $response->content_length, 'No Content-Length for streaming responses');
  42             is(length $response->content, 12, 'Response content' );
  43         }
  44
  45         is( $response->content,, <<'EOF', 'Content is a stream' );
  46 foo
  47 bar
  48 baz
  49 EOF
  50     }
  51
  52     # test streaming by passing a handle to $c->res->body
  53   SKIP:
  54     {
  55         if ( $ENV{CATALYST_SERVER} ) {
  56             skip "Using remote server", 10;
  57         }
  58
  59         my $file = "$FindBin::Bin/../lib/TestApp/Controller/Action/Streaming.pm";
  60         my $fh = IO::File->new( $file, 'r' );
  61         my $buffer;
  62         if ( defined $fh ) {
  63             $fh->read( $buffer, 2048 );
  64             $fh->close;
  65         }
  66
  67         ok( my $response = request('http://localhost/action/streaming/body'),
  68             'Request' );
  69         ok( $response->is_success, 'Response Successful 2xx' );
  70         is( $response->content_type, 'text/plain', 'Response Content-Type' );
  71         is( $response->content_length, -s $file, 'Response Content-Length' );
  72         is( $response->header('X-Test-Header'), 'valid', 'Headers sent properly' );
  73         is( $response->header('X-Test-Header-Call-Count'), 1);
  74         is( $response->content, $buffer, 'Content is read from filehandle' );
  75
  76         ok( $response = request('http://localhost/action/streaming/body_glob'),
  77             'Request' );
  78         ok( $response->is_success, 'Response Successful 2xx' );
  79         is( $response->content_type, 'text/plain', 'Response Content-Type' );
  80         is( $response->content_length, -s $file, 'Response Content-Length' );
  81         is( $response->header('X-Test-Header'), 'valid', 'Headers sent properly' );
  82         is( $response->header('X-Test-Header-Call-Count'), 1);
  83         is( $response->content, $buffer, 'Content is read from filehandle' );
  84     }
  85
  86     {
  87         my $size = 128 * 1024; # more than one read with the default chunksize
  88
  89         ok( my $response = request('http://localhost/action/streaming/body_large'), 'Request' );
  90         ok( $response->is_success, 'Response Successful 2xx' );
  91         is( $response->content_type, 'text/plain', 'Response Content-Type' );
  92         is( $response->header('X-Test-Header'), 'valid', 'Headers sent properly' );
  93         is( $response->header('X-Test-Header-Call-Count'), 1);
  94         is( $response->content_length, $size, 'Response Content-Length' );
  95         is( $response->content, "\0" x $size, 'Content is read from filehandle' );
  96     }
  97 }
  98
  99 done_testing;