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 = system ("$self->{_workdir}/../src/$self->{_cmdline}");
  85
  86     # Shutdown server
  87     kill ('TERM', $pid);
  88     # print "Killed server\n";
  89
  90     # Verify download
  91     unless ($errcode == $self->{_errcode}) {
  92         $result_message = "Test failed: wrong code returned (was: $errcode, expected: $self->{_errcode})\n";
  93     }
  94     if (my $error_str = $self->_verify_download()) {
  95         $result_message = $error_str;
  96     }
  97
  98     # Cleanup
  99     $self->_cleanup();
 100
 101     print $result_message;
 102 }
 103
 104
 105 sub _setup {
 106     my $self = shift;
 107
 108     #print $self->{_name}, "\n";
 109     chdir ($self->{_workdir});
 110
 111     # Create temporary directory
 112     mkdir ($self->{_name});
 113     chdir ($self->{_name});
 114     mkdir ("input");
 115     mkdir ("output");
 116
 117     # Setup existing files
 118     chdir ("output");
 119     foreach my $filename (keys %{$self->{_existing}}) {
 120         open (FILE, ">$filename")
 121             or return "Test failed: cannot open pre-existing file $filename\n";
 122
 123         print FILE $self->{_existing}->{$filename}->{content}
 124             or return "Test failed: cannot write pre-existing file $filename\n";
 125
 126         close (FILE);
 127     }
 128
 129     chdir ("input");
 130     $self->_setup_server();
 131
 132     chdir ($self->{_workdir});
 133 }
 134
 135
 136 sub _cleanup {
 137     my $self = shift;
 138
 139     chdir ($self->{_workdir});
 140     File::Path::rmtree ($self->{_name});
 141 }
 142
 143
 144 sub _verify_download {
 145     my $self = shift;
 146
 147     chdir ("$self->{_workdir}/$self->{_name}/output");
 148
 149     # use slurp mode to read file content
 150     my $old_input_record_separator = $/;
 151     undef $/;
 152
 153     while (my ($filename, $filedata) = each %{$self->{_output}}) {
 154         open (FILE, $filename)
 155             or return "Test failed: file $filename not downloaded\n";
 156
 157         my $content = <FILE>;
 158         $content eq $filedata->{'content'}
 159             or return "Test failed: wrong content for file $filename\n";
 160
 161         if (exists($filedata->{'timestamp'})) {
 162             my ($dev, $ino, $mode, $nlink, $uid, $gid, $rdev, $size,
 163                 $atime, $mtime, $ctime, $blksize, $blocks) = stat FILE;
 164
 165             $mtime == $filedata->{'timestamp'}
 166                 or return "Test failed: wrong timestamp for file $filename\n";
 167         }
 168
 169         close (FILE);
 170     }
 171
 172     $/ = $old_input_record_separator;
 173
 174     # make sure no unexpected files were downloaded
 175     chdir ("$self->{_workdir}/$self->{_name}/output");
 176
 177     __dir_walk('.', sub { push @unexpected_downloads, $_[0] unless (exists $self->{_output}{$_[0]}) }, sub { shift; return @_ } );
 178     if (@unexpected_downloads) {
 179         return "Test failed: unexpected downloaded files [" . join(', ', @unexpected_downloads) . "]\n";
 180     }
 181
 182     return "";
 183 }
 184
 185
 186 sub __dir_walk {
 187     my ($top, $filefunc, $dirfunc) = @_;
 188
 189     my $DIR;
 190
 191     if (-d $top) {
 192         my $file;
 193         unless (opendir $DIR, $top) {
 194             warn "Couldn't open directory $DIR: $!; skipping.\n";
 195             return;
 196         }
 197
 198         my @results;
 199         while ($file = readdir $DIR) {
 200             next if $file eq '.' || $file eq '..';
 201             my $nextdir = $top eq '.' ? $file : "$top/$file";
 202             push @results, __dir_walk($nextdir, $filefunc, $dirfunc);
 203         }
 204
 205         return $dirfunc ? $dirfunc->($top, @results) : () ;
 206     } else {
 207         return $filefunc ? $filefunc->($top) : () ;
 208     }
 209 }
 210
 211 1;
 212
 213 # vim: et ts=4 sw=4
 214