lib/WWW/Mechanize/Script/Plugin/ContentSizeTest.pm

   1 package WWW::Mechanize::Script::Plugin::ContentSizeTest;
   2
   3 use strict;
   4 use warnings;
   5
   6 use parent qw(WWW::Mechanize::Script::Plugin);
   7
   8 # ABSTRACT: check for size of received content
   9
  10 our $VERSION = '0.001_003';
  11
  12 use 5.014;
  13
  14 sub check_value_names
  15 {
  16     return qw(min_bytes max_bytes);
  17 }
  18
  19 sub check_response
  20 {
  21     my ( $self, $check, $mech ) = @_;
  22
  23     my $code = 0;
  24     my $msg;
  25
  26     my $min_bytes = $self->get_check_value( $check, "min_bytes" );
  27     my $max_bytes = $self->get_check_value( $check, "max_bytes" );
  28     my $content_len = length $mech->response()->content();
  29
  30     if ( defined($min_bytes) and $min_bytes > $content_len )
  31     {
  32         my $err_code = $self->get_check_value( $check, "min_bytes_code" ) // 1;
  33         $code = &{ $check->{compute_code} }( $code, $err_code );
  34         $msg = "received $content_len bytes exceeds lower threshold ($min_bytes)";
  35     }
  36
  37     if ( defined($max_bytes) and $max_bytes < $content_len )
  38     {
  39         my $err_code = $self->get_check_value( $check, "max_bytes_code" ) // 1;
  40         $code = &{ $check->{compute_code} }( $code, $err_code );
  41         if ($msg)
  42         {
  43             $msg .= " and upper threshold ($max_bytes )";
  44         }
  45         else
  46         {
  47             $msg = "received $content_len bytes exceeds upper limit ($max_bytes)";
  48         }
  49     }
  50
  51     return ( $code, ( $msg ? ($msg) : () ) );
  52 }
  53
  54 1;