driver_helper.tt in ruby_raider-0.8.8

- old
+ new

@@ -19,21 +19,28 @@
     YAML.load_file('config/capabilities.yml')
   end
 end
 <%- else -%>
 # frozen_string_literal: true
+<% if axe? -%>
+require 'axe-selenium'
+<% end -%>
 require 'yaml'
 <% if automation == 'selenium' -%>
 require 'active_support/inflector'
 require 'selenium-webdriver'
 <% else -%>
 require 'appium_lib'
 <% end -%>
 
 module DriverHelper
-  <% if automation == 'selenium' -%>
+  <% if automation == 'selenium' && axe? == false -%>
   def driver(*opts)
     @driver ||= create_driver(*opts)
+  end
+  <% elsif axe? == true -%>
+  def driver(browser = :chrome, js_path = nil, skip_iframes = nil)
+    @driver ||= create_driver(browser, js_path, skip_iframes)
   end
   <% else -%>
   def driver
       @driver ||= create_driver
   end
\ No newline at end of file