t/aggregate/live_component_controller_action_streaming.t

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