initialize Screenshot class with Browser instance not a Driver instance

watir · May 31, 2018 · 0943e7e · 0943e7e
1 parent 274042b
commit 0943e7e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/lib/watir/browser.rb b/lib/watir/browser.rb
@@ -290,7 +290,7 @@ def send_keys(*args)
     #
 
     def screenshot
-      Screenshot.new driver
+      Screenshot.new self
     end
 
     #

diff --git a/lib/watir/screenshot.rb b/lib/watir/screenshot.rb
@@ -1,8 +1,14 @@
 module Watir
   class Screenshot
 
-    def initialize(driver)
-      @driver = driver
+    def initialize(browser)
+      if browser.is_a? Selenium::WebDriver::Driver
+        Watir.logger.deprecate "Initializing `Watir::Screenshot` with a `Selenium::Driver` instance", "a `Watir::Browser` instance"
+        @driver = browser
+      else
+        @browser = browser
+        @driver = browser.wd
+      end
     end
 
     #