Add Watir support

bootstraponline · May 5, 2015 · da5276e · da5276e
1 parent 5cc9ddf
commit da5276e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/angular_webdriver/protractor/protractor.rb b/lib/angular_webdriver/protractor/protractor.rb
@@ -17,6 +17,9 @@ def initialize opts={}
     @driver = opts[:driver]
     raise 'Must supply Selenium::WebDriver' unless @driver
 
+    watir = defined?(Watir::Browser) && @driver.is_a?(Watir::Browser)
+    @driver = watir ? @driver.driver : @driver
+
     # The css selector for an element on which to find Angular. This is usually
     # 'body' but if your ng-app is on a subsection of the page it may be
     # a subelement.