sjero.net Git - wget/blob - tests/Test.pm

   1 #!/usr/bin/perl -w
   2
   3 package Test;
   4 $VERSION = 0.01;
   5
   6 use strict;
   7
   8 use Cwd;
   9 use File::Path;
  10
  11 my @unexpected_downloads = ();
  12
  13 {
  14     my %_attr_data = ( # DEFAULT
  15         _cmdline      => "",
  16         _workdir      => Cwd::getcwd(),
  17         _errcode      => 0,
  18         _existing     => {},
  19         _input        => {},
  20         _name         => "",
  21         _output       => {},
  22     );
  23
  24         sub _default_for
  25         {
  26                 my ($self, $attr) = @_;
  27                 $_attr_data{$attr};
  28         }
  29
  30         sub _standard_keys
  31         {
  32                 keys %_attr_data;
  33         }
  34 }
  35
  36
  37 sub new {
  38     my ($caller, %args) = @_;
  39     my $caller_is_obj = ref($caller);
  40     my $class = $caller_is_obj || $caller;
  41     #print STDERR "class = ", $class, "\n";
  42     #print STDERR "_attr_data {workdir} = ", $Test::_attr_data{_workdir}, "\n";
  43     my $self = bless {}, $class;
  44     foreach my $attrname ($self->_standard_keys()) {
  45         #print STDERR "attrname = ", $attrname, " value = ";
  46         my ($argname) = ($attrname =~ /^_(.*)/);
  47         if (exists $args{$argname}) {
  48             #printf STDERR "Setting up $attrname\n";
  49             $self->{$attrname} = $args{$argname};
  50         } elsif ($caller_is_obj) {
  51             #printf STDERR "Copying $attrname\n";
  52             $self->{$attrname} = $caller->{$attrname};
  53         } else {
  54             #printf STDERR "Using default for $attrname\n";
  55             $self->{$attrname} = $self->_default_for($attrname);
  56         }
  57         #print STDERR $attrname, '=', $self->{$attrname}, "\n";
  58     }
  59     #printf STDERR "_workdir default = ", $self->_default_for("_workdir");
  60     return $self;
  61 }
  62
  63
  64 sub run {
  65     my $self = shift;
  66     my $result_message = "Test successful.\n";
  67
  68     printf "Running test $self->{_name}\n";
  69
  70     # Setup
  71     $self->_setup();
  72     chdir ("$self->{_workdir}/$self->{_name}/input");
  73
  74     # Launch server
  75     my $pid = fork();
  76     if($pid == 0) {
  77         $self->_launch_server();
  78     }
  79     # print STDERR "Spawned server with pid: $pid\n";
  80
  81     # Call wget
  82     chdir ("$self->{_workdir}/$self->{_name}/output");
  83     # print "Calling $self->{_cmdline}\n";
  84     my $errcode =
  85         ($self->{_cmdline} =~ m{^/.*})
  86             ? system ($self->{_cmdline})
  87             : system ("$self->{_workdir}/../src/$self->{_cmdline}");
  88
  89     # Shutdown server
  90     kill ('TERM', $pid);
  91     # print "Killed server\n";
  92
  93     # Verify download
  94     unless ($errcode == $self->{_errcode}) {
  95         $result_message = "Test failed: wrong code returned (was: $errcode, expected: $self->{_errcode})\n";
  96     }
  97     if (my $error_str = $self->_verify_download()) {
  98         $result_message = $error_str;
  99     }
 100
 101     # Cleanup
 102     $self->_cleanup();
 103
 104     print $result_message;
 105 }
 106
 107
 108 sub _setup {
 109     my $self = shift;
 110
 111     #print $self->{_name}, "\n";
 112     chdir ($self->{_workdir});
 113
 114     # Create temporary directory
 115     mkdir ($self->{_name});
 116     chdir ($self->{_name});
 117     mkdir ("input");
 118     mkdir ("output");
 119
 120     # Setup existing files
 121     chdir ("output");
 122     foreach my $filename (keys %{$self->{_existing}}) {
 123         open (FILE, ">$filename")
 124             or return "Test failed: cannot open pre-existing file $filename\n";
 125
 126         print FILE $self->{_existing}->{$filename}->{content}
 127             or return "Test failed: cannot write pre-existing file $filename\n";
 128
 129         close (FILE);
 130     }
 131
 132     chdir ("input");
 133     $self->_setup_server();
 134
 135     chdir ($self->{_workdir});
 136 }
 137
 138
 139 sub _cleanup {
 140     my $self = shift;
 141
 142     chdir ($self->{_workdir});
 143     File::Path::rmtree ($self->{_name});
 144 }
 145
 146
 147 sub _verify_download {
 148     my $self = shift;
 149
 150     chdir ("$self->{_workdir}/$self->{_name}/output");
 151
 152     # use slurp mode to read file content
 153     my $old_input_record_separator = $/;
 154     undef $/;
 155
 156     while (my ($filename, $filedata) = each %{$self->{_output}}) {
 157         open (FILE, $filename)
 158             or return "Test failed: file $filename not downloaded\n";
 159
 160         my $content = <FILE>;
 161         $content eq $filedata->{'content'}
 162             or return "Test failed: wrong content for file $filename\n";
 163
 164         if (exists($filedata->{'timestamp'})) {
 165             my ($dev, $ino, $mode, $nlink, $uid, $gid, $rdev, $size,
 166                 $atime, $mtime, $ctime, $blksize, $blocks) = stat FILE;
 167
 168             $mtime == $filedata->{'timestamp'}
 169                 or return "Test failed: wrong timestamp for file $filename\n";
 170         }
 171
 172         close (FILE);
 173     }
 174
 175     $/ = $old_input_record_separator;
 176
 177     # make sure no unexpected files were downloaded
 178     chdir ("$self->{_workdir}/$self->{_name}/output");
 179
 180     __dir_walk('.', sub { push @unexpected_downloads, $_[0] unless (exists $self->{_output}{$_[0]}) }, sub { shift; return @_ } );
 181     if (@unexpected_downloads) {
 182         return "Test failed: unexpected downloaded files [" . join(', ', @unexpected_downloads) . "]\n";
 183     }
 184
 185     return "";
 186 }
 187
 188
 189 sub __dir_walk {
 190     my ($top, $filefunc, $dirfunc) = @_;
 191
 192     my $DIR;
 193
 194     if (-d $top) {
 195         my $file;
 196         unless (opendir $DIR, $top) {
 197             warn "Couldn't open directory $DIR: $!; skipping.\n";
 198             return;
 199         }
 200
 201         my @results;
 202         while ($file = readdir $DIR) {
 203             next if $file eq '.' || $file eq '..';
 204             my $nextdir = $top eq '.' ? $file : "$top/$file";
 205             push @results, __dir_walk($nextdir, $filefunc, $dirfunc);
 206         }
 207
 208         return $dirfunc ? $dirfunc->($top, @results) : () ;
 209     } else {
 210         return $filefunc ? $filefunc->($top) : () ;
 211     }
 212 }
 213
 214 1;
 215
 216 # vim: et ts=4 sw=4
 217