diff --git a/lib/cuke-pack/support/wait_for.rb b/lib/cuke-pack/support/wait_for.rb index 9de85de..fd83bed 100644 --- a/lib/cuke-pack/support/wait_for.rb +++ b/lib/cuke-pack/support/wait_for.rb @@ -6,6 +6,9 @@ def _wait_for_exceptions if defined?(Capybara::Driver::Webkit::Node::ElementNotDisplayedError) exceptions << Capybara::Driver::Webkit::Node::ElementNotDisplayedError end + if defined?(Selenium::WebDriver::Error::StaleElementReferenceError) + exceptions << Selenium::WebDriver::Error::StaleElementReferenceError + end exceptions end @@ -15,6 +18,9 @@ def _wait_for_not_exceptions if defined?(Capybara::Driver::Webkit::NodeNotAttachedError) exceptions << Capybara::Driver::Webkit::NodeNotAttachedError end + if defined?(Selenium::WebDriver::Error::StaleElementReferenceError) + exceptions << Selenium::WebDriver::Error::StaleElementReferenceError + end exceptions end