lib/generators/templates/helpers/driver_helper.tt in ruby_raider-0.4.2 vs lib/generators/templates/helpers/driver_helper.tt in ruby_raider-0.4.3

@@ -7,10 +7,23 @@
 require 'appium_lib'
 <% end -%>
 
 module Raider
   module DriverHelper
-      attr_reader :driver
+    attr_reader :driver
 
-      <%= ERB.new(File.read(File.expand_path('./partials/driver_and_options.tt', __dir__))).result(binding).strip! %>
+    <%= ERB.new(File.read(File.expand_path('./partials/driver_and_options.tt', __dir__))).result(binding).strip! %>
+
+  <% if automation == 'cross_platform' -%>
+  private
+
+    def config
+      YAML.load_file('config/config.yml')
+    end
+<% end -%>
+<% if automation != 'selenium'  -%>
+    def caps
+      YAML.load_file('config/capabilities.yml')
+    end
+    <% end -%>
   end
 end
\ No newline at end of file