X-Git-Url: http://sjero.net/git/?p=wget;a=blobdiff_plain;f=tests%2FTest.pm;h=930c54e938264480215361fea33e0a45f0e93219;hp=a84f8624ad286e221362b655f10f7f9e72df129f;hb=4ad743db989fbd62e541c15ab3e65c6ac8d7715d;hpb=823228830e57766ebabe529b75765816cb2507dc

diff --git a/tests/Test.pm b/tests/Test.pm
index a84f8624..930c54e9 100755
--- a/tests/Test.pm
+++ b/tests/Test.pm
@@ -5,9 +5,19 @@ $VERSION = 0.01;
 
 use strict;
 
+use Cwd;
+use File::Path;
+
+my @unexpected_downloads = ();
 
 {
     my %_attr_data = ( # DEFAULT
+        _cmdline      => "",
+        _cwd          => Cwd::getcwd(),
+        _errcode      => 0,
+        _input        => {},
+        _name         => "",
+        _output       => {},
     );
     
 	sub _default_for
@@ -22,27 +32,169 @@ use strict;
 	}
 }
 
+
 sub new {
     my ($caller, %args) = @_;
     my $caller_is_obj = ref($caller);
     my $class = $caller_is_obj || $caller;
+    #print STDERR "class = ", $class, "\n";
+    #print STDERR "_attr_data {cwd} = ", $Test::_attr_data{_cwd}, "\n";
     my $self = bless {}, $class;
     foreach my $attrname ($self->_standard_keys()) {
+        #print STDERR "attrname = ", $attrname, " value = ";
         my ($argname) = ($attrname =~ /^_(.*)/);
         if (exists $args{$argname}) {
             #printf STDERR "Setting up $attrname\n";
             $self->{$attrname} = $args{$argname};
         } elsif ($caller_is_obj) {
             #printf STDERR "Copying $attrname\n";
-            $self->{$attrname} = $caller->{$argname};
+            $self->{$attrname} = $caller->{$attrname};
         } else {
             #printf STDERR "Using default for $attrname\n";
-            $self->{$attrname} = $self->_default_for($argname);
+            $self->{$attrname} = $self->_default_for($attrname);
         }
+        #print STDERR $attrname, '=', $self->{$attrname}, "\n";
     }
+    #printf STDERR "_cwd default = ", $self->_default_for("_cwd");
     return $self;
 }
 
+
+sub run {
+    my $self = shift;
+    my $result_message = "Test successful.\n";
+   
+    printf "Running test $self->{_name}\n";
+    
+    # Setup 
+    $self->_setup();
+    chdir ("$self->{_cwd}/$self->{_name}/input");
+    
+    # Launch server
+    my $pid = fork();
+    if($pid == 0) {
+        $self->_launch_server();
+    }
+    # print STDERR "Spawned server with pid: $pid\n"; 
+    
+    # Call wget
+    chdir ("$self->{_cwd}/$self->{_name}/output");
+    # print "Calling $self->{_cmdline}\n";
+    my $errcode = system ("$self->{_cwd}/../src/$self->{_cmdline}");
+
+    # Shutdown server
+    kill ('TERM', $pid);
+    # print "Killed server\n";
+
+    # Verify download
+    unless ($errcode == $self->{_errcode}) {
+        $result_message = "Test failed: wrong code returned (was: $errcode, expected: $self->{_errcode})\n";
+    }
+    if (my $error_str = $self->_verify_download()) {
+        $result_message = $error_str;
+    }
+
+    # Cleanup
+    $self->_cleanup();
+
+    print $result_message;
+}
+
+
+sub _setup {
+    my $self = shift;
+
+    #print $self->{_name}, "\n";
+    chdir ($self->{_cwd});
+
+    # Create temporary directory
+    mkdir ($self->{_name});
+    chdir ($self->{_name});
+    mkdir ("input");
+    mkdir ("output");
+    chdir ("input");
+
+    $self->_setup_server();
+
+    chdir ($self->{_cwd});
+}
+
+
+sub _cleanup {
+    my $self = shift;
+
+    chdir ($self->{_cwd});
+    File::Path::rmtree ($self->{_name});
+}
+
+
+sub _verify_download {
+    my $self = shift;
+
+    chdir ("$self->{_cwd}/$self->{_name}/output");
+    
+    # use slurp mode to read file content
+    my $old_input_record_separator = $/;
+    undef $/;
+    
+    while (my ($filename, $filedata) = each %{$self->{_output}}) {
+        open (FILE, $filename) 
+            or return "Test failed: file $filename not downloaded\n";
+        
+        my $content = <FILE>;
+        $content eq $filedata->{'content'} 
+            or return "Test failed: wrong content for file $filename\n";
+
+        if (exists($filedata->{'timestamp'})) {
+            my ($dev, $ino, $mode, $nlink, $uid, $gid, $rdev, $size,
+                $atime, $mtime, $ctime, $blksize, $blocks) = stat FILE;
+
+            $mtime == $filedata->{'timestamp'} 
+                or return "Test failed: wrong timestamp for file $filename\n";
+        }
+        
+        close (FILE);
+    } 
+    
+    $/ = $old_input_record_separator;    
+
+    # make sure no unexpected files were downloaded
+    chdir ("$self->{_cwd}/$self->{_name}/output");
+
+    __dir_walk('.', sub { push @unexpected_downloads, $_[0] unless (exists $self->{_output}{$_[0]}) }, sub { shift; return @_ } );
+    if (@unexpected_downloads) { 
+        return "Test failed: unexpected downloaded files [" . join(', ', @unexpected_downloads) . "]\n";
+    }
+
+    return "";
+}
+
+
+sub __dir_walk {
+    my ($top, $filefunc, $dirfunc) = @_;
+
+    my $DIR;
+
+    if (-d $top) {
+        my $file;
+        unless (opendir $DIR, $top) {
+            warn "Couldn't open directory $DIR: $!; skipping.\n";
+            return;
+        }
+
+        my @results;
+        while ($file = readdir $DIR) {
+            next if $file eq '.' || $file eq '..';
+            my $nextdir = $top eq '.' ? $file : "$top/$file";
+            push @results, __dir_walk($nextdir, $filefunc, $dirfunc);
+        }
+
+        return $dirfunc ? $dirfunc->($top, @results) : () ;
+    } else {
+        return $filefunc ? $filefunc->($top) : () ;
+    }
+}
+
 1;
 
 # vim: et ts=4 sw=4