fix unit tests

teodesian · Mar 8, 2020 · 28f0207 · 28f0207
1 parent b06fcf8
commit 28f0207
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/t/01-driver.t b/t/01-driver.t
@@ -6,6 +6,7 @@ use Test::More;
 use LWP::UserAgent;
 use Test::LWP::UserAgent;
 use IO::Socket::INET;
+use Test::MockModule;
 use Selenium::Remote::Driver;
 use Selenium::Remote::Mock::Commands;
 use Selenium::Remote::Mock::RemoteConnection;
@@ -133,6 +134,8 @@ GRID_STARTUP: {
     };
     $tua->map_response(qr{(?:grid/api/hub/status|session)}, $ok);
 
+    my $mock = Test::MockModule->new('Selenium::Remote::RemoteConnection');
+    $mock->redefine('check_status', sub { $grid_status_count++; 1 });
     my $grid_driver = Selenium::Remote::Driver->new(ua => $tua);
 
     ok(defined $grid_driver, 'Grid: Object loaded fine using grid/api/hub/status');