class ActiveRecordSafetyListener

def scenario_started(*args)
  if defined?(ActiveRecord::Base)
    if ActiveRecord::Base.connection.respond_to?(:increment_open_transactions)
      ActiveRecord::Base.connection.increment_open_transactions
    else
      ActiveRecord::Base.__send__ :increment_open_transactions
    end
  end
  ActiveRecord::Base.connection.begin_db_transaction
end

def scenario_succeeded(*args)
  if defined?(ActiveRecord::Base)
    ActiveRecord::Base.connection.rollback_db_transaction
    if ActiveRecord::Base.connection.respond_to?(:decrement_open_transactions)
      ActiveRecord::Base.connection.decrement_open_transactions
    else
      ActiveRecord::Base.__send__ :decrement_open_transactions
    end
  end
end