use LWP::Simple;
use Child;
+our $VERSION = '0.01';
+
use Test::More;
BEGIN {
package Test::Builder;
sub new {
my $class = shift;
my %args = @_;
- if( $ENV{SELENIUM_RC_HOST} &&
- $ENV{SELENIUM_RC_PORT} &&
- $ENV{SELENIUM_RC_START} ) {
- $args{selenium_rc}{host} = $ENV{SELENIUM_RC_HOST};
- $args{selenium_rc}{port} = $ENV{SELENIUM_RC_PORT};
- $args{selenium_rc}{start} = $ENV{SELENIUM_RC_START};
- }
- $args{selenium_rc}{xvnc_display} //= '0';
+ my $selrc = ($args{selenium_rc} ||={});
+ $selrc->{$_} = $ENV{"SELENIUM_RC_${\uc $_}"}
+ for grep exists $ENV{"SELENIUM_RC_${\uc $_}"},
+ qw(host port start start_xvnc xvnc_display);
+ $selrc->{xvnc_display} ||= '0';
+ $selrc->{host} ||= 'localhost';
+ $selrc->{port} ||= 4444;
+ $args{browser} ||= '*firefox';
my $self = \%args;
bless $self, $class;
}
sub start_selenium_server {
my($self) = @_;
- if($self->{selenium_rc}{start}) {
- $self->{selenium_rc}{xvnc_server_proc} = Child->new(sub {
- system('ssh', $self->{selenium_rc}{host}, 'vncserver',
- ":$self->{selenium_rc}{xvnc_display}");
- }
- );
- $self->{selenium_rc}{xvnc_server_proc}->start;
- $self->{selenium_rc}{selenium_server_proc} = Child->new(sub {
- system('ssh', $self->{selenium_rc}{host}, 'env',
- "DISPLAY=:$self->{selenium_rc}{xvnc_display}", 'selenium-rc', '-port',
- $self->{selenium_rc}{port} );
+ my $selrc = $self->{selenium_rc};
+ if($selrc->{start}) {
+ my ($host, $display, $port) = @{$selrc}{qw(host xvnc_display port)};
+ my @do_ssh = $host eq 'localhost' ? () : ('ssh', $host);
+ if ($selrc->{start_xvnc}) {
+ $selrc->{xvnc_server_proc} = Child->new(
+ sub {
+ exec(
+ @do_ssh,
+ 'vncserver', ":${display}",
+ );
+ }
+ )->start;
+ $selrc->{xvnc_started} = 1;
+ sleep 3;
+ }
+ $selrc->{selenium_server_proc} = Child->new(
+ sub {
+ exec(
+ @do_ssh,
+ 'env', "DISPLAY=:${display}", 'selenium-rc', '-port', $port
+ )
}
- );
- $self->{selenium_rc}{selenium_server_proc}->start;
+ )->start;
+ sleep 1;
}
my $tries = 0;
while($tries < 5) {
eval {
+ # if we don't create the ::Selenium object ourselves, then
+ # wikifixture shuts the session down after the first test table
+ # is run, at which point KABOOM when you try and run a second one.
$self->{src} = Socialtext::WikiFixture::Selenese->new(
- host => $self->{selenium_rc}{host},
- port => $self->{selenium_rc}{port},
- browser => $self->{browser},
- browser_url => $self->{app_base},
+ selenium => Test::WWW::Selenium->new(
+ host => $self->{selenium_rc}{host},
+ port => $self->{selenium_rc}{port},
+ browser => $self->{browser},
+ browser_url => $self->{app_base},
+ )
);
};
$tries++;
last;
}
}
- die "timed out waiting for selenium server to start at
- http://$self->{selenium_rc}{host}:$self->{selenium_rc}{port}" if $tries == 5;
+ if($tries == 5) {
+ diag "timed out waiting for selenium server to start at
+ http://$self->{selenium_rc}{host}:$self->{selenium_rc}{port}" if $tries == 5;
+ $self->done;
+ die;
+ }
}
sub stop_selenium_server {
my($self) = @_;
- # okay, we're done, kill the server.
- my $url = sprintf
- "http://%s:%s/selenium-server/driver/?cmd=shutDownSeleniumServer",
- $self->{selenium_rc}{host}, $self->{selenium_rc}{port};
- get($url);
- delete $self->{src};
- $self->{selenium_rc}{selenium_server_proc}->wait;
- $self->{selenium_rc}{xvnc_server_proc}->wait;
+ if (my $proc = delete $self->{selenium_rc}{selenium_server_proc}) {
+ my $url = sprintf
+ "http://%s:%s/selenium-server/driver/?cmd=shutDownSeleniumServer",
+ $self->{selenium_rc}{host}, $self->{selenium_rc}{port};
+ eval { get($url); }; # will fail if it never started
+ delete $self->{src};
+ $proc->kill("KILL");
+ }
+ if (delete $self->{selenium_rc}{xvnc_started}) {
+ my $host = $self->{selenium_rc}{host};
+ my @do_ssh = $host eq 'localhost' ? () : ('ssh', $host);
+ Child->new(sub {
+ exec(@do_ssh, 'vncserver', '-kill',
+ ":$self->{selenium_rc}{xvnc_display}");
+ })->start->wait;
+ }
}
sub start_app_server {
my($self) = @_;
- $self->{app_server_proc} = Child->new(sub { exec($self->{app_server_cmd}) } );
- $self->{app_server_proc}->start;
+ my $child = Child->new(sub { exec($self->{app_server_cmd}) } );
+ $self->{app_server_proc} = $child->start;
}
sub stop_app_server {
my($self) = @_;
- $self->{app_server_proc}->complete || $self->{app_server_proc}->kill("KILL");
+ $self->{app_server_proc}->kill("KILL");
}
sub test_directory {
return \@rows;
}
-sub DESTROY {
+sub done {
my($self) = @_;
if(exists $self->{selenium_rc}{xvnc_server_proc} and
exists $self->{selenium_rc}{selenium_server_proc}) {
- $self->{selenium_rc}{xvnc_server_proc}->complete ||
- $self->{selenium_rc}{xvnc_server_proc}->kill("KILL");
- $self->{selenium_rc}{selenium_server_proc}->complete ||
- $self->{selenium_rc}{selenium_server_proc}->kill("KILL");
+ $self->stop_selenium_server;
}
$self->stop_app_server;
}