module Cucumber::Rails

def self.bypass_rescue
  ActionController::Base.class_eval do
    alias_method :rescue_action_without_bypass, :rescue_action
    def rescue_action(exception)
      raise exception
    end
  end
  begin
    ActionController::Failsafe.class_eval do
      alias_method :failsafe_response_without_bypass, :failsafe_response
    
      def failsafe_response(exception)
        raise exception
      end
    end
  rescue NameError # Failsafe was introduced in Rails 2.3.2
    ActionController::Dispatcher.class_eval do
      def self.failsafe_response(output, status, exception = nil)
        raise exception
      end
    end
  end
end

def self.failsafe_response(output, status, exception = nil)
  raise exception
end

def self.use_transactional_fixtures
  unless ::Rails.configuration.cache_classes
    warn "WARNING: You have set Rails' config.cache_classes to false (most likely in config/environments/test.rb).  This setting is known to break Cucumber's use_transactional_fixtures method. Set config.cache_classes to true if you want to use transactional fixtures.  For more information see https://rspec.lighthouseapp.com/projects/16211/tickets/165."
  end
  World.use_transactional_fixtures = true
  if defined?(ActiveRecord::Base)
    $__cucumber_toplevel.Before do
      @__cucumber_ar_connection = ActiveRecord::Base.connection
      if @__cucumber_ar_connection.respond_to?(:increment_open_transactions)
        @__cucumber_ar_connection.increment_open_transactions
      else
        ActiveRecord::Base.__send__(:increment_open_transactions)
      end
      @__cucumber_ar_connection.begin_db_transaction
      ActionMailer::Base.deliveries = [] if defined?(ActionMailer::Base)
    end
    
    $__cucumber_toplevel.After do
      @__cucumber_ar_connection.rollback_db_transaction
      if @__cucumber_ar_connection.respond_to?(:decrement_open_transactions)
        @__cucumber_ar_connection.decrement_open_transactions
      else
        ActiveRecord::Base.__send__(:decrement_open_transactions)
      end
    end
  end
end

def failsafe_response(exception)
  raise exception
end

def rescue_action(exception)
  raise exception
end