From 29de0c6e3051b2d9a136c6713c17763dd8cf25ab Mon Sep 17 00:00:00 2001
From: John Bintz <john@coswellproductions.com>
Date: Tue, 27 Nov 2012 07:27:21 -0500
Subject: [PATCH] fixes for connection problems

---
 lib/persistent_selenium/browser.rb | 2 ++
 lib/persistent_selenium/driver.rb  | 6 +++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/lib/persistent_selenium/browser.rb b/lib/persistent_selenium/browser.rb
index e9c8eba..1c68194 100644
--- a/lib/persistent_selenium/browser.rb
+++ b/lib/persistent_selenium/browser.rb
@@ -26,7 +26,9 @@ module PersistentSelenium
 
     def set_app_host(host)
       @app_host = host
+    end
 
+    def load_splash_page
       browser.navigate.to("data:text/html;base64,#{Base64.encode64(starting_page)}")
     end
 
diff --git a/lib/persistent_selenium/driver.rb b/lib/persistent_selenium/driver.rb
index 17ae4b2..93d8993 100644
--- a/lib/persistent_selenium/driver.rb
+++ b/lib/persistent_selenium/driver.rb
@@ -1,10 +1,14 @@
 require 'persistent_selenium'
+require 'capybara/selenium/driver'
+
+# make sure these classes exist on this end
+[ Selenium::WebDriver::Error::StaleElementReferenceError, Selenium::WebDriver::Error::UnhandledError, Selenium::WebDriver::Error::ElementNotVisibleError ]
 
 Capybara.register_driver :persistent_selenium do |app|
   require 'drb'
 
   DRb.start_service
-  browser = DRbObject.new nil, PersistentSelenium.url 
+  browser = DRbObject.new nil, PersistentSelenium.url
 
   server = Capybara::Server.new(app)
   server.boot