lib/testgen/generators/project/env.rb.tt in testgen-0.2.3 vs lib/testgen/generators/project/env.rb.tt in testgen-0.2.4

@@ -1,9 +1,18 @@
+<% if with_lib == 'true' -%>
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '../../', 'lib'))
+<% end -%>
+
 require 'rspec-expectations'
 <% unless pageobject_driver.downcase == 'none' -%>
 require 'page-object'
 require 'page-object/page_factory'
 <% end %>
+<% if with_lib == 'true' -%>
+require 'require_all'
+
+require_all 'lib'
+<% end -%>
 
 
 <% unless pageobject_driver.downcase == 'none' -%>
 World(PageObject::PageFactory)
 <% end %>