features/support/env.rb in mechanical-cuke-0.4.3 vs features/support/env.rb in mechanical-cuke-0.5.0

@@ -1,11 +1,17 @@
 $LOAD_PATH.unshift(File.dirname(__FILE__) + '/../../lib')
+require 'open4'
 
-require 'mechanical-cuke'
+if ENV['CAPYBARA']
+  require 'capybara/cucumber'
+  require 'mechanical-cuke/capybara'
+  Capybara.default_driver = :mechanical_cuke
+  Capybara.save_and_open_page_path = File.dirname(__FILE__) + '/../../tmp'
+else
+  require 'mechanical-cuke'
+end
 
 require 'test/unit/assertions'
-
-require 'open4'
 
 World(Test::Unit::Assertions)
 
 Before do
   @server ||= {}